- First commit

backup.py

@@ -0,0 +1,98 @@
+#!/usr/bin/python3
+
+from datetime import date, timedelta, datetime
+from glob import iglob
+from re import match
+import os
+from shutil import rmtree
+from shlex import split
+from subprocess import call
+from argparse import ArgumentParser
+
+
+def dates(sdate):
+    # sdate: "20180508"
+    sdate = date(int(sdate[:4]), int(sdate[4:6]), int(sdate[6:]))
+    edate = datetime.now()
+    edate = date(edate.year, edate.month, edate.day)
+    return [(sdate + timedelta(i)) for i in range(int((edate - sdate).days))]
+
+
+def days(d, n=7):
+    return d[-n:]
+
+
+def weeks(d, n=4):
+    l = {}
+    for i in d[-7*n:]:
+        l[(i.year, i.isocalendar()[1])] = i
+    return list(l.values())
+
+
+def months(d, n=12):
+    l = {}
+    for i in d[-int(30.4375*n):]:
+        l[(i.year, i.month)] = i
+    return list(l.values())
+
+
+def years(d, n=10):
+    l = {}
+    for i in d[-int(365.25*n):]:
+        l[i.year] = i
+    return list(l.values())
+
+
+if __name__ == '__main__':
+    parser = ArgumentParser(description='Backup files.')
+    parser.add_argument('source', help='source path')
+    parser.add_argument('dest', help='destination path')
+    parser.add_argument('-r', '--rsync', help='arguments for rsync', default='-a')
+    parser.add_argument('-v', '--verbose', help='verbose', action='store_true')
+    parser.add_argument('-d', '--dry-run', help='dry-run', action='store_true')
+    parser.add_argument('-l', '--latest', help='name of latest backup', default='latest')
+
+    args = parser.parse_args()
+    verbose = args.verbose
+    run = not args.dry_run
+
+    if not run:
+        print('Dry-run: do not actually change anything.')
+
+    backup_path_full = os.path.join(args.dest, datetime.now().strftime('%Y%m%d-%H%M%S'))
+    if verbose:
+        print('Backing up to: {}'.format(backup_path_full))
+    if run:
+        os.makedirs(backup_path_full, exist_ok=True)
+    latest_path = os.path.join(args.dest, args.latest)
+    if os.path.exists(latest_path):
+        latest = '--link-dest="{}" '.format(os.path.join(latest_path, args.source[1:], ''))
+    else:
+        latest = ''
+
+    os.makedirs(os.path.join(backup_path_full, args.source[1:]))
+    rsync = 'rsync {} --delete "{}" {} "{}"'.format(args.rsync, os.path.join(args.source, ''), latest,
+        os.path.join(backup_path_full, args.source[1:]))
+
+    if verbose:
+        print('Running rsync: {}'.format(rsync))
+    if run:
+        call(split(rsync))
+        if os.path.exists(latest_path):
+            os.remove(latest_path)
+        os.symlink(backup_path_full, latest_path)
+
+    # Deleting old backups
+    f = [i for i in iglob(os.path.join(args.dest, '*')) if not match('\d{8}-.*', os.path.split(i)[1]) is None]
+    fn = [os.path.split(i)[1] for i in f]
+    f = {date(int(j[:4]), int(j[4:6]), int(j[6:8])): i for i, j in zip(f, fn)}
+    keys = sorted(list(f.keys()))
+    keep = list(set(days(keys) + weeks(keys) + months(keys) + years(keys)))
+    keep = {f[i] for i in keep}
+    delete = set(f.values()) - keep
+    if verbose:
+        print('Deleting old backups: {}'.format(sorted(list(delete))))
+    if run:
+        for i in delete:
+            if os.path.exists(i):
+                rmtree(i, True)