def add_model_img(store, model, obj_id, data):
obj = store.find(model, model.id == obj_id).one()
if obj is not None:
if obj.picture is None:
obj.picture = models.File()
obj.picture.data = base64.b64encode(data)
def add_model_img(store, model, obj_id, data):
data = base64.b64encode(data)
obj = store.find(model, model.id == obj_id).one()
if obj.img_id is not None:
picture = store.find(models.File, models.File.id == obj.img_id).one()
else:
picture = models.File({'data': data})
store.add(picture)
obj.img_id = picture.id
picture.data = data
def db_add_file(store, data, key=None):
file_obj = None
if key is not None:
file_obj = store.find(models.File, id=key).one()
if file_obj is None:
file_obj = models.File()
if key is not None:
file_obj.id = key
store.add(file_obj)
file_obj.data = base64.b64encode(data)
def db_add_file(session, tid, file_id, name, path):
"""
Transaction to register a file on a tenant
:param session: An ORM session
:param tid: The tenant ID
:param file_id: The ID of the file to be added
:param name: The file name
:param data: The file content
"""
file_obj = models.File()
file_obj.tid = tid
file_obj.id = file_id
file_obj.name = name
session.merge(file_obj)
def db_add_file(session, tid, id, name, data):
if id is not None:
file_obj = session.query(models.File).filter(models.File.tid == tid, models.File.id == id).one_or_none()
if file_obj is None:
file_obj = models.File()
file_obj.tid = tid
if id is not None:
file_obj.id = id
session.add(file_obj)
file_obj.name = name
file_obj.data = data
