def add_name(unit_id, lang, short, full, desc):
date = datetime.utcnow()
if not isinstance(unit_id, uuid.UUID):
unit_id = uuid.UUID(unit_id)
with UnitRepository.atomic() as txn:
try:
name_id = UnitRepository.add_name(unit_id, lang, short, full,
desc)
LogWriter.push_as_user(name_id, get_user_id(), None,
EntityLogModifyType.CREATE, None, None)
return {'id': str(name_id)}
except Exception as ex:
txn.rollback()
raise ex
def create_unit():
date = datetime.utcnow()
user_id = get_user_id()
with UnitRepository.atomic() as txn:
try:
unit_id = UnitRepository.create_unit()
UnitRepository.add_ownership_user(user_id, unit_id,
Privacy.PRIVATE)
LogWriter.push_as_user(unit_id, user_id, None,
EntityLogModifyType.CREATE, None, None)
return {'id': str(unit_id)}
except Exception as ex:
txn.rollback()
raise ex
def search_units(search_query, in_maintenance=0):
user_id = get_user_id()
units = UnitRepository.search_for_user(user_id, search_query,
in_maintenance)
units_mapped = [UnitMapper.map_unit(unit) for unit in units]
return {'units': units_mapped, 'count': len(units_mapped)}
def get_units(with_names=False, in_maintenance=0):
user_id = get_user_id()
units = UnitRepository.list_for_user(user_id, with_names,
in_maintenance)
units_mapped = [UnitMapper.map_unit(unit) for unit in units]
return {'units': units_mapped, 'count': len(units_mapped)}