class CITaskConfigManager(object):
def __init__(self, mongo_host, mongo_port, db, collection):
self.host = mongo_host
self.port = mongo_port
self.db = db
self.collection = collection
self.default_db = "doraemon"
self.default_collection = "ci"
self.mongo_helper = MongodbHelper(self.host, self.port)
def save(self, value):
return self.mongo_helper.save(self.db, self.collection, value)
def remove(self, doc_id):
result = self.mongo_helper.remove(self.db, self.collection, doc_id)
if result == None:
result = self.mongo_helper.remove(self.default_db,
self.default_collection, doc_id)
return result
def get(self, doc_id):
result = self.mongo_helper.get(self.db, self.collection, doc_id)
if result == None:
result = self.mongo_helper.get(self.default_db,
self.default_collection, doc_id)
return result
def save(db, collection, value):
HOST = MONGODB['default']['HOST']
PORT = MONGODB['default']['PORT']
mongo_helper = MongodbHelper(HOST, PORT)
return mongo_helper.save(db, collection, value)
def save_replace_config(self, value):
mongo_helper = MongodbHelper(CIDeployServiceManager.host,
CIDeployServiceManager.port)
return mongo_helper.save(CIDeployServiceManager.db,
CIDeployServiceManager.collection, value)
def save_value(self, value):
mongo_helper = MongodbHelper(self.host, self.port)
doc_id = mongo_helper.save(self.db, self.collection, value)
return doc_id