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.")
|
