fix: Remote typer dependency

This commit is contained in:
2023-04-02 22:35:52 +02:00
parent cb39b395c6
commit f8febafacb
+21 -23
View File
@@ -1,10 +1,11 @@
#!/usr/bin/env python3
import httpx
import subprocess
import argparse
import logging
import typer
import subprocess
from pathlib import Path
import httpx
logging.basicConfig(
level=logging.DEBUG,
format="%(levelname)s %(asctime)s: %(message)s",
@@ -15,8 +16,6 @@ logging.getLogger("httpx").setLevel(logging.WARNING)
# get api token from https://real-debrid.com/apitoken
key = "P4SUGS4VSXLDIRXSOAB46DXMU2HVFJTOWYCACNULX4V7TOZRW2BA"
cli = typer.Typer()
http = httpx.Client(
base_url="https://api.real-debrid.com/rest/1.0/",
headers={"Authorization": f"Bearer {key}"},
@@ -29,9 +28,7 @@ def get_download_url(url: str) -> str:
return res.json()["download"]
def download_with_aria2(
url: str, cwd: Path = Path.cwd(), extra_args: list[str] = None
) -> None:
def download_with_aria2(url: str, cwd: Path = Path.cwd(), extra_args: list[str] = None) -> None:
if not extra_args:
extra_args = []
args = ["aria2c", "-x", "5", url, *extra_args]
@@ -39,33 +36,34 @@ def download_with_aria2(
subprocess.run(args, text=True, cwd=str(cwd.resolve()))
@cli.command(
context_settings={
"ignore_unknown_options": True,
"allow_extra_args": True,
}
def parse_args():
arger = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
arger.add_argument('url', help='URL to unrestrict & download')
arger.add_argument(
'--output-dir', '-o', '--out', dest='cwd', help='Output directory', type=Path, default=Path.cwd()
)
def main(
ctx: typer.Context,
url: str,
cwd: Path = typer.Option(Path("."), "-o", "--out", "--cwd", writable=True),
):
args, extra = arger.parse_known_args()
return args, extra
def main():
args, aria2c_args = parse_args()
try:
logging.debug("generating premium link")
download_url = get_download_url(url)
download_url = get_download_url(args.url)
except (httpx.HTTPStatusError, KeyError):
logging.error("cannot generate premium link")
raise typer.Exit(1)
raise SystemExit(1)
logging.info("got url=%s", download_url)
logging.debug("passing url to aria2")
try:
download_with_aria2(download_url, cwd=cwd, extra_args=ctx.args)
download_with_aria2(download_url, cwd=args.cwd, extra_args=aria2c_args)
except KeyboardInterrupt:
logging.error("download interrupted. exiting")
raise typer.Exit(2)
raise SystemExit(1)
if __name__ == "__main__":
cli()
main()