docsTitle

"""
TRUST Lab API - Complete Workflow Example
This script demonstrates how to search, get details, and download a movie
"""

import requests
from urllib.parse import quote

BASE_URL = "https://your-domain.com"

def search_movie(query):
    """Search for movies by title"""
    response = requests.get(f"{BASE_URL}/api/search", params={"query": query})
    return response.json()

def get_details(subject_id):
    """Get movie details"""
    response = requests.get(f"{BASE_URL}/api/item-details", params={"subjectId": subject_id})
    return response.json()

def get_media_links(subject_id, detail_path, season=0, episode=0):
    """Get streaming and download links"""
    response = requests.get(f"{BASE_URL}/api/media", params={
        "subjectId": subject_id,
        "detailPath": detail_path,
        "season": season,
        "episode": episode
    })
    return response.json()

def download_movie(url, filename):
    """Download the movie file"""
    print(f"Downloading: {filename}")
    response = requests.get(url, stream=True)
    with open(filename, "wb") as f:
        for chunk in response.iter_content(chunk_size=8192):
            f.write(chunk)
    print(f"Download complete: {filename}")

# Example usage
if __name__ == "__main__":
    # 1. Search for a movie
    search_results = search_movie("Avengers Endgame")
    
    if search_results.get("data", {}).get("subjects"):
        movie = search_results["data"]["subjects"][0]
        subject_id = movie["subjectId"]
        detail_path = movie["detailPath"]
        title = movie["title"]
        
        print(f"Found: {title}")
        
        # 2. Get media links
        media = get_media_links(subject_id, detail_path)
        
        if media.get("data", {}).get("downloads", {}).get("data", {}).get("downloads"):
            download = media["data"]["downloads"]["data"]["downloads"][0]
            
            # 3. Download using the provided URL (already includes movie name)
            download_url = download["downloadUrl"]
            print(f"Download URL: {download_url}")
            
            # The filename will be: trustlab_avengers_endgame_1080p.mp4