Migrate from Mux to Bunny Stream

"""Turns on temporary master access for every ready Mux asset, then waits until each master file is ready."""

import base64
import getpass
import json
import time
import urllib.error
import urllib.request

API = "https://api.mux.com/video/v1"
TIMEOUT_HOURS = 6

token_id = input("Mux Token ID: ").strip()
token_secret = getpass.getpass("Mux Secret Key: ").strip()
auth = base64.b64encode(f"{token_id}:{token_secret}".encode()).decode()

def mux(method, path, body=None):
    request = urllib.request.Request(
        API + path,
        method=method,
        data=json.dumps(body).encode() if body else None,
        headers={"Authorization": f"Basic {auth}", "Content-Type": "application/json"},
    )
    try:
        with urllib.request.urlopen(request) as response:
            return json.load(response)["data"]
    except urllib.error.HTTPError as error:
        raise SystemExit(f"Mux returned {error.code} for {method} {path}. Check the token and its permissions.")

def request_master(asset_id):
    mux("PUT", f"/assets/{asset_id}/master-access", {"master_access": "temporary"})
    time.sleep(0.2)  # stay under Mux's API rate limit

# Collect every ready asset, 100 per page.
assets, page = [], 1
while True:
    items = mux("GET", f"/assets?limit=100&page={page}")
    assets += [asset["id"] for asset in items if asset["status"] == "ready"]
    if len(items) < 100:
        break
    page += 1

print(f"Requesting master access for {len(assets)} assets...")
for asset_id in assets:
    request_master(asset_id)

# Poll only the assets that are still preparing, so each pass gets shorter.
print("Waiting for Mux to prepare the master files...")
pending, ready, errored = set(assets), [], []
deadline = time.time() + TIMEOUT_HOURS * 3600
while True:
    for asset_id in list(pending):
        asset = mux("GET", f"/assets/{asset_id}")
        status = (asset.get("master") or {}).get("status")
        if status == "ready":
            pending.discard(asset_id)
            ready.append(asset_id)
        elif status == "errored":
            pending.discard(asset_id)
            errored.append(asset_id)
        elif asset.get("master_access") != "temporary":
            request_master(asset_id)  # the first request did not take, so ask again
        time.sleep(0.2)
    if not pending or time.time() > deadline:
        break
    print(f"{len(pending)} of {len(assets)} master files still preparing...")
    time.sleep(30)

if errored:
    print(f"Mux could not prepare a master file for {len(errored)} assets: {', '.join(errored)}")
if pending:
    print(f"Stopped waiting after {TIMEOUT_HOURS} hours with {len(pending)} master files still preparing. Run the script again to check on them.")
print(f"{len(ready)} master files are ready. Run the import within 24 hours, before the master URLs expire.")