| | |
|---|
| | | import uuid |
|---|
| | | |
|---|
| | | from app.Log import logger |
|---|
| | | from app.models import OrganizationModel, GroupModel |
|---|
| | | from app.models.role_model import RoleModel |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | async def get_organization_list(db, deptName: str): |
|---|
| | | query = db.query(OrganizationModel).filter(OrganizationModel.organization_id == None) |
|---|
| | | if deptName: |
|---|
| | | query = query.filter(OrganizationModel.name.like('%{}%'.format(deptName))) |
|---|
| | | depts = query.order_by(OrganizationModel.seq.desc()).all() |
|---|
| | | return {"total": query.count(), "rows": [dept.to_json() for dept in depts]} |
|---|
| | | |
|---|
| | | |
|---|
| | | async def create_dept(db, dept_name, leader, phone, address, status, order_num, roles, groups,parent_id): |
|---|
| | | try: |
|---|
| | | dept_model = OrganizationModel(id=str(uuid.uuid4()),name=dept_name, address=address,leader=leader,phone=phone,seq=order_num,status=status) |
|---|
| | | if parent_id: |
|---|
| | | dept_model.parent = db.get(OrganizationModel, parent_id) |
|---|
| | | if roles: |
|---|
| | | dept_model.roles = [db.get(RoleModel, roleId) for roleId in roles] |
|---|
| | | # if groups: |
|---|
| | | # dept_model.groups = [db.get(GroupModel, groupId) for groupId in groups] |
|---|
| | | db.add(dept_model) |
|---|
| | | db.commit() |
|---|
| | | db.refresh(dept_model) |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | db.rollback() |
|---|
| | | return False |
|---|
| | | return True |
|---|
| | | |
|---|
| | | |
|---|
| | | async def edit_dept_data(db, dept_id, dept_name, leader, phone, address, status, order_num, roles, groups,parent_id): |
|---|
| | | try: |
|---|
| | | dept_model = db.query(OrganizationModel).filter(OrganizationModel.id == dept_id).first() |
|---|
| | | dept_model.name = dept_name |
|---|
| | | dept_model.address = address |
|---|
| | | dept_model.phone = phone |
|---|
| | | dept_model.leader = leader |
|---|
| | | dept_model.status = status |
|---|
| | | dept_model.seq = order_num |
|---|
| | | # if parent_id: |
|---|
| | | # dept_model.parent = db.get(OrganizationModel, parent_id) |
|---|
| | | if roles: |
|---|
| | | dept_model.roles = [db.get(RoleModel, roleId) for roleId in roles] |
|---|
| | | # if groups: |
|---|
| | | # dept_model.groups = [db.get(GroupModel, groupId) for groupId in groups] |
|---|
| | | # db.add(dept_model) |
|---|
| | | db.commit() |
|---|
| | | db.refresh(dept_model) |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | db.rollback() |
|---|
| | | return False |
|---|
| | | return True |
|---|
| | | |
|---|
| | | |
|---|
| | | async def edit_dept_parent(db, dept_id, parent_id, order_num): |
|---|
| | | try: |
|---|
| | | dept_model = db.query(OrganizationModel).filter(OrganizationModel.id == dept_id).first() |
|---|
| | | dept_model.parent = db.get(OrganizationModel, parent_id) |
|---|
| | | dept_model.seq = order_num |
|---|
| | | db.commit() |
|---|
| | | db.refresh(dept_model) |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | db.rollback() |
|---|
| | | return False |
|---|
| | | return True |
|---|
| | | |
|---|
| | | |
|---|
| | | async def get_organization_info(db, dept_id: str): |
|---|
| | | dept = db.query(OrganizationModel).filter(OrganizationModel.id.__eq__(dept_id)).first() |
|---|
| | | |
|---|
| | | return {"total": 0, "data": dept.to_json()} |
|---|
| | | |
|---|
| | | |
|---|
| | | async def delete_organization_info(db, dept_id: str): |
|---|
| | | try: |
|---|
| | | db.query(OrganizationModel).filter(OrganizationModel.id.__eq__(dept_id)).delete() |
|---|
| | | db.commit() |
|---|
| | | except Exception as e: |
|---|
| | | logger.error(e) |
|---|
| | | db.rollback() |
|---|
| | | return False |
|---|
| | | return True |
|---|
