File: dl_tqdm.py

package info (click to toggle)
mozjs78 78.15.0-7
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 739,892 kB
  • sloc: javascript: 1,344,214; cpp: 1,215,708; python: 526,544; ansic: 433,835; xml: 118,736; sh: 26,176; asm: 16,664; makefile: 11,537; yacc: 4,486; perl: 2,564; ada: 1,681; lex: 1,414; pascal: 1,139; cs: 879; exp: 499; java: 164; ruby: 68; sql: 45; csh: 35; sed: 18; lisp: 2
file content (45 lines) | stat: -rw-r--r-- 1,028 bytes parent folder | download | duplicates (35) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
import argparse
import tqdm

from six.moves.urllib.parse import urlparse

from dlmanager import Download


def parse_args(argv=None):
    parser = argparse.ArgumentParser()
    parser.add_argument("url", help="url to download")
    return parser.parse_args(argv)


def download_progress(bar):
    last_b = [0]

    def inner(_, current, total):
        if total is not None:
            bar.total = total
        delta = current - last_b[0]
        last_b[0] = current

        if delta > 0:
            bar.update(delta)
    return inner


def download_file(url, dest=None):
    if dest is None:
        dest = urlparse(url).path.split('/')[-1]

    with tqdm.tqdm(unit='B', unit_scale=True, miniters=1, dynamic_ncols=True,
                   desc=dest) as bar:
        dl = Download(url, dest, progress=download_progress(bar))
        dl.start()
        dl.wait()


if __name__ == '__main__':
    options = parse_args()
    try:
        download_file(options.url)
    except KeyboardInterrupt:
        print("\nInterrupted.")