Merge branch 'online_backup' of https://github.com/RCasatta/opentimestamps-server into online_backup

Author: RCasatta

backup.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+# Copyright (C) 2016 The OpenTimestamps developers
+#
+# This file is part of the OpenTimestamps Server.
+#
+# It is subject to the license terms in the LICENSE file found in the top-level
+# directory of this distribution.
+#
+# No part of the OpenTimestamps Server, including this file, may be copied,
+# modified, propagated, or distributed except according to the terms contained
+# in the LICENSE file.
+
+import argparse
+import logging.handlers
+import os
+import sys
+import otsserver.calendar
+import otsserver.backup
+
+parser = argparse.ArgumentParser(description="OpenTimestamps Backup Server")
+
+parser.add_argument("-q", "--quiet", action="count", default=0,
+                    help="Be more quiet.")
+parser.add_argument("-v", "--verbose", action="count", default=0,
+                    help="Be more verbose. Both -v and -q may be used multiple times.")
+parser.add_argument("-p", "--path", type=str,
+                    dest='base_path',
+                    default='~/.otsd/backups',
+                    help="Location of the calendar (default: '%(default)s')")
+
+parser.add_argument('-c', '--calendar', metavar='calendar', action='append', type=str,
+                    default=[], help='Add a calendar URL to the ones to backup.')
+
+parser.add_argument("--debug-file", type=str,
+                    dest='debug_file',
+                    default='~/.otsd/backups/debug.log',
+                    help="Location of the debug log")
+parser.add_argument("--debug-file-max-size", type=int,
+                    dest='debug_file_max_size',
+                    default=10000000,
+                    help="Max size of the debug log (default: %(default)d bytes) ")
+
+parser.add_argument("--rpc-port", type=int,
+                    default=14799,
+                    help="RPC port (default: %(default)d)")
+parser.add_argument("--rpc-address", type=str,
+                    default='localhost',
+                    help="RPC address (default: %(default)s)")
+
+
+args = parser.parse_args()
+args.parser = parser
+
+base_path = os.path.expanduser(args.base_path)
+os.makedirs(base_path, exist_ok=True)
+db_dir = base_path + '/db'
+os.makedirs(db_dir, exist_ok=True)
+print("db dir is %s" % db_dir)
+
+debugfile = os.path.expanduser(args.debug_file)
+handler = logging.handlers.RotatingFileHandler(filename=debugfile, maxBytes=args.debug_file_max_size)
+fmt = logging.Formatter("%(asctime)-15s %(message)s")
+handler.setFormatter(fmt)
+logger = logging.getLogger('')
+logger.addHandler(handler)
+ch = logging.StreamHandler(sys.stdout)
+logger.addHandler(ch)
+
+args.verbosity = args.verbose - args.quiet
+
+if args.verbosity == 0:
+    logging.root.setLevel(logging.INFO)
+elif args.verbosity > 0:
+    logging.root.setLevel(logging.DEBUG)
+elif args.verbosity == -1:
+    logging.root.setLevel(logging.WARNING)
+elif args.verbosity < -1:
+    logging.root.setLevel(logging.ERROR)
+
+
+db = otsserver.calendar.LevelDbCalendar(db_dir)
+calendar = otsserver.backup.BackupCalendar(db)
+server = otsserver.backup.BackupServer((args.rpc_address, args.rpc_port), calendar)
+
+for calendar_url in args.calendar:
+    print("Starting calendar checker for %s" % calendar_url)
+    ask_thread = otsserver.backup.AskBackup(db, calendar_url, base_path)
+    ask_thread.start()
+
+try:
+    print("Starting at %s:%s" % (args.rpc_address, args.rpc_port))
+    server.serve_forever()
+except KeyboardInterrupt:
+    sys.exit(0)
+
+# vim:syntax=python filetype=python