-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathsynch.py
55 lines (47 loc) · 1.95 KB
/
synch.py
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
import sys
import argparse
import time
import logging
import subprocess
from watchdog.observers import Observer
from watchdog.events import LoggingEventHandler, FileSystemEventHandler
class RsyncEventHandler(FileSystemEventHandler):
def __init__(self, host, src_dir, dst_dir, flags, exclude):
self.host = host
self.src = src_dir
self.dst = dst_dir
self.flags = flags
self.exclude = exclude
self.sync()
def on_any_event(self, event):
if not event.is_directory:
self.sync()
def sync(self):
cmd = ['rsync', self.flags, self.src, '--exclude-from', self.exclude, '{}:{}'.format(self.host, self.dst)]
print(' '.join(cmd))
subprocess.call(cmd)
def main(args):
logging.basicConfig(level=logging.INFO,
format='%(asctime)s - %(message)s',
datefmt='%Y-%m-%d %H:%M:%S')
event_handler = LoggingEventHandler()
rsync_handler = RsyncEventHandler(args.host, args.source, args.destination, args.rsync, args.exclude_file)
observer = Observer()
observer.schedule(event_handler, args.source, recursive=True)
observer.schedule(rsync_handler, args.source, recursive=True)
observer.start()
try:
while True:
time.sleep(1)
except KeyboardInterrupt:
observer.stop()
observer.join()
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('host', type=str, help='Host')
parser.add_argument('destination', type=str, help='Destination directory')
parser.add_argument('-s', '--source', type=str, default='.', help='Source directory (default current directory)')
parser.add_argument('-rs', '--rsync', type=str, default='-azvq', help='rsync flags (default -avzq')
parser.add_argument('-ef', '--exclude-file', type=str, default='.rsync-exclude', help='File containing exclusion rules (default .gitignore)')
args = parser.parse_args()
main(args)