def __init__(self): super(MyDBDialog, self).__init__() self.setupUi(self) self.setWindowIcon(QIcon('./images/db.png')) self.buttonBox.clicked.connect(self.set_db_info) self.db_dict = utils.get_db_config() self.init_db_info()
def init_db_connect(): db_dict = utils.get_db_config() db_password = utils.HashManager().back_aes_ecb(db_dict['db_password']) db_connect = DBSession(account=db_dict['db_account'], password=db_password, ip=db_dict['db_ip'], port=db_dict['db_port'], dbname=db_dict['db_dbname']) return db_connect
def main(): import argparse parser = argparse.ArgumentParser() parser.add_argument('-d', '--drop', action="store_true", default=False, help="drop existing collections before importing") parser.add_argument("-s", "--source", default=None, type=str, help="db to import from") parser.add_argument("-t", "--target", default=None, type=str, help="db to import to") args = parser.parse_args() host, port = utils.get_db_config(args.source) utils.log("SOURCE: %s:%s" % (host, port)) old_host = host old_connection = pymongo.Connection(host, port) old_database = old_connection['stamped'] collections = old_database.collection_names() new_host = args.target if new_host is None: dest = MongoDBConfig.getInstance() new_host = dest.host utils.log("DEST: %s:%s" % (new_host, port)) if not os.path.isdir('/stamped/tmp/stamped/'): os.makedirs('/stamped/tmp/stamped') ignore = set([ 'tempentities', 'logs', 'logstats', ]) for collection in collections: print 'RUN %s' % collection if collection in ignore: print 'PASS' else: ret = mongoExportImport(collection, old_host, new_host) if 0 == ret: print 'COMPLETE' else: print "ERROR restoring collection '%s'" % collection print try: utils.runMongoCommand('db.runCommand( {createCollection:"logs", capped:true, size:500000} )') except: utils.printException()
def issueQueries(queries): # TODO: Pull this out to a named constant somewhere, or be more intelligent about this, or something. config = { 'mongodb' : { 'hosts': [get_db_config('peach.db3')] } } devDbConfig = MongoDBConfig() devDbConfig.config = AttributeDict(config) devDb = devDbConfig.connection.stamped # TODO: This whole "stamped_fixtures" thing really needs to be a constant. localDb = MongoDBConfig.getInstance().connection.stamped_fixtures for (collectionName, query) in queries: devCollection = getattr(devDb, collectionName) results = list(devCollection.find(query)) localCollection = getattr(localDb, collectionName) localCollection.insert(results)