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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
import os
import re
import subprocess
import humanfriendly
from ..structures import Progress
from .runnerbase import RunnerBase, neutral_locale_variables
class Download(RunnerBase):
_rsync_progress_re = re.compile(r'^\s*(\S+)\s+(\d+)%\s+(\S+)\s+(\S+)\s+$')
def __init__(self, src, dst):
self.proc = subprocess.Popen(
[
'rsync',
'--partial',
# --no-h: not human, easier to parse (but speed still appears
# in human form).
'--no-h',
'--info=progress2',
src,
dst,
],
stdout=subprocess.PIPE,
encoding='utf8',
env={**os.environ,
**neutral_locale_variables,
},
)
self.last_progress = Progress()
def get_read_fd(self):
return self.proc.stdout
def progress_read(self):
line = self.proc.stdout.readline()
if match := self._rsync_progress_re.search(line):
self.last_progress = Progress(
nbbytes=int(match.group(1)),
percent=int(match.group(2)),
speed=humanfriendly.parse_size(match.group(3), binary=True),
eta=match.group(4),
)
return self.last_progress
def terminate(self):
self.proc.terminate()
def poll(self):
return self.proc.poll()
def close(self):
self.proc.wait()
if __name__ == '__main__':
import sys
import contextlib
with contextlib.ExitStack() as stack:
stack.enter_context(contextlib.suppress(KeyboardInterrupt))
runner = stack.enter_context(Download(sys.argv[1], sys.argv[2]))
while (rc := runner.poll()) is None:
print(runner.progress_read())
print('runner ended with code:', rc)
print('test main ended')
|
