.gitignore

/*.json

bitly_export.py

#! /usr/bin/env python3
"""Download a backup of my bitly links."""

import datetime as dt
import json
import subprocess
import sys
from pathlib import Path
from urllib.request import Request, urlopen


def get_keychain_credential(account, service):
    """Get credentials from MacOS keychain."""
    cmd = f"security find-generic-password -a {account} -s {service} -w"
    process = subprocess.run(cmd.split(), stdout=subprocess.PIPE)
    data = process.stdout
    return data.decode("utf8").strip()


def request(url: str, access_token: str) -> dict:
    req = Request(url)
    req.add_header("Authorization", "Bearer {}".format(access_token))

    with urlopen(req) as r:
        resp = json.loads(r.read())

    return resp


def main(access_token: str):
    """Run the main code logic."""
    output = Path(".") / f"{dt.datetime.today():%Y%m%d}_bitly_export.json"

    url = "https://api-ssl.bitly.com/v4/groups/Ba9b2FIuUtv/bitlinks?archived=both&size=50"
    all_data = []
    while True:
        resp = request(url, access_token)
        all_data.append(resp)
        url = resp.get("pagination", {}).get("next")
        if not url:
            break

    output.write_text(json.dumps(all_data, indent=4))


def cli() -> dict:
    config = dict(access_token=sys.argv[1])
    return config


if __name__ == "__main__":
    main(**cli())