[tor-commits] [bridgedb/develop] Add script for assigning unallocated bridges to another distributor.

[isis at torproject.org]

commit 3095c905600431cc3c0f52885cc8ad219228fafe
Author: Isis Lovecruft <isis at torproject.org>
Date:   Wed Feb 14 00:25:48 2018 +0000

    Add script for assigning unallocated bridges to another distributor.
    
     * FIXES #25246: https://bugs.torproject.org/25246
---
 scripts/assign-reserve-bridges.py | 81 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)

diff --git a/scripts/assign-reserve-bridges.py b/scripts/assign-reserve-bridges.py
new file mode 100755
index 0000000..0585634
--- /dev/null
+++ b/scripts/assign-reserve-bridges.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python2
+# -*- coding: utf-8 -*-
+#
+# This file is part of BridgeDB, a Tor bridge distribution system.
+#
+# :authors: Isis Lovecruft 0xA3ADB67A2CDB8B35 <isis at torproject.org>
+#           please also see AUTHORS file
+# :copyright: (c) 2013-2018 Isis Lovecruft
+#             (c) 2007-2018, The Tor Project, Inc.
+#             (c) 2007-2018, all entities within the AUTHORS file
+# :license: 3-clause BSD, see included LICENSE for information
+
+"""Assign all unallocated bridges to another distributor."""
+
+from __future__ import print_function
+
+import argparse
+import logging
+
+import bridgedb.Storage
+
+#: The path to the sqlite database file containing bridges.
+DB_FILENAME = "bridgedist.db.sqlite"
+
+logger = logging.getLogger("assign-reserve-bridges")
+logger.setLevel(logging.DEBUG)
+
+def setDBFilename(filename):
+    global DB_FILENAME
+    DB_FILENAME = filename
+
+def getDBFilename():
+    return DB_FILENAME
+
+def getOptions():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "distributor", type=type(''),
+        help="The new distributor to assign the unallocated bridges to")
+    parser.add_argument(
+        "-f", "--db-filename", type=type(''),
+        help="The path to the database file")
+
+    return parser.parse_args()
+
+def checkOptions(options):
+    assert options.distributor
+    assert options.distributor in ["https", "email", "moat"]
+    return options
+
+def getDB():
+    return bridgedb.Storage.Database(getDBFilename())
+
+def assignBridgesToDistributor(db, distributor):
+    unallocated = db.getBridgesForDistributor('unallocated')
+
+    logging.info("Assigning %d unallocated bridges to distributor %s"
+                 % (len(unallocated), distributor))
+    print("Assigning %d unallocated bridges to distributor %s"
+                 % (len(unallocated), distributor))
+
+    for bridge in unallocated:
+        db.updateDistributorForHexKey(distributor, bridge.hex_key)
+        db.commit()
+
+    remaining_bridges = db.getBridgesForDistributor('unallocated')
+    assert len(remaining_bridges) == 0
+
+    logging.info("Done!")
+    print("Done!")
+
+
+if __name__ == "__main__":
+    options = checkOptions(getOptions())
+    
+    if options.db_filename:
+        setDBFilename(options.db_filename)
+
+    db = getDB()
+
+    assignBridgesToDistributor(db, options.distributor)