89 lines
No EOL
2.6 KiB
Python
89 lines
No EOL
2.6 KiB
Python
import os
|
|
import requests
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
# Configure logging
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
|
|
|
# Environment variables
|
|
IMMICH_API_KEY = os.getenv('IMMICH_API_KEY')
|
|
IMMICH_INSTANCE_URL = os.getenv('IMMICH_INSTANCE_URL')
|
|
# ALBUM_NAME = os.getenv('IMMICH_ALBUM_NAME')
|
|
ALBUM_ID = os.getenv('IMMICH_ALBUM_ID')
|
|
DOWNLOAD_PATH = Path(os.getenv('IMMICH_DOWNLOAD_PATH', './downloads'))
|
|
|
|
|
|
# def get_album_id():
|
|
# """Get album ID by name from Immich API"""
|
|
# headers = {'x-api-key': IMMICH_API_KEY}
|
|
# response = requests.get(
|
|
# f"{IMMICH_INSTANCE_URL}/albums",
|
|
# headers=headers,
|
|
# params={'albumName': ALBUM_NAME}
|
|
# )
|
|
# response.raise_for_status()
|
|
#
|
|
# for album in response.json():
|
|
# if album['albumName'] == ALBUM_NAME:
|
|
# return album['id']
|
|
# raise ValueError(f"Album '{ALBUM_NAME}' not found")
|
|
|
|
|
|
def get_album_assets(album_id):
|
|
"""Retrieve all assets in album with pagination"""
|
|
headers = {'x-api-key': IMMICH_API_KEY}
|
|
assets = []
|
|
page = 1
|
|
|
|
while True:
|
|
response = requests.get(
|
|
f"{IMMICH_INSTANCE_URL}/albums/{album_id}",
|
|
headers=headers,
|
|
params={'page': page}
|
|
)
|
|
response.raise_for_status()
|
|
|
|
assets.extend(response.json()["assets"])
|
|
total_pages = int(response.headers.get('X-Pagination-Count', 1))
|
|
if page >= total_pages:
|
|
break
|
|
page += 1
|
|
# break
|
|
|
|
return assets
|
|
|
|
|
|
def download_assets(assets):
|
|
"""Download missing assets to target directory"""
|
|
DOWNLOAD_PATH.mkdir(parents=True, exist_ok=True)
|
|
headers = {'x-api-key': IMMICH_API_KEY}
|
|
|
|
for asset in assets:
|
|
filename = f"{asset['originalFileName']}"
|
|
filepath = DOWNLOAD_PATH / filename
|
|
|
|
if not filepath.exists():
|
|
logging.info(f"Downloading {filename}")
|
|
response = requests.get(
|
|
f"{IMMICH_INSTANCE_URL}/assets/{asset['id']}/original",
|
|
headers=headers,
|
|
stream=True
|
|
)
|
|
response.raise_for_status()
|
|
|
|
with open(filepath, 'wb') as f:
|
|
for chunk in response.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
else:
|
|
logging.debug(f"Skipping existing file {filename}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
assets = get_album_assets(ALBUM_ID)
|
|
download_assets(assets)
|
|
logging.info(f"Sync complete. {len(assets)} assets processed")
|
|
except Exception as e:
|
|
logging.error(f"Sync failed: {str(e)}")
|
|
raise |