Fix API compatibility and add user/role/permission and asset import/export
This commit is contained in:
332
backend_new/app/crud/allocation.py
Normal file
332
backend_new/app/crud/allocation.py
Normal file
@@ -0,0 +1,332 @@
|
||||
"""
|
||||
资产分配相关CRUD操作
|
||||
"""
|
||||
from typing import List, Optional, Tuple
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import and_, or_
|
||||
from app.models.allocation import AssetAllocationOrder, AssetAllocationItem
|
||||
from app.models.asset import Asset
|
||||
from app.schemas.allocation import AllocationOrderCreate, AllocationOrderUpdate
|
||||
|
||||
|
||||
class AllocationOrderCRUD:
|
||||
"""分配单CRUD操作"""
|
||||
|
||||
def get(self, db: Session, id: int) -> Optional[AssetAllocationOrder]:
|
||||
"""根据ID获取分配单"""
|
||||
return db.query(AssetAllocationOrder).filter(
|
||||
AssetAllocationOrder.id == id
|
||||
).first()
|
||||
|
||||
def get_by_code(self, db: Session, order_code: str) -> Optional[AssetAllocationOrder]:
|
||||
"""根据单号获取分配单"""
|
||||
return db.query(AssetAllocationOrder).filter(
|
||||
AssetAllocationOrder.order_code == order_code
|
||||
).first()
|
||||
|
||||
def get_multi(
|
||||
self,
|
||||
db: Session,
|
||||
skip: int = 0,
|
||||
limit: int = 20,
|
||||
order_type: Optional[str] = None,
|
||||
approval_status: Optional[str] = None,
|
||||
execute_status: Optional[str] = None,
|
||||
applicant_id: Optional[int] = None,
|
||||
target_organization_id: Optional[int] = None,
|
||||
keyword: Optional[str] = None
|
||||
) -> Tuple[List[AssetAllocationOrder], int]:
|
||||
"""获取分配单列表"""
|
||||
query = db.query(AssetAllocationOrder)
|
||||
|
||||
# 筛选条件
|
||||
if order_type:
|
||||
query = query.filter(AssetAllocationOrder.order_type == order_type)
|
||||
if approval_status:
|
||||
query = query.filter(AssetAllocationOrder.approval_status == approval_status)
|
||||
if execute_status:
|
||||
query = query.filter(AssetAllocationOrder.execute_status == execute_status)
|
||||
if applicant_id:
|
||||
query = query.filter(AssetAllocationOrder.applicant_id == applicant_id)
|
||||
if target_organization_id:
|
||||
query = query.filter(AssetAllocationOrder.target_organization_id == target_organization_id)
|
||||
if keyword:
|
||||
query = query.filter(
|
||||
or_(
|
||||
AssetAllocationOrder.order_code.like(f"%{keyword}%"),
|
||||
AssetAllocationOrder.title.like(f"%{keyword}%")
|
||||
)
|
||||
)
|
||||
|
||||
# 排序
|
||||
query = query.order_by(AssetAllocationOrder.created_at.desc())
|
||||
|
||||
# 总数
|
||||
total = query.count()
|
||||
|
||||
# 分页
|
||||
items = query.offset(skip).limit(limit).all()
|
||||
|
||||
return items, total
|
||||
|
||||
def create(
|
||||
self,
|
||||
db: Session,
|
||||
obj_in: AllocationOrderCreate,
|
||||
order_code: str,
|
||||
applicant_id: int
|
||||
) -> AssetAllocationOrder:
|
||||
"""创建分配单"""
|
||||
# 创建分配单
|
||||
db_obj = AssetAllocationOrder(
|
||||
order_code=order_code,
|
||||
order_type=obj_in.order_type,
|
||||
title=obj_in.title,
|
||||
source_organization_id=obj_in.source_organization_id,
|
||||
target_organization_id=obj_in.target_organization_id,
|
||||
applicant_id=applicant_id,
|
||||
expect_execute_date=obj_in.expect_execute_date,
|
||||
remark=obj_in.remark,
|
||||
created_by=applicant_id,
|
||||
approval_status="pending",
|
||||
execute_status="pending"
|
||||
)
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
|
||||
# 创建分配单明细
|
||||
self._create_items(
|
||||
db=db,
|
||||
order_id=db_obj.id,
|
||||
asset_ids=obj_in.asset_ids,
|
||||
target_org_id=obj_in.target_organization_id
|
||||
)
|
||||
|
||||
return db_obj
|
||||
|
||||
def update(
|
||||
self,
|
||||
db: Session,
|
||||
db_obj: AssetAllocationOrder,
|
||||
obj_in: AllocationOrderUpdate,
|
||||
updater_id: int
|
||||
) -> AssetAllocationOrder:
|
||||
"""更新分配单"""
|
||||
update_data = obj_in.model_dump(exclude_unset=True)
|
||||
for field, value in update_data.items():
|
||||
setattr(db_obj, field, value)
|
||||
db_obj.updated_by = updater_id
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def approve(
|
||||
self,
|
||||
db: Session,
|
||||
db_obj: AssetAllocationOrder,
|
||||
approval_status: str,
|
||||
approver_id: int,
|
||||
approval_remark: Optional[str] = None
|
||||
) -> AssetAllocationOrder:
|
||||
"""审批分配单"""
|
||||
from datetime import datetime
|
||||
|
||||
db_obj.approval_status = approval_status
|
||||
db_obj.approver_id = approver_id
|
||||
db_obj.approval_time = datetime.utcnow()
|
||||
db_obj.approval_remark = approval_remark
|
||||
|
||||
# 如果审批通过，自动设置为可执行状态
|
||||
if approval_status == "approved":
|
||||
db_obj.execute_status = "pending"
|
||||
elif approval_status == "rejected":
|
||||
db_obj.execute_status = "cancelled"
|
||||
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def execute(
|
||||
self,
|
||||
db: Session,
|
||||
db_obj: AssetAllocationOrder,
|
||||
executor_id: int
|
||||
) -> AssetAllocationOrder:
|
||||
"""执行分配单"""
|
||||
from datetime import datetime, date
|
||||
|
||||
db_obj.execute_status = "completed"
|
||||
db_obj.actual_execute_date = date.today()
|
||||
db_obj.executor_id = executor_id
|
||||
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def cancel(self, db: Session, db_obj: AssetAllocationOrder) -> AssetAllocationOrder:
|
||||
"""取消分配单"""
|
||||
db_obj.approval_status = "cancelled"
|
||||
db_obj.execute_status = "cancelled"
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, db: Session, id: int) -> bool:
|
||||
"""删除分配单"""
|
||||
obj = self.get(db, id)
|
||||
if obj:
|
||||
db.delete(obj)
|
||||
db.commit()
|
||||
return True
|
||||
return False
|
||||
|
||||
def get_statistics(
|
||||
self,
|
||||
db: Session,
|
||||
applicant_id: Optional[int] = None
|
||||
) -> dict:
|
||||
"""获取分配单统计信息"""
|
||||
query = db.query(AssetAllocationOrder)
|
||||
|
||||
if applicant_id:
|
||||
query = query.filter(AssetAllocationOrder.applicant_id == applicant_id)
|
||||
|
||||
total = query.count()
|
||||
pending = query.filter(AssetAllocationOrder.approval_status == "pending").count()
|
||||
approved = query.filter(AssetAllocationOrder.approval_status == "approved").count()
|
||||
rejected = query.filter(AssetAllocationOrder.approval_status == "rejected").count()
|
||||
executing = query.filter(AssetAllocationOrder.execute_status == "executing").count()
|
||||
completed = query.filter(AssetAllocationOrder.execute_status == "completed").count()
|
||||
|
||||
return {
|
||||
"total": total,
|
||||
"pending": pending,
|
||||
"approved": approved,
|
||||
"rejected": rejected,
|
||||
"executing": executing,
|
||||
"completed": completed
|
||||
}
|
||||
|
||||
def _create_items(
|
||||
self,
|
||||
db: Session,
|
||||
order_id: int,
|
||||
asset_ids: List[int],
|
||||
target_org_id: int
|
||||
):
|
||||
"""创建分配单明细"""
|
||||
# 查询资产信息
|
||||
assets = db.query(Asset).filter(Asset.id.in_(asset_ids)).all()
|
||||
|
||||
for asset in assets:
|
||||
item = AssetAllocationItem(
|
||||
order_id=order_id,
|
||||
asset_id=asset.id,
|
||||
asset_code=asset.asset_code,
|
||||
asset_name=asset.asset_name,
|
||||
from_organization_id=asset.organization_id,
|
||||
to_organization_id=target_org_id,
|
||||
from_status=asset.status,
|
||||
to_status=self._get_target_status(asset.status),
|
||||
execute_status="pending"
|
||||
)
|
||||
db.add(item)
|
||||
|
||||
db.commit()
|
||||
|
||||
def _get_target_status(self, current_status: str) -> str:
|
||||
"""根据当前状态获取目标状态"""
|
||||
status_map = {
|
||||
"in_stock": "transferring",
|
||||
"in_use": "transferring",
|
||||
"maintenance": "in_stock"
|
||||
}
|
||||
return status_map.get(current_status, "transferring")
|
||||
|
||||
|
||||
class AllocationItemCRUD:
|
||||
"""分配单明细CRUD操作"""
|
||||
|
||||
def get_by_order(self, db: Session, order_id: int) -> List[AssetAllocationItem]:
|
||||
"""根据分配单ID获取明细列表"""
|
||||
return db.query(AssetAllocationItem).filter(
|
||||
AssetAllocationItem.order_id == order_id
|
||||
).order_by(AssetAllocationItem.id).all()
|
||||
|
||||
def get_multi(
|
||||
self,
|
||||
db: Session,
|
||||
skip: int = 0,
|
||||
limit: int = 20,
|
||||
order_id: Optional[int] = None,
|
||||
execute_status: Optional[str] = None
|
||||
) -> Tuple[List[AssetAllocationItem], int]:
|
||||
"""获取明细列表"""
|
||||
query = db.query(AssetAllocationItem)
|
||||
|
||||
if order_id:
|
||||
query = query.filter(AssetAllocationItem.order_id == order_id)
|
||||
if execute_status:
|
||||
query = query.filter(AssetAllocationItem.execute_status == execute_status)
|
||||
|
||||
total = query.count()
|
||||
items = query.offset(skip).limit(limit).all()
|
||||
|
||||
return items, total
|
||||
|
||||
def update_execute_status(
|
||||
self,
|
||||
db: Session,
|
||||
item_id: int,
|
||||
execute_status: str,
|
||||
failure_reason: Optional[str] = None
|
||||
) -> AssetAllocationItem:
|
||||
"""更新明细执行状态"""
|
||||
from datetime import datetime
|
||||
|
||||
item = db.query(AssetAllocationItem).filter(
|
||||
AssetAllocationItem.id == item_id
|
||||
).first()
|
||||
|
||||
if item:
|
||||
item.execute_status = execute_status
|
||||
item.execute_time = datetime.utcnow()
|
||||
item.failure_reason = failure_reason
|
||||
db.add(item)
|
||||
db.commit()
|
||||
db.refresh(item)
|
||||
|
||||
return item
|
||||
|
||||
def batch_update_execute_status(
|
||||
self,
|
||||
db: Session,
|
||||
order_id: int,
|
||||
execute_status: str
|
||||
):
|
||||
"""批量更新明细执行状态"""
|
||||
from datetime import datetime
|
||||
|
||||
items = db.query(AssetAllocationItem).filter(
|
||||
and_(
|
||||
AssetAllocationItem.order_id == order_id,
|
||||
AssetAllocationItem.execute_status == "pending"
|
||||
)
|
||||
).all()
|
||||
|
||||
for item in items:
|
||||
item.execute_status = execute_status
|
||||
item.execute_time = datetime.utcnow()
|
||||
db.add(item)
|
||||
|
||||
db.commit()
|
||||
|
||||
|
||||
# 创建全局实例
|
||||
allocation_order = AllocationOrderCRUD()
|
||||
allocation_item = AllocationItemCRUD()
|
||||
Reference in New Issue
Block a user