mirror of
https://github.com/RsyncProject/rsync.git
synced 2026-11-04 02:55:08 -05:00
134 lines
4.7 KiB
Python
134 lines
4.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
|
|
from . import protocol as rp
|
|
|
|
DEFAULT_PORT = 873
|
|
CAPS = 'e.LsfxCIu'
|
|
|
|
def parse_remote(spec):
|
|
if spec.startswith('rsync://'):
|
|
rest = spec[len('rsync://'):]
|
|
hostport, _, path = rest.partition('/')
|
|
host, _, port = hostport.partition(':')
|
|
return host, int(port) if port else DEFAULT_PORT, path
|
|
if '::' in spec:
|
|
host, _, path = spec.partition('::')
|
|
return host, None, path
|
|
return None
|
|
|
|
def server_opts(args, listing):
|
|
flags = ''
|
|
if args.recurse:
|
|
flags += 'r'
|
|
if args.links:
|
|
flags += 'l'
|
|
if args.perms:
|
|
flags += 'p'
|
|
if args.times:
|
|
flags += 't'
|
|
if listing and not args.recurse:
|
|
flags += 'd'
|
|
return '-' + flags + CAPS
|
|
|
|
def do_list(host, port, module, path, args, factory):
|
|
c = factory(host, port)
|
|
c.handshake(module, ['--server', '--sender', server_opts(args, True),
|
|
'.', path], greeting_version=30)
|
|
entries = rp.sort_entries(c.recv_flist(preserve_links=args.links))
|
|
for e in entries:
|
|
tgt = ''
|
|
if e.is_link and e.link_target is not None:
|
|
tgt = ' -> ' + e.link_target.decode('utf-8', 'surrogateescape')
|
|
print('%s %15d %s%s' % (rp.mode_to_perms(e.mode), e.length,
|
|
e.name.decode('utf-8', 'surrogateescape'), tgt))
|
|
c.finish_no_transfer()
|
|
c.drain(timeout=1.0)
|
|
c.close()
|
|
return 0
|
|
|
|
def do_pull(host, port, module, path, dest, args, factory):
|
|
c = factory(host, port)
|
|
c.handshake(module, ['--server', '--sender', server_opts(args, False),
|
|
'.', path], greeting_version=30)
|
|
c.pull(dest, verbose=args.verbose, preserve_times=args.times,
|
|
preserve_perms=args.perms)
|
|
c.drain(timeout=1.0)
|
|
c.close()
|
|
return 0
|
|
|
|
def do_push(srcs, host, port, module, path, args, factory):
|
|
files = []
|
|
for s in srcs:
|
|
if os.path.isfile(s):
|
|
with open(s, 'rb') as fh:
|
|
files.append((os.path.basename(s), fh.read()))
|
|
elif os.path.isdir(s):
|
|
for root, _dirs, names in os.walk(s):
|
|
for nm in names:
|
|
full = os.path.join(root, nm)
|
|
if os.path.isfile(full) and not os.path.islink(full):
|
|
rel = os.path.relpath(full, s)
|
|
with open(full, 'rb') as fh:
|
|
files.append((rel, fh.read()))
|
|
c = factory(host, port)
|
|
c.handshake(module, ['--server', server_opts(args, False), '.', path],
|
|
greeting_version=30)
|
|
c.push(files)
|
|
if args.verbose:
|
|
for name, _ in files:
|
|
print(name)
|
|
c.drain(timeout=1.0)
|
|
c.close()
|
|
return 0
|
|
|
|
def main(argv=None, client_factory=rp.DaemonClient):
|
|
p = argparse.ArgumentParser(prog='protocol-client.py', add_help=True)
|
|
p.add_argument('-a', '--archive', action='store_true', help='= -rlpt')
|
|
p.add_argument('-r', '--recursive', dest='recurse', action='store_true')
|
|
p.add_argument('-l', '--links', action='store_true')
|
|
p.add_argument('-p', '--perms', action='store_true')
|
|
p.add_argument('-t', '--times', action='store_true')
|
|
p.add_argument('-v', '--verbose', action='store_true')
|
|
p.add_argument('--list-only', action='store_true')
|
|
p.add_argument('--port', type=int, default=None)
|
|
p.add_argument('paths', nargs='+')
|
|
args = p.parse_args(argv)
|
|
if args.archive:
|
|
args.recurse = args.links = args.perms = args.times = True
|
|
|
|
paths = args.paths
|
|
remotes = [parse_remote(x) for x in paths]
|
|
|
|
if args.list_only or (len(paths) == 1 and remotes[0] is not None):
|
|
host, port, path = remotes[0]
|
|
port = args.port or port or DEFAULT_PORT
|
|
module = path.split('/')[0]
|
|
return do_list(host, port, module, path, args, client_factory)
|
|
|
|
if len(paths) < 2:
|
|
p.error('need a source and a destination')
|
|
*srcs, dest = paths
|
|
*src_remotes, dest_remote = remotes
|
|
|
|
if dest_remote is not None and all(r is None for r in src_remotes):
|
|
host, port, path = dest_remote
|
|
port = args.port or port or DEFAULT_PORT
|
|
module = path.split('/')[0]
|
|
return do_push(srcs, host, port, module, path, args, client_factory)
|
|
|
|
if len(srcs) == 1 and src_remotes[0] is not None and dest_remote is None:
|
|
host, port, path = src_remotes[0]
|
|
port = args.port or port or DEFAULT_PORT
|
|
module = path.split('/')[0]
|
|
return do_pull(host, port, module, path, dest, args, client_factory)
|
|
|
|
p.error('unsupported source/destination combination (one side must be '
|
|
'host::module/path and the other local)')
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|