mirror of
https://github.com/upa/mscp.git
synced 2026-02-04 03:24:58 +08:00
11e024c1da
- fix libmscp install path by setup.py with data_files - fix return values of mscp_get_stats() - add examples directory for mscp python binding
64 lines
1.6 KiB
Python
Executable File
64 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""mscp.py
|
|
|
|
An example python script running mscp
|
|
"""
|
|
|
|
import argparse
|
|
import time
|
|
import sys
|
|
|
|
from rich.progress import Progress
|
|
|
|
import mscp
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-f", "--from", dest = "fr",
|
|
metavar = "REMOTE", default = None,
|
|
help = "copy a file from this remote host")
|
|
parser.add_argument("-t", "--to", metavar = "REMOTE", default = None,
|
|
help = "copy a file to this remote host")
|
|
parser.add_argument("source", help = "path to source file to be copied")
|
|
parser.add_argument("destination", help = "path of copy destination")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.fr and args.to:
|
|
print("-f and -t are exclusive", file = sys.stderr)
|
|
sys.exit(1)
|
|
elif args.fr:
|
|
d = mscp.REMOTE2LOCAL
|
|
remote = args.fr
|
|
elif args.to:
|
|
d = mscp.LOCAL2REMOTE
|
|
remote = args.to
|
|
else:
|
|
print("-f or -t must be specified", file = sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
m = mscp.mscp(remote, d)
|
|
m.connect()
|
|
m.add_src_path(args.source)
|
|
m.set_dst_path(args.destination)
|
|
m.scan()
|
|
m.start()
|
|
|
|
total, done, finished = m.stats()
|
|
with Progress() as progress:
|
|
|
|
task = progress.add_task("[green]Copying...", total = total)
|
|
|
|
while not progress.finished:
|
|
total, done, finished = m.stats()
|
|
progress.update(task, completed = done)
|
|
time.sleep(0.5)
|
|
|
|
m.join()
|
|
m.cleanup()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|