foreflight-logbook-export / fflogex / cli.py

import sys
import time
from typing import cast

import fflogex
from getpass import getpass
import argparse
from dataclasses import dataclass


@dataclass
class Args:
    username: str
    password: str | None
    output: str


def main():
    parser = argparse.ArgumentParser(
        prog="fflogex",
        description="Export Foreflight Logbook",
    )
    _ = parser.add_argument("-u", "--username", required=True)
    _ = parser.add_argument(
        "-p",
        "--password",
        help="Specify password inline; You can also skip this and the program will ask you during the execution instead. BE CAREFUL: Your shell might record your password if you choose to use this option.",
    )
    _ = parser.add_argument(
        "-o", "--output", help="Output file name (.csv)", default="logbook.csv"
    )

    opts = cast(Args, cast(object, parser.parse_args()))
    if opts.password is None:
        opts.password = getpass()
    exporter = fflogex.ForeflightLogbookExporter()

    print("Logging in...")
    exporter.login(opts.username, opts.password)

    print("Requesting logbook export...", end="", flush=True)
    requestId = exporter.request_export()
    print(f" [Done] Request ID: {requestId}")

    print("Waiting ...", end="", flush=True)
    elapsed = 0
    while elapsed <= 60:
        elapsed += 1
        time.sleep(1)
        print(".", end="", flush=True)
        link = exporter.get_link(requestId)
        if link is not None:
            print(" [OK]")
            print("Downloading...", end="", flush=True)
            with open(opts.output, "w", encoding="utf-8") as f:
                _ = f.write(exporter.download_link(link))
            print(f" [OK] Wrote to {opts.output}")
            sys.exit(0)

    print("")
    print("[FAILED] Timeout while waiting for the export")
    sys.exit(1)


if __name__ == "__main__":
    main()