205 lines
5.8 KiB
Python
205 lines
5.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
生成AppleScript来创建照片相册 - 卡若AI·卡资（金）
|
||
"""
|
||
import json
|
||
import os
|
||
import subprocess
|
||
|
||
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 create_album_applescript(album_name, uuids, folder_name=None):
|
||
"""生成创建单个相册的AppleScript"""
|
||
if not uuids:
|
||
return ""
|
||
|
||
# 限制每个相册最多处理的UUID数量（避免AppleScript太长）
|
||
max_uuids = 500
|
||
uuids_to_add = uuids[:max_uuids]
|
||
|
||
uuid_list = '", "'.join(uuids_to_add)
|
||
|
||
script = f'''
|
||
-- 创建相册: {album_name}
|
||
tell application "Photos"
|
||
-- 检查相册是否存在
|
||
set albumExists to false
|
||
repeat with a in albums
|
||
if name of a is "{album_name}" then
|
||
set albumExists to true
|
||
set targetAlbum to a
|
||
exit repeat
|
||
end if
|
||
end repeat
|
||
|
||
-- 如果不存在则创建
|
||
if not albumExists then
|
||
set targetAlbum to make new album named "{album_name}"
|
||
end if
|
||
|
||
-- 添加照片（通过UUID）
|
||
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
|
||
end if
|
||
end tell
|
||
'''
|
||
return script
|
||
|
||
def create_folder_structure_script():
|
||
"""创建文件夹结构的AppleScript"""
|
||
folders = ["📱 个人照片", "📷 内容素材", "🎬 视频素材", "📸 截图收藏", "💬 社交记录", "📅 年份回顾"]
|
||
|
||
script = '''
|
||
-- 创建相册文件夹结构
|
||
tell application "Photos"
|
||
'''
|
||
for folder in folders:
|
||
script += f'''
|
||
-- 创建文件夹: {folder}
|
||
try
|
||
set folderExists to false
|
||
repeat with f in folders
|
||
if name of f is "{folder}" then
|
||
set folderExists to true
|
||
exit repeat
|
||
end if
|
||
end repeat
|
||
if not folderExists then
|
||
make new folder named "{folder}"
|
||
end if
|
||
on error
|
||
-- 忽略错误继续
|
||
end try
|
||
'''
|
||
script += '''
|
||
end tell
|
||
'''
|
||
return script
|
||
|
||
def generate_batch_scripts(data):
|
||
"""生成批量创建相册的脚本"""
|
||
scripts_dir = os.path.join(OUTPUT_DIR, "applescripts")
|
||
os.makedirs(scripts_dir, exist_ok=True)
|
||
|
||
# 相册映射到文件夹
|
||
folder_mapping = {
|
||
"个人照片": "📱 个人照片",
|
||
"内容素材": "📷 内容素材",
|
||
"视频素材": "🎬 视频素材",
|
||
"截图收藏": "📸 截图收藏",
|
||
"社交记录": "💬 社交记录",
|
||
"其他": "📷 内容素材"
|
||
}
|
||
|
||
# 1. 创建文件夹结构脚本
|
||
folder_script = create_folder_structure_script()
|
||
with open(os.path.join(scripts_dir, "01_create_folders.scpt"), "w") as f:
|
||
f.write(folder_script)
|
||
|
||
script_index = 2
|
||
|
||
# 2. 按类别创建相册脚本
|
||
for category, albums in data["categories"].items():
|
||
folder_name = folder_mapping.get(category, "📷 内容素材")
|
||
|
||
for album_name, uuids in albums.items():
|
||
if len(uuids) < 10: # 少于10张的跳过
|
||
continue
|
||
|
||
# 生成相册全名
|
||
full_album_name = f"{album_name}"
|
||
|
||
script = create_album_applescript(full_album_name, uuids, folder_name)
|
||
|
||
filename = f"{script_index:02d}_{category}_{album_name}.scpt"
|
||
filename = filename.replace("/", "_").replace(" ", "_")
|
||
|
||
with open(os.path.join(scripts_dir, filename), "w") as f:
|
||
f.write(script)
|
||
|
||
script_index += 1
|
||
|
||
# 3. 年份相册
|
||
for year_album, uuids in data["year_albums"].items():
|
||
script = create_album_applescript(year_album, uuids, "📅 年份回顾")
|
||
|
||
filename = f"{script_index:02d}_年份_{year_album}.scpt"
|
||
with open(os.path.join(scripts_dir, filename), "w") as f:
|
||
f.write(script)
|
||
|
||
script_index += 1
|
||
|
||
print(f"✅ 已生成 {script_index - 1} 个AppleScript脚本")
|
||
print(f" 保存位置: {scripts_dir}")
|
||
|
||
return scripts_dir
|
||
|
||
def create_master_runner():
|
||
"""创建主执行脚本"""
|
||
scripts_dir = os.path.join(OUTPUT_DIR, "applescripts")
|
||
|
||
runner_script = '''#!/bin/bash
|
||
# 照片相册批量创建脚本 - 卡若AI
|
||
# 使用方法: ./run_all.sh
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
echo "🚀 开始创建照片相册..."
|
||
|
||
# 先打开照片应用
|
||
open -a "Photos"
|
||
sleep 3
|
||
|
||
# 依次执行所有AppleScript
|
||
for script in "$SCRIPT_DIR"/*.scpt; do
|
||
if [ -f "$script" ]; then
|
||
name=$(basename "$script")
|
||
echo "📁 执行: $name"
|
||
osascript "$script" 2>/dev/null
|
||
sleep 1
|
||
fi
|
||
done
|
||
|
||
echo "✅ 相册创建完成！"
|
||
'''
|
||
|
||
runner_path = os.path.join(scripts_dir, "run_all.sh")
|
||
with open(runner_path, "w") as f:
|
||
f.write(runner_script)
|
||
os.chmod(runner_path, 0o755)
|
||
|
||
print(f"✅ 主执行脚本: {runner_path}")
|
||
return runner_path
|
||
|
||
def main():
|
||
print("📊 加载分类数据...")
|
||
data = load_classification()
|
||
|
||
print("📝 生成AppleScript脚本...")
|
||
scripts_dir = generate_batch_scripts(data)
|
||
|
||
print("📝 创建主执行脚本...")
|
||
runner_path = create_master_runner()
|
||
|
||
print(f"\n{'='*60}")
|
||
print("🎉 脚本生成完成！")
|
||
print(f"{'='*60}")
|
||
print(f"\n执行以下命令开始创建相册:")
|
||
print(f" cd {scripts_dir} && ./run_all.sh")
|
||
print(f"\n或者手动在「脚本编辑器」中打开执行单个脚本")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|