immich_asset_add_time_buckets_061324

import requests
import json
import time

def login_to_api(server_url, email, password):
    print("Attempting to log in...")
    login_url = f"{server_url}/api/auth/login"
    login_payload = json.dumps({"email": email, "password": password})
    login_headers = {'Content-Type': 'application/json'}
    response = requests.post(login_url, headers=login_headers, data=login_payload)
    print(response.text)

    if response.status_code in [200, 201]:
        print(f"Login successful: {response.status_code}")
        return response.json()
    else:
        print(f"Failed to login. Status code: {response.status_code}")
        return None

def validate_token(server_url, token):
    url = f"{server_url}/api/auth/validateToken"
    headers = {'Authorization': f'Bearer {token}', 'Accept': 'application/json'}
    response = requests.post(url, headers=headers)
    if response.status_code == 200:
        return response.json().get('authStatus', False)
    else:
        print(f"Failed to validate token. Status code: {response.status_code}, Response text: {response.text}")
        return False

def get_person_assets(server_url, token, person_id):
    url = f"{server_url}/api/people/{person_id}/assets"
    headers = {'Authorization': f'Bearer {token}', 'Accept': 'application/json'}
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        return response.json()
    else:
        print(f"Failed to fetch assets. Status code: {response.status_code}, Response text: {response.text}")
        return []

def get_time_buckets(server_url, token, user_id, size='MONTH'):
    print("Fetching time buckets...")
    url = f"{server_url}/api/timeline/buckets"
    headers = {'Authorization': f'Bearer {token}', 'Accept': 'application/json'}
    params = {'userId': user_id, 'size': size}
    response = requests.get(url, headers=headers, params=params)
    if response.status_code == 200:
        print("Time buckets fetched successfully.")
        buckets = response.json()
        print(f"Time buckets: {buckets}")
        return buckets
    else:
        print(f"Failed to fetch time buckets. Status code: {response.status_code}, Response text: {response.text}")
        return []

def add_assets_to_album(server_url, token, album_id, asset_ids, key=None):
    url = f"{server_url}/api/albums/{album_id}/assets"
    headers = {
        'Authorization': f'Bearer {token}',
        'Content-Type': 'application/json',
        'Accept': 'application/json'
    }
    payload = json.dumps({"ids": asset_ids})
    params = {'key': key} if key else {}
    print(f"Sending request to {url} with payload: {payload} and params: {params}")

    response = requests.put(url, headers=headers, data=payload, params=params)
    print(response.text)
    print(f"Add assets response: {response.status_code}, Body: {response.text}")

    if response.status_code == 200:
        print("Assets successfully added to the album.")
        return True
    else:
        try:
            error_response = response.json()
            print(f"Error adding assets to album: {error_response.get('error', 'Unknown error')}")
        except json.JSONDecodeError:
            print(f"Failed to decode JSON response. Status code: {response.status_code}, Response text: {response.text}")
        return False

def chunker(seq, size):
    return (seq[pos:pos + size] for pos in range(0, len(seq), size))

def main():
    #dont forget to change the entries:
    server_url = "https://immich.yurserver.xy"
    email = "test@test.xy"
    password = "yuorpassword"
    person_id = "1dde757d-00c5-4aab-b610-934a1105f1f0"
    album_id = "c5cb853e-ae21-48c3-aeea-b8efe72da9c0"

    login_response = login_to_api(server_url, email, password)
    if login_response:
        token = login_response['accessToken']
        user_id = login_response['userId']

        if validate_token(server_url, token):
            time_buckets = get_time_buckets(server_url, token, user_id, size='MONTH')

            unique_asset_ids = set()
            for bucket in time_buckets:
                assets = get_person_assets(server_url, token, person_id)
                for asset in assets:
                    unique_asset_ids.add(asset['id'])

            print(f"Total unique assets: {len(unique_asset_ids)}")
            asset_ids_list = list(unique_asset_ids)

            for asset_chunk in chunker(asset_ids_list, 500):
                add_assets_to_album(server_url, token, album_id, asset_chunk, key=None)
                time.sleep(2)
        else:
            print("Invalid token; cannot proceed with fetching assets.")
    else:
        print("Failed to log in; cannot proceed with fetching assets.")

if __name__ == "__main__":
    main()