def get_companies(company_id=None, count=None, offset=None):
result_list = []
company_id = str(company_id) if company_id else ''
if company_id:
company = CompanyDbObject.query.filter_by(id=company_id, deleted=False, _owner=current_user,
_batch=None, _copy_id=None).first()
if not company:
raise errors.EntityNotFound()
company_doc = CompanyObject.db_obj_to_field(company)
result_list.append(company_doc.get_api_structure())
result_count = 1
result_total = 1
else:
cur = CompanyDbObject.query.filter_by(deleted=False, _owner=current_user, _batch=None, _copy_id=None).order_by(
CompanyDbObject.id.asc())
result_total = cur.count()
cur = CompanyDbObject.query.filter_by(deleted=False, _owner=current_user, _batch=None, _copy_id=None).order_by(
CompanyDbObject.id.asc())
if count is not None:
cur = cur.limit(offset)
if offset is not None:
cur = cur.offset(offset)
result_count = cur.count()
for company in cur:
company_doc = CompanyObject.db_obj_to_field(company)
result_list.append(company_doc.get_api_structure())
return {'result': {
'companies': result_list,
'total': result_total,
'count': result_count
}}
def delete_company(company_id=None):
company_id = str(company_id) if company_id else None
company_db_obj = CompanyDbObject.query.filter_by(id=company_id, _owner=current_user, _batch=None).first()
if not company_db_obj:
raise errors.EntityNotFound()
company_db_obj.deleted = True
sqldb.session.commit()
return {"result": True}
def delete_person(person_id=None):
person_db_obj = PrivatePersonDbObject.query.filter_by(id=unicode(person_id), _owner=current_user,
_batch=None).first()
if not person_db_obj:
raise errors.EntityNotFound()
person_db_obj.deleted = True
sqldb.session.commit()
return {"result": True}
def update_company(company_id=None, company=None):
company_id = str(company_id) if company_id else None
company_db_obj = CompanyDbObject.query.filter_by(id=company_id, _owner=current_user, _batch=None).first()
if not company_db_obj:
raise errors.EntityNotFound()
old_company_doc = CompanyObject.db_obj_to_field(company_db_obj)
old_company_doc.update(company)
CompanyObject.update_db_obj(company_db_obj, company, api_data=True)
sqldb.session.commit()
company_doc = CompanyObject.db_obj_to_field(company_db_obj)
return {"result": company_doc.get_api_structure()}
def update_person(person_id=None, person=None):
person_db_obj = PrivatePersonDbObject.query.filter_by(id=unicode(person_id), _owner=current_user,
_batch=None).first()
if not person_db_obj:
raise errors.EntityNotFound()
PrivatePerson.update_db_obj(person_db_obj, person, api_data=True)
# validator = IsMyObjectVisitor(current_user._id) # todo:!!!
# try:
# validator.process(old_person_doc)
# except NotMineException:
# raise InvalidParameterValue('person')
sqldb.session.commit()
person_doc = PrivatePerson.db_obj_to_field(person_db_obj)
return {"result": person_doc.get_api_structure()}
def get_persons(person_id=None, count=None, offset=None):
result_list = []
person_id = str(person_id) if person_id else ''
if person_id:
person = PrivatePersonDbObject.query.filter_by(id=person_id, deleted=False, _owner=current_user,
_batch=None, _copy_id=None).first()
if not person:
raise errors.EntityNotFound('person_id')
person_doc = PrivatePerson.db_obj_to_field(person)
result_list.append(person_doc.get_api_structure())
result_count = 1
result_total = 1
else:
cur = PrivatePersonDbObject.query.filter_by(deleted=False, _owner=current_user, _batch=None, _copy_id=None).order_by(
PrivatePersonDbObject.id.asc())
result_total = cur.count()
cur = PrivatePersonDbObject.query.filter_by(deleted=False, _owner=current_user, _batch=None, _copy_id=None).order_by(
PrivatePersonDbObject.id.asc())
if count is not None:
cur = cur.limit(offset)
if offset is not None:
cur = cur.offset(offset)
result_count = cur.count()
for person in cur:
person_doc = PrivatePerson.db_obj_to_field(person)
result_list.append(person_doc.get_api_structure())
return {'result': {
'persons': result_list,
'total': result_total,
'count': result_count
}}