AddOfflineToAlbum.py

#!/usr/bin/env python3

# Note: you might need to run "pip install requests halo tabulate tqdm" if these dependencies are missing on your machine

import argparse
import json
import requests
from urllib.parse import urlparse
from halo import Halo
from tabulate import tabulate
from tqdm import tqdm

def parse_arguments():
    parser = argparse.ArgumentParser(description='Fetch file report and add orphaned media assets to Album from Immich.')
    # Make API key and Immich address arguments optional
    parser.add_argument('--admin_apikey', help='Immich admin API key for fetching reports', nargs='?', default=None)
    parser.add_argument('--user_apikey', help='User-specific Immich API key for deletion', nargs='?', default=None)
    parser.add_argument('--user_id', help='Immich admin Profile ID', nargs='?', default=None)
    parser.add_argument('--immichaddress', help='Full address for Immich, including protocol and port', nargs='?', default=None)
    parser.add_argument('--no_prompt', action='store_true', help='Delete orphaned media assets without confirmation')
    return parser.parse_args()

def filter_entities(response_json, entity_type):
    return [
        {'pathValue': entity['pathValue'], 'entityId': entity['entityId'], 'entityType': entity['entityType']}
        for entity in response_json.get('orphans', []) if entity.get('entityType') == entity_type
    ]

def main():
    args = parse_arguments()

    # Prompt for admin API key if not provided
    admin_api_key = args.admin_apikey if args.admin_apikey else input('Enter the Immich admin API key: ')
    # Prompt for user API key if not provided
    user_api_key = args.user_apikey if args.user_apikey else input('Enter the Immich user API key for deletion: ')
    # Prompt for USER ID if not provided
    user_id = args.user_id if args.user_id else input('Enter the Immich ProfileID: ')
    # Prompt for Immich address if not provided
    immich_server = args.immichaddress if args.immichaddress else input('Enter the full web address for Immich, including protocol and port: ')

    if not admin_api_key or not user_api_key:
        print("Both admin and user API keys are required.")
        return

    immich_parsed_url = urlparse(immich_server)
    base_url = f'{immich_parsed_url.scheme}://{immich_parsed_url.netloc}'
    api_url = f'{base_url}/api'
    file_report_url = api_url + '/reports'
    headers = {'x-api-key': admin_api_key}

    print()
    spinner = Halo(text='Retrieving list of orphaned media assets...', spinner='dots')
    spinner.start()

    try:
        response = requests.get(file_report_url, headers=headers)
        response.raise_for_status()
        spinner.succeed('Success!')
    except requests.exceptions.RequestException as e:
        spinner.fail(f'Failed to fetch assets: {str(e)}')
        return

    orphan_media_assets = filter_entities(response.json(), 'asset')
    num_entries = len(orphan_media_assets)

    if num_entries == 0:
        print('No orphaned media assets found; exiting.')
        return

    if not args.no_prompt:
        table_data = [[asset['pathValue'], asset['entityId']] for asset in orphan_media_assets]
        print(tabulate(table_data, headers=['Path Value', 'Entity ID'], tablefmt='pretty'))
        print()

        summary = f'There {"is" if num_entries == 1 else "are"} {num_entries} orphaned media asset{"s" if num_entries != 1 else ""}. Would you like to add to album {"them" if num_entries != 1 else "it"} in Immich? (yes/no): '
        user_input = input(summary).lower()
        print()

        if user_input not in ('y', 'yes'):
            print('Exiting without making any changes.')
            return

    headers['x-api-key'] = user_api_key  # Use user API key for deletion
    with tqdm(total=num_entries, desc="Managing orphaned media assets", unit="asset") as progress_bar:
        for asset in orphan_media_assets:
            entity_id = asset['entityId']
            asset_url = f'{api_url}/albums'
            url = f'{api_url}/albums'
            payload = json.dumps({'force': True, 'ids': [entity_id]})
            payload = json.dumps({
            	"albumName": "OfflineAssetPyScript",
            	"albumUsers": [
            	{
            		"role": "editor",
            		"userId": "{user_id}"
            	}
            	],
            	"assetIds": [
            		"{entity_id}"
            	],
            	"description": "Files From Script"
            	})
            headers = {
            		'Content-Type': 'application/json',
            		'Accept': 'application/json'
            	}  
            headers = {'Content-Type': 'application/json', 'x-api-key': user_api_key}
            try:
                response = requests.request("POST", url, headers=headers, data=payload)
                response.raise_for_status()
            except requests.exceptions.HTTPError as e:
                if response.status_code == 400:
                    print(f"Failed to manage asset {entity_id} due to potential API key mismatch. Ensure you're using the asset owners API key as the User API key.")
                else:
                    print(f"Failed to manage asset {entity_id}: {str(e)}")
                continue
            progress_bar.update(1)
    print('Orphaned media assets managed successfully!')

if __name__ == '__main__':
    main()