commit 3095c905600431cc3c0f52885cc8ad219228fafe Author: Isis Lovecruft <isis@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@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)