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 delete_value(self, file_id): mongo_helper = MongodbHelper(self.host, self.port) file_id = mongo_helper.remove(self.db, self.collection, file_id) file_id = mongo_helper.remove(self.default_db, self.default_collection, file_id) return file_id