select bitcoin network

Author: RCasatta

otsd-backup.py

@@ -47,6 +47,12 @@
                     default='localhost',
                     help="RPC address (default: %(default)s)")
 
+parser.add_argument('--btc-testnet', dest='btc_net', action='store_const',
+                    const='testnet', default='mainnet',
+                    help='Use Bitcoin testnet rather than mainnet')
+parser.add_argument('--btc-regtest', dest='btc_net', action='store_const',
+                    const='regtest',
+                    help='Use Bitcoin regtest rather than mainnet')
 
 args = parser.parse_args()
 args.parser = parser
@@ -77,14 +83,13 @@
 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 = otsserver.backup.AskBackup(db, calendar_url, base_path, args.btc_net)
     ask_thread.start()
 
 try:

otsserver/backup.py

@@ -40,6 +40,7 @@ def __getitem__(self, chunk):
         if cached_kv_bytes is not None:
             return cached_kv_bytes
 
+        # TODO next code must be run only once, LOCK or something
         backup_map = {}
         start = chunk*PAGING
         end = start+PAGING
@@ -216,24 +217,31 @@ def serve_forever(self):
 # theoretically serve timestamp in place of every calendar server which supports this incremental live backup mechanism
 class AskBackup(threading.Thread):
 
-    def __init__(self, db, calendar_url, base_path):
+    def __init__(self, db, calendar_url, base_path, btc_net):
         self.db = db
         self.calendar_url = calendar_url
         calendar_url_parsed = urlparse(calendar_url)
         self.up_to_path = os.path.join(base_path, calendar_url_parsed.netloc)
+        self.btc_net = btc_net
 
         super().__init__(target=self.loop)
 
     def loop(self):
         logging.info("Starting loop for %s" % self.calendar_url)
 
         try:
+            logging.debug("Opening %s" % self.up_to_path)
             with open(self.up_to_path, 'r') as up_to_fd:
                 last_known = int(up_to_fd.read().strip())
         except FileNotFoundError as exp:
             last_known = -1
         logging.info("Checking calendar " + str(self.calendar_url) + ", last_known commitment:" + str(last_known))
 
+        if self.btc_net == 'testnet':
+            bitcoin.SelectParams('testnet')
+        elif self.btc_net == 'regtest':
+            bitcoin.SelectParams('regtest')
+
         while True:
             start_time = time.time()
             backup_url = urljoin(self.calendar_url, "/experimental/backup/%d" % (last_known + 1))