130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
向已创建的相册添加照片 - 卡若AI·卡资（金）
|
||
使用AppleScript批量添加照片到相册
|
||
"""
|
||
import json
|
||
import os
|
||
import subprocess
|
||
import time
|
||
|
||
OUTPUT_DIR = os.path.expanduser("~/Documents/个人/卡若AI/01_卡资（金）/照片分类")
|
||
|
||
def load_classification():
|
||
with open(os.path.join(OUTPUT_DIR, "classification.json"), "r") as f:
|
||
return json.load(f)
|
||
|
||
def add_photos_to_album_batch(album_name, uuids, batch_size=100):
|
||
"""分批向相册添加照片"""
|
||
total = len(uuids)
|
||
added = 0
|
||
|
||
for i in range(0, total, batch_size):
|
||
batch = uuids[i:i+batch_size]
|
||
uuid_list = '", "'.join(batch)
|
||
|
||
script = f'''
|
||
tell application "Photos"
|
||
set targetAlbum to missing value
|
||
repeat with a in albums
|
||
if name of a is "{album_name}" then
|
||
set targetAlbum to a
|
||
exit repeat
|
||
end if
|
||
end repeat
|
||
|
||
if targetAlbum is not missing value then
|
||
set uuidList to {{"{uuid_list}"}}
|
||
set photosToAdd to {{}}
|
||
repeat with uuid in uuidList
|
||
try
|
||
set thePhoto to media item id uuid
|
||
set end of photosToAdd to thePhoto
|
||
end try
|
||
end repeat
|
||
|
||
if (count of photosToAdd) > 0 then
|
||
add photosToAdd to targetAlbum
|
||
return count of photosToAdd
|
||
end if
|
||
end if
|
||
return 0
|
||
end tell
|
||
'''
|
||
try:
|
||
result = subprocess.run(
|
||
['osascript', '-e', script],
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=120
|
||
)
|
||
if result.returncode == 0:
|
||
try:
|
||
added += int(result.stdout.strip())
|
||
except:
|
||
added += len(batch)
|
||
except subprocess.TimeoutExpired:
|
||
print(f" ⚠️ 批次超时，跳过")
|
||
except Exception as e:
|
||
print(f" ⚠️ 错误: {e}")
|
||
|
||
# 进度显示
|
||
progress = min(i + batch_size, total)
|
||
print(f" 进度: {progress}/{total} ({progress*100//total}%)", end='\r')
|
||
time.sleep(0.5)
|
||
|
||
print(f" ✓ 完成: {added}/{total}张 ")
|
||
return added
|
||
|
||
def main():
|
||
print("🚀 照片分类自动归档")
|
||
print("="*60)
|
||
|
||
data = load_classification()
|
||
|
||
# 相册映射
|
||
album_mapping = {
|
||
("个人照片", "手机拍摄"): "📱 手机拍摄",
|
||
("个人照片", "自拍美颜"): "🤳 自拍美颜",
|
||
("截图收藏", "屏幕截图"): "📸 屏幕截图",
|
||
("视频素材", "手机拍摄视频"): "🎬 手机视频",
|
||
("视频素材", "剪辑导出"): "🎬 剪辑导出",
|
||
("视频素材", "屏幕录制"): "🎬 屏幕录制",
|
||
("视频素材", "抖音视频"): "🎬 抖音视频",
|
||
("内容素材", "网络图片"): "🌐 网络图片",
|
||
}
|
||
|
||
year_mapping = {
|
||
"2018年及以前": "📅 2018年及以前",
|
||
"2019年回顾": "📅 2019年回顾",
|
||
"2020年回顾": "📅 2020年回顾",
|
||
"2021年回顾": "📅 2021年回顾",
|
||
"2022年回顾": "📅 2022年回顾",
|
||
"2023年回顾": "📅 2023年回顾",
|
||
"2024年照片": "📅 2024年照片",
|
||
"2025年照片": "📅 2025年照片",
|
||
}
|
||
|
||
# 1. 按类别添加照片
|
||
print("\n📁 按内容类型分类...")
|
||
for (category, sub_album), target_album in album_mapping.items():
|
||
if category in data["categories"] and sub_album in data["categories"][category]:
|
||
uuids = data["categories"][category][sub_album]
|
||
print(f"\n → {target_album} ({len(uuids)}张)")
|
||
add_photos_to_album_batch(target_album, uuids)
|
||
|
||
# 2. 按年份添加照片
|
||
print("\n📅 按年份分类...")
|
||
for year_album, target_album in year_mapping.items():
|
||
if year_album in data["year_albums"]:
|
||
uuids = data["year_albums"][year_album]
|
||
print(f"\n → {target_album} ({len(uuids)}张)")
|
||
add_photos_to_album_batch(target_album, uuids)
|
||
|
||
print("\n" + "="*60)
|
||
print("✅ 照片归档完成！")
|
||
print("="*60)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|