203 lines
6.4 KiB
Python
203 lines
6.4 KiB
Python
"""站点 CRUD（sys_site）"""
|
||
from __future__ import annotations
|
||
|
||
import math
|
||
from datetime import datetime
|
||
from typing import Any, Dict, List, Optional, Tuple
|
||
|
||
from sqlalchemy import and_, func, or_
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.core.utils import generate_uuid
|
||
from app.models.models import SysSite
|
||
|
||
|
||
def _row_to_dict(s: SysSite) -> Dict[str, Any]:
|
||
return {
|
||
"id": s.id,
|
||
"siteName": s.site_name,
|
||
"siteCode": s.site_code,
|
||
"siteType": s.site_type or "",
|
||
"status": s.status,
|
||
"address": s.address or "",
|
||
"contactName": s.contact_name or "",
|
||
"contactPhone": s.contact_phone or "",
|
||
"businessScope": s.business_scope or "",
|
||
"siteImage": s.site_image or "",
|
||
"tenantNo": s.tenant_no,
|
||
"orderNum": s.order_num if s.order_num is not None else 0,
|
||
"remark": s.remark or "",
|
||
"createTime": s.create_time.strftime("%Y-%m-%d %H:%M:%S") if s.create_time else "",
|
||
"updateTime": s.update_time.strftime("%Y-%m-%d %H:%M:%S") if s.update_time else "",
|
||
}
|
||
|
||
|
||
def list_sites(
|
||
db: Session,
|
||
*,
|
||
page: int = 1,
|
||
page_size: int = 10,
|
||
keyword: Optional[str] = None,
|
||
tenant_no: Optional[str] = None,
|
||
user_info=None,
|
||
) -> Tuple[List[Dict[str, Any]], int]:
|
||
def _filtered():
|
||
q0 = db.query(SysSite).filter(SysSite.del_flag == 0)
|
||
if tenant_no:
|
||
q0 = q0.filter(SysSite.tenant_no == tenant_no)
|
||
kw = (keyword or "").strip()
|
||
if kw:
|
||
like = f"%{kw}%"
|
||
q0 = q0.filter(or_(SysSite.site_name.like(like), SysSite.site_code.like(like)))
|
||
if user_info:
|
||
from app.core.data_scope import apply_data_scope
|
||
q0 = apply_data_scope(q0, SysSite, user_info, db, tenant_no)
|
||
return q0
|
||
|
||
total = _filtered().order_by(None).with_entities(func.count(SysSite.id)).scalar() or 0
|
||
page = max(1, page)
|
||
page_size = min(max(1, page_size), 100)
|
||
rows = (
|
||
_filtered()
|
||
.order_by(SysSite.create_time.desc())
|
||
.offset((page - 1) * page_size)
|
||
.limit(page_size)
|
||
.all()
|
||
)
|
||
return [_row_to_dict(x) for x in rows], int(total)
|
||
|
||
|
||
def get_site(db: Session, site_id: str) -> Optional[SysSite]:
|
||
return (
|
||
db.query(SysSite)
|
||
.filter(and_(SysSite.id == site_id, SysSite.del_flag == 0))
|
||
.first()
|
||
)
|
||
|
||
|
||
def site_code_exists(db: Session, code: str, tenant_no: str, exclude_id: Optional[str] = None) -> bool:
|
||
q = db.query(SysSite).filter(and_(SysSite.site_code == code.strip(), SysSite.tenant_no == tenant_no, SysSite.del_flag == 0))
|
||
if exclude_id:
|
||
q = q.filter(SysSite.id != exclude_id)
|
||
return q.first() is not None
|
||
|
||
|
||
def create_site(
|
||
db: Session,
|
||
*,
|
||
site_name: str,
|
||
site_code: str,
|
||
site_type: Optional[str] = None,
|
||
status: str = "1",
|
||
address: Optional[str] = None,
|
||
contact_name: Optional[str] = None,
|
||
contact_phone: Optional[str] = None,
|
||
business_scope: Optional[str] = None,
|
||
site_image: Optional[str] = None,
|
||
tenant_no: str = "zkxw",
|
||
order_num: Optional[int] = None,
|
||
remark: Optional[str] = None,
|
||
user_id: Optional[str] = None,
|
||
) -> SysSite:
|
||
sc = (site_code or "").strip()
|
||
if not sc:
|
||
raise ValueError("站点编码不能为空")
|
||
if site_code_exists(db, sc, tenant_no):
|
||
raise ValueError("站点编码已存在")
|
||
obj = SysSite(
|
||
id=generate_uuid(),
|
||
site_name=(site_name or "").strip() or sc,
|
||
site_code=sc,
|
||
site_type=(site_type or "").strip() or None,
|
||
status=status or "1",
|
||
address=(address or "").strip() or None,
|
||
contact_name=(contact_name or "").strip() or None,
|
||
contact_phone=(contact_phone or "").strip() or None,
|
||
business_scope=(business_scope or "").strip() or None,
|
||
site_image=(site_image or "").strip() or None,
|
||
tenant_no=tenant_no,
|
||
order_num=order_num if order_num is not None else 0,
|
||
remark=(remark or "").strip() or None,
|
||
create_by=user_id,
|
||
create_dept_id="",
|
||
create_time=datetime.now(),
|
||
update_by=user_id,
|
||
update_time=datetime.now(),
|
||
del_flag=0,
|
||
)
|
||
db.add(obj)
|
||
db.commit()
|
||
db.refresh(obj)
|
||
return obj
|
||
|
||
|
||
def update_site(
|
||
db: Session,
|
||
site_id: str,
|
||
*,
|
||
site_name: Optional[str] = None,
|
||
site_code: Optional[str] = None,
|
||
site_type: Optional[str] = None,
|
||
status: Optional[str] = None,
|
||
address: Optional[str] = None,
|
||
contact_name: Optional[str] = None,
|
||
contact_phone: Optional[str] = None,
|
||
business_scope: Optional[str] = None,
|
||
site_image: Optional[str] = None,
|
||
order_num: Optional[int] = None,
|
||
remark: Optional[str] = None,
|
||
user_id: Optional[str] = None,
|
||
) -> SysSite:
|
||
obj = get_site(db, site_id)
|
||
if not obj:
|
||
raise ValueError("站点不存在")
|
||
if site_code is not None:
|
||
new_code = site_code.strip()
|
||
if not new_code:
|
||
raise ValueError("站点编码不能为空")
|
||
if new_code != obj.site_code and site_code_exists(db, new_code, obj.tenant_no, exclude_id=site_id):
|
||
raise ValueError("站点编码已存在")
|
||
obj.site_code = new_code
|
||
if site_name is not None:
|
||
obj.site_name = (site_name or "").strip() or obj.site_name
|
||
if site_type is not None:
|
||
obj.site_type = (site_type or "").strip() or None
|
||
if status is not None:
|
||
obj.status = status
|
||
if address is not None:
|
||
obj.address = (address or "").strip() or None
|
||
if contact_name is not None:
|
||
obj.contact_name = (contact_name or "").strip() or None
|
||
if contact_phone is not None:
|
||
obj.contact_phone = (contact_phone or "").strip() or None
|
||
if business_scope is not None:
|
||
obj.business_scope = (business_scope or "").strip() or None
|
||
if site_image is not None:
|
||
obj.site_image = (site_image or "").strip() or None
|
||
if order_num is not None:
|
||
obj.order_num = order_num
|
||
if remark is not None:
|
||
obj.remark = (remark or "").strip() or None
|
||
obj.update_by = user_id
|
||
obj.update_time = datetime.now()
|
||
db.add(obj)
|
||
db.commit()
|
||
db.refresh(obj)
|
||
return obj
|
||
|
||
|
||
def soft_delete_site(db: Session, site_id: str) -> None:
|
||
obj = get_site(db, site_id)
|
||
if not obj:
|
||
raise ValueError("站点不存在")
|
||
obj.del_flag = 1
|
||
obj.update_time = datetime.now()
|
||
db.add(obj)
|
||
db.commit()
|
||
|
||
|
||
def total_pages(total: int, page_size: int) -> int:
|
||
if total <= 0:
|
||
return 0
|
||
return int(math.ceil(total / float(page_size)))
|