def save_content(self, content, file_name, file_property): mongo_helper = MongodbHelper(self.host, self.port) file_bucket_fs = mongo_helper.pub_file_bucket(self.db, self.collection, file_name, file_property) file_bucket_fs.write(content.encode(encoding="utf-8")) file_bucket_fs.close() return file_bucket_fs._id
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 get_replace_config(self, doc_id): mongo_helper = MongodbHelper(CIDeployServiceManager.host, CIDeployServiceManager.port) result = mongo_helper.get(CIDeployServiceManager.db, CIDeployServiceManager.collection, doc_id) if result == None: result = mongo_helper.get(CIDeployServiceManager.default_db, CIDeployServiceManager.default_collection, doc_id) return result
def save_bucket(self,file_chunks,file_name,file_property): mongo_helper=MongodbHelper(self.host,self.port) file_bucket_fs=mongo_helper.pub_file_bucket(self.db,self.collection,file_name,file_property) for chunk in file_chunks: file_bucket_fs.write(chunk) file_bucket_fs.close() return file_bucket_fs._id
def get(db, collection, doc_id): HOST = MONGODB['default']['HOST'] PORT = MONGODB['default']['PORT'] mongo_helper = MongodbHelper(HOST, PORT) result = mongo_helper.get(db, collection, doc_id) if result == None: result = mongo_helper.get('doraemon', 'ci', doc_id) return result
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 get(self, file_id): mongo_helper = MongodbHelper(self.host, self.port) grid_out = mongo_helper.get_file_bucket(self.db, self.collection, file_id) if grid_out == None: grid_out = mongo_helper.get_file_bucket(self.default_db, self.default_collection, file_id) return grid_out
def copy_bucket(self, grid_out): mongo_helper = MongodbHelper(self.host, self.port) file_bucket_fs = mongo_helper.pub_file_bucket(self.db, self.collection, grid_out.name, grid_out.metadata) while True: chunk = grid_out.read(size=1024 * 1024 * 10) if chunk: file_bucket_fs.write(chunk) else: break file_bucket_fs.close() return file_bucket_fs._id
def delete_file(self,file_id): try: mongo_helper=MongodbHelper(self.host,self.port) mongo_helper.delete_file(self.db,self.collection, file_id) mongo_helper.delete_file(self.default_db,self.default_collection, file_id) except Exception as ex: SimpleLogger.exception(ex)
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
def get_value(self, doc_id): mongo_helper = MongodbHelper(self.host, self.port) grid_out = mongo_helper.get(self.db, self.collection, doc_id) return grid_out
def save_replace_config(self, value): mongo_helper = MongodbHelper(CIDeployServiceManager.host, CIDeployServiceManager.port) return mongo_helper.save(CIDeployServiceManager.db, CIDeployServiceManager.collection, value)
def get(db, collection, doc_id): HOST = MONGODB['default']['HOST'] PORT = MONGODB['default']['PORT'] mongo_helper = MongodbHelper(HOST, PORT) result = mongo_helper.get(db, 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 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
def save(self, file_bytes, file_property): mongo_helper = MongodbHelper(self.host, self.port) file_id = mongo_helper.put_file(self.db, self.collection, file_bytes, file_property) return file_id
def update_value(self, doc_id, value): mongo_helper = MongodbHelper(self.host, self.port) doc_id = mongo_helper.update(self.db, self.collection, doc_id, value) return doc_id