python-urlopen 0.0.1.2__tar.gz → 0.0.1.3__tar.gz

{python_urlopen-0.0.1.2 → python_urlopen-0.0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-urlopen
-Version: 0.0.1.2
+Version: 0.0.1.3
 Summary: Python urlopen wrapper.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-urlopen
 License: MIT

{python_urlopen-0.0.1.2 → python_urlopen-0.0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-urlopen"
-version = "0.0.1.2"
+version = "0.0.1.3"
 description = "Python urlopen wrapper."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"
@@ -30,6 +30,10 @@ python = "^3.10"
 http_response = "*"
 python-filewrap = "*"
 
+[tool.poetry.scripts]
+python-urlopen = "urlopen.__main__:main"
+urlopen = "urlopen.__main__:main"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

{python_urlopen-0.0.1.2 → python_urlopen-0.0.1.3}/urlopen/__main__.py

@@ -7,8 +7,8 @@ __doc__ = "python url downloader"
 from argparse import ArgumentParser, RawTextHelpFormatter
 
 parser = ArgumentParser(description=__doc__, formatter_class=RawTextHelpFormatter)
-parser.add_argument("urls", nargs="*", metavar="url", help="URL(s) to be downloaded (one URL per line), if omitted, read from stdin")
-parser.add_argument("-d", "--savedir", default="", help="directory to the downloading files")
+parser.add_argument("url", nargs="?", help="URL to be downloaded")
+parser.add_argument("-o", "--output-file", help="file path to be downloaded, if omitted, print into stdout")
 parser.add_argument("-r", "--resume", action="store_true", help="skip downloaded data")
 parser.add_argument("-hs", "--headers", help="dictionary of HTTP Headers to send with")
 parser.add_argument("-v", "--version", action="store_true", help="print the current version")
@@ -17,9 +17,11 @@ if args.version:
     from urlopen import __version__
     print(".".join(map(str, __version__)))
     raise SystemExit(0)
+url = args.url
+if not url:
+    parser.parse_args(["-h"])
 
 from collections import deque
-from os import makedirs
 from time import perf_counter
 
 from urlopen import download
@@ -45,35 +47,35 @@ def progress(total=None):
             print(f"\r\x1b[K{read_num} | {speed:.2f} MB/s", end="", flush=True)
         dq.append((read_num, cur_t))
 
-urls = args.urls
-if not urls:
-    from sys import stdin
-    urls = (l.removesuffix("\n") for l in stdin)
-savedir = args.savedir
-if savedir:
-    makedirs(savedir, exist_ok=True)
-
-try:
+def main():
     headers = args.headers
     if headers is not None:
         headers = headers_str_to_dict(headers)
-    for url in urls:
-        if not url:
-            continue
-        try:
-            file = download(
-                url, 
-                savedir, 
-                resume=args.resume, 
-                make_reporthook=progress, 
-                headers=headers, 
-            )
-            print(f"\r\x1b[K\x1b[1;32mDOWNLOADED\x1b[0m \x1b[4;34m{url!r}\x1b[0m\n |_ ⏬ \x1b[4;34m{file!r}\x1b[0m")
-        except BaseException as e:
-            print(f"\r\x1b[K\x1b[1;31mERROR\x1b[0m \x1b[4;34m{url!r}\x1b[0m\n  |_ 🙅 \x1b[1;31m{type(e).__qualname__}\x1b[0m: {e}")
-except (EOFError, KeyboardInterrupt):
-    pass
-except BrokenPipeError:
-    from sys import stderr
-    stderr.close()
+
+    output_file = args.output_file
+    if output_file:
+        from os.path import dirname
+        dir_ = dirname(output_file)
+        if dir_:
+            from os import makedirs
+            makedirs(dir_, exist_ok=True)
+
+        download(
+            url, 
+            output_file, 
+            resume=args.resume, 
+            make_reporthook=progress, 
+            headers=headers, 
+        )
+    else: 
+        from sys import stdout
+
+        download(
+            url, 
+            stdout, 
+            headers=headers, 
+        )
+
+if __name__ == "__main__":
+    main()