def partner_to_json(item):
partner_id = item.pop('partner_id', None)
if partner_id:
partner = get_db()['partners'].find_one({'_id': partner_id})
item['partner'] = item_to_json(partner, ['_id', '_cls', 'name'])
else:
item['partner'] = None
return item
def update(self, id, data):
return get_db(cfg.database__database_name).get_collection('users'). \
update_one({'_id': ObjectId(id)}, {"$set": dictionary_user(data)},
upsert=False)
def get_by_field(self, field, value):
return get_db(cfg.database__database_name).get_collection('users') \
.find_one({field: value})
def get(self, id):
return get_db(cfg.database__database_name).get_collection('users') \
.find_one({'_id': ObjectId(id)})
def get(self):
cursor = get_db(
cfg.database__database_name).get_collection('token').find()
return cursor.__getitem__(0)['token']
def __init__(self, name_image):
self.grid_fs = GridFS(get_db(_cfg.database__database_repository))
self.name_image = name_image