File: dl_progressbar.py

package info (click to toggle)
thunderbird 1%3A115.16.0esr-1~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 3,476,252 kB
  • sloc: cpp: 6,972,150; javascript: 5,209,211; ansic: 3,507,222; python: 1,137,609; asm: 432,531; xml: 205,149; java: 175,761; sh: 116,485; makefile: 22,152; perl: 13,971; objc: 12,561; yacc: 4,583; pascal: 2,840; lex: 1,720; ruby: 1,075; exp: 762; sql: 666; awk: 580; php: 436; lisp: 430; sed: 70; csh: 10
file content (41 lines) | stat: -rw-r--r-- 1,074 bytes parent folder | download | duplicates (40) 
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
 
import argparse

# for python 3, use https://github.com/coagulant/progressbar-python3
from progressbar import ProgressBar, Percentage, RotatingMarker, ETA, \
    FileTransferSpeed, Bar

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_file(url, dest=None):
    if dest is None:
        dest = urlparse(url).path.split('/')[-1]

    widgets = ['Download: ', Percentage(), ' ', Bar(marker=RotatingMarker()),
               ' ', ETA(), ' ', FileTransferSpeed()]
    bar = ProgressBar(widgets=widgets).start()

    def download_progress(_, current, total):
        bar.maxval = total
        bar.update(current)

    dl = Download(url, dest, progress=download_progress)
    dl.start()
    dl.wait()
    bar.finish()


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