"""Dashboard 统计服务 — 上帝视角（全厂全系统数据，不按用户过滤）""" from datetime import datetime from sqlalchemy import select, func, or_ from sqlalchemy.ext.asyncio import AsyncSession from pydantic import BaseModel # ============================================================ # Schemas # ============================================================ class DashboardStats(BaseModel): products_total: int products_pending: int products_in_progress: int products_completed: int tasks_total: int tasks_pending: int tasks_in_progress: int tasks_completed: int tasks_rejected: int tasks_rework: int unread_notifications: int = 0 unread_messages: int = 0 class WipTask(BaseModel): task_id: str task_name: str assignee: str product_sn: str # 16位HEX身份证 external_serial: str | None # 业务序列号 material_name: str # 设备名称 spec_model: str # 规格型号 status: str received_at: str duration_hours: float class CompletedTask(BaseModel): task_id: str task_name: str # 完成的工序节点 assignee: str # 完成人中文姓名 product_sn: str # 16位HEX身份证 external_serial: str | None # 业务序列号 material_name: str # 产品名称（物料名称） spec_model: str # 规格型号 completed_at: str # 完成时间 ISO class PersonDevice(BaseModel): product_id: str serial_number: str # 16位HEX身份证 external_serial: str | None # 业务序列号 material_name: str # 产品名称 spec_model: str # 规格型号 task_status: str # 该设备名下的状态 WIP/PENDING duration_hours: float # 滞留时长（在当前人手上多久，小时） received_at: str | None # 接收时间（北京时间 MM-DD HH:mm） class PersonWorkload(BaseModel): assignee_id: str assignee_name: str # 中文姓名 device_count: int devices: list[PersonDevice] class PersonHistoryRecord(BaseModel): task_id: str task_name: str # 工序/任务名 assignee_id: str # 负责人ID assignee_name: str # 负责人姓名 product_sn: str # 16位身份证 external_serial: str | None # 业务序列号 material_name: str # 产品名称 spec_model: str # 规格型号 status: str # 状态 WIP/PENDING/COMPLETED received_at: str | None # 开始时间 MM-DD HH:mm completed_at: str | None # 结束时间 MM-DD HH:mm（进行中为 None） duration_hours: float # 总耗时（小时） latest_valid_remark: str | None = None # 最新有效备注（排除系统转交类） record_count: int = 0 # 记录总数（含系统备注） class ProductMessageItem(BaseModel): id: str content: str operator_name: str # 留言人中文姓名 product_sn: str # 16位HEX系统追溯码 external_serial: str | None # 业务产品序列号（如 25022） material_name: str # 物料名称 created_at: str # ISO时间字符串 class ProductMessageList(BaseModel): items: list[ProductMessageItem] total: int # ============================================================ # 看板统计（时间快照语义） # ============================================================ async def get_dashboard_stats( db: AsyncSession, since: datetime | None = None, until: datetime | None = None, ) -> DashboardStats: """ 上帝视角 — 全厂全系统统计。 时间筛选规则： - PENDING / WIP / 总数：永远忽略时间筛选，返回实时快照。 - COMPLETED / REJECTED / 完成率：严格按 since~until 过滤（用于时段报表）。 """ from app.models.product import Product from app.models.task import ( Task, TASK_STATUS_PENDING, TASK_STATUS_WIP, TASK_STATUS_COMPLETED, TASK_STATUS_REJECTED, ) from app.models.notification import Notification from app.models.message import ProductMessage # ── 产品（实时快照，不过滤） ── p_total = await db.scalar(select(func.count(Product.id))) p_pending = await db.scalar(select(func.count(Product.id)).where(Product.status == "pending")) p_progress = await db.scalar(select(func.count(Product.id)).where(Product.status == "in_progress")) p_done = await db.scalar(select(func.count(Product.id)).where(Product.status == "completed")) # ── 任务实时快照（PENDING/WIP/返工 — 永远不过滤） ── t_pending = await db.scalar(select(func.count(Task.id)).where(Task.status == TASK_STATUS_PENDING)) t_progress = await db.scalar(select(func.count(Task.id)).where(Task.status == TASK_STATUS_WIP)) t_rework = await db.scalar(select(func.count(Task.id)).where(Task.is_rework.is_(True))) # ── 任务已完成/驳回（时间可过滤） ── t_done_q = select(func.count(Task.id)).where(Task.status == TASK_STATUS_COMPLETED) t_rej_q = select(func.count(Task.id)).where(Task.status == TASK_STATUS_REJECTED) if since: t_done_q = t_done_q.where(Task.completed_at >= since) t_rej_q = t_rej_q.where(Task.completed_at >= since) if until: t_done_q = t_done_q.where(Task.completed_at <= until) t_rej_q = t_rej_q.where(Task.completed_at <= until) t_done = await db.scalar(t_done_q) t_rejected = await db.scalar(t_rej_q) # ── 任务总数 = 实时快照段 + 时间过滤段（确保进度条段总和=总数） ── t_total = (t_pending or 0) + (t_progress or 0) + (t_done or 0) + (t_rejected or 0) # ── 通知 & 留言（实时快照） ── unread_notif = await db.scalar( select(func.count(Notification.id)).where(Notification.is_read.is_(False)) ) unread_msg = await db.scalar(select(func.count(ProductMessage.id))) return DashboardStats( products_total=p_total or 0, products_pending=p_pending or 0, products_in_progress=p_progress or 0, products_completed=p_done or 0, tasks_total=t_total or 0, tasks_pending=t_pending or 0, tasks_in_progress=t_progress or 0, tasks_completed=t_done or 0, tasks_rejected=t_rejected or 0, tasks_rework=t_rework or 0, unread_notifications=unread_notif or 0, unread_messages=unread_msg or 0, ) # ============================================================ # 在制品看板（永远实时） # ============================================================ async def get_wip_tasks(db: AsyncSession, limit: int = 20) -> list[WipTask]: from app.models.task import Task, TASK_STATUS_PENDING, TASK_STATUS_WIP from app.models.product import Product from app.core.time_utils import get_beijing_time, BEIJING_TZ stmt = ( select(Task, Product.serial_number, Product.external_serial, Product.material_name, Product.spec_model) .join(Product, Task.product_id == Product.id) .where(Task.status.in_([TASK_STATUS_PENDING, TASK_STATUS_WIP])) .limit(limit * 2) ) result = await db.execute(stmt) rows = result.all() raw_ids = list({t.assignee_id for t, *_ in rows if t.assignee_id}) name_map: dict[str, str] = {} if raw_ids: from app.services.mom_cache import get_display_names name_map = get_display_names(raw_ids) now = get_beijing_time() wip_list: list[WipTask] = [] for task, product_sn, ext_sn, mat_name, spec in rows: start = task.received_at or task.created_at if start: if start.tzinfo is None: start = start.replace(tzinfo=BEIJING_TZ) else: start = start.astimezone(BEIJING_TZ) hours = round((now - start).total_seconds() / 3600, 1) recv_str = start.strftime("%m-%d %H:%M") else: hours = 0 recv_str = "" wip_list.append(WipTask( task_id=str(task.id), task_name=task.task_name, assignee=name_map.get(task.assignee_id or "", task.assignee_id or "未分配"), product_sn=product_sn or "", external_serial=ext_sn or None, material_name=mat_name or "", spec_model=spec or "", status=task.status, received_at=recv_str, duration_hours=hours, )) # 统一按滞留时间降序排列（无视 status，纯数值排序） wip_list.sort(key=lambda t: t.duration_hours, reverse=True) return wip_list[:limit] # ============================================================ # 已完成任务明细（流转完成率下钻 — 按时段过滤） # ============================================================ async def get_completed_tasks( db: AsyncSession, since: datetime | None = None, until: datetime | None = None, limit: int = 200, ) -> list[CompletedTask]: """按时段查询已完成任务明细（上帝视角），用于「流转完成率」卡片下钻。""" from app.models.task import Task, TASK_STATUS_COMPLETED from app.models.product import Product from app.core.time_utils import BEIJING_TZ stmt = ( select(Task, Product.serial_number, Product.external_serial, Product.material_name, Product.spec_model) .join(Product, Task.product_id == Product.id) .where(Task.status == TASK_STATUS_COMPLETED) ) if since: stmt = stmt.where(Task.completed_at >= since) if until: stmt = stmt.where(Task.completed_at <= until) stmt = stmt.order_by(Task.completed_at.desc()).limit(limit) result = await db.execute(stmt) rows = result.all() raw_ids = list({t.assignee_id for t, *_ in rows if t.assignee_id}) name_map: dict[str, str] = {} if raw_ids: from app.services.mom_cache import get_display_names name_map = get_display_names(raw_ids) items: list[CompletedTask] = [] for task, sn, ext, mat, spec in rows: t = task.completed_at if t: if t.tzinfo is None: from datetime import timezone as dt_timezone t = t.replace(tzinfo=dt_timezone.utc).astimezone(BEIJING_TZ) else: t = t.astimezone(BEIJING_TZ) time_str = t.isoformat() if t else "" items.append(CompletedTask( task_id=str(task.id), task_name=task.task_name, assignee=name_map.get(task.assignee_id or "", task.assignee_id or "未分配"), product_sn=sn or "", external_serial=ext or None, material_name=mat or "", spec_model=spec or "", completed_at=time_str, )) return items # ============================================================ # 人员负载（按人聚合在制品设备 — 独立「人员看板」） # ============================================================ async def get_people_workload(db: AsyncSession) -> list[PersonWorkload]: """上帝视角 — 按负责人聚合当前在制品设备（WIP/PENDING 任务，product 去重，含滞留时长）。""" from app.models.task import Task, TASK_STATUS_PENDING, TASK_STATUS_WIP from app.models.product import Product from app.core.time_utils import get_beijing_time, BEIJING_TZ stmt = ( select( Task.assignee_id, Product.id, Product.serial_number, Product.external_serial, Product.material_name, Product.spec_model, Task.status, Task.received_at, Task.created_at, ) .join(Product, Task.product_id == Product.id) .where( Task.status.in_([TASK_STATUS_PENDING, TASK_STATUS_WIP]), Task.assignee_id.isnot(None), ) .order_by(Task.assignee_id, Product.created_at.desc()) ) result = await db.execute(stmt) rows = result.all() def _to_bj(dt): if not dt: return None if dt.tzinfo is None: from datetime import timezone as dt_timezone dt = dt.replace(tzinfo=dt_timezone.utc).astimezone(BEIJING_TZ) else: dt = dt.astimezone(BEIJING_TZ) return dt now = get_beijing_time() # 中间聚合：by_assignee[assignee][product_id] -> 设备快照 + 最早接手时间 agg: dict[str, dict[str, dict]] = {} for row in rows: assignee = row[0] product_id = str(row[1]) status = row[6] or "" received_dt = _to_bj(row[7] or row[8]) entry = agg.setdefault(assignee, {}).setdefault(product_id, { "status": "", "earliest_dt": None, "serial": row[2] or "", "ext": row[3] or None, "mat": row[4] or "", "spec": row[5] or "", }) # 状态优先级 WIP > PENDING if status == "WIP": entry["status"] = "WIP" elif not entry["status"]: entry["status"] = status # 取最早接手时间 if received_dt and (entry["earliest_dt"] is None or received_dt < entry["earliest_dt"]): entry["earliest_dt"] = received_dt raw_ids = list(agg.keys()) name_map: dict[str, str] = {} if raw_ids: from app.services.mom_cache import get_display_names name_map = get_display_names(raw_ids) workloads: list[PersonWorkload] = [] for assignee, prods in agg.items(): devices: list[PersonDevice] = [] for product_id, e in prods.items(): dt = e["earliest_dt"] hours = round((now - dt).total_seconds() / 3600, 1) if dt else 0.0 received_str = dt.strftime("%m-%d %H:%M") if dt else None devices.append(PersonDevice( product_id=product_id, serial_number=e["serial"], external_serial=e["ext"], material_name=e["mat"], spec_model=e["spec"], task_status=e["status"], duration_hours=hours, received_at=received_str, )) workloads.append(PersonWorkload( assignee_id=assignee, assignee_name=name_map.get(assignee, assignee), device_count=len(devices), devices=devices, )) workloads.sort(key=lambda w: w.device_count, reverse=True) return workloads # ============================================================ # 历史人员看板（按人聚合历史备注记录，按时段过滤） # ============================================================ async def get_people_history( db: AsyncSession, since: datetime | None = None, until: datetime | None = None, assignee_id: str | None = None, spec_model: str | None = None, product_sn: str | None = None, task_name: str | None = None, ) -> list[PersonHistoryRecord]: """上帝视角 — 人员效能与工时台账（平铺 Task 明细，含 WIP/PENDING/COMPLETED）。""" from app.models.task import Task, TASK_STATUS_WIP, TASK_STATUS_PENDING, TASK_STATUS_COMPLETED from app.models.product import Product from app.core.time_utils import get_beijing_time, BEIJING_TZ from sqlalchemy import func now = get_beijing_time() # ── 关联 TaskRecord：最新有效备注 + 记录总数 ── from app.models.task import TaskRecord from sqlalchemy import and_, or_ # 系统自动备注关键字（转交/移交/撤回/派发/驳回等） sys_remark = or_( TaskRecord.remark.ilike("%转交%"), TaskRecord.remark.ilike("%移交%"), TaskRecord.remark.ilike("%撤回%"), TaskRecord.remark.ilike("%重新接手%"), TaskRecord.remark.ilike("%派发%"), TaskRecord.remark.ilike("%分配%"), TaskRecord.remark.ilike("%驳回%"), TaskRecord.remark.ilike("%返工%"), TaskRecord.remark.ilike("%完工%"), ) # 最新有效备注（排除系统备注，按时间倒序取第一条） valid_remark_subq = ( select( TaskRecord.task_id, TaskRecord.remark.label("latest_valid_remark"), func.row_number().over( partition_by=TaskRecord.task_id, order_by=TaskRecord.created_at.desc(), ).label("rn"), ) .where( TaskRecord.remark.isnot(None), func.trim(TaskRecord.remark) != "", ~sys_remark, ) ).subquery("vr") # 记录总数（含系统备注） record_count_subq = ( select( TaskRecord.task_id, func.count(TaskRecord.id).label("record_count"), ) .group_by(TaskRecord.task_id) ).subquery("rc") stmt = ( select( Task.id, Task.task_name, Task.assignee_id, Task.status, Task.received_at, Task.created_at, Task.completed_at, Product.serial_number, Product.external_serial, Product.material_name, Product.spec_model, valid_remark_subq.c.latest_valid_remark, func.coalesce(record_count_subq.c.record_count, 0).label("record_count"), ) .join(Product, Task.product_id == Product.id) .outerjoin( valid_remark_subq, and_(valid_remark_subq.c.task_id == Task.id, valid_remark_subq.c.rn == 1), ) .outerjoin(record_count_subq, record_count_subq.c.task_id == Task.id) .where( Task.status.in_([TASK_STATUS_WIP, TASK_STATUS_PENDING, TASK_STATUS_COMPLETED]), Task.assignee_id.isnot(None), ) ) # 多维筛选（模糊/精确匹配） if assignee_id: stmt = stmt.where(Task.assignee_id == assignee_id) if spec_model: stmt = stmt.where(Product.spec_model.ilike(f"%{spec_model.strip()}%")) if product_sn: stmt = stmt.where(Product.serial_number.ilike(f"%{product_sn.strip()}%")) if task_name: stmt = stmt.where(Task.task_name.ilike(f"%{task_name.strip()}%")) # 时间交集：start = COALESCE(received_at, created_at)，end = COALESCE(completed_at, now) start_expr = func.coalesce(Task.received_at, Task.created_at) end_expr = func.coalesce(Task.completed_at, now) if until: stmt = stmt.where(start_expr <= until) if since: stmt = stmt.where(end_expr >= since) # 排序：先进行中（WIP/PENDING），后已完成（COMPLETED），状态内按开始时间降序 from sqlalchemy import case status_prio = case( (Task.status == TASK_STATUS_WIP, 0), (Task.status == TASK_STATUS_PENDING, 1), (Task.status == TASK_STATUS_COMPLETED, 2), else_=3, ) stmt = stmt.order_by( status_prio.asc(), func.coalesce(Task.received_at, Task.created_at).desc(), ) result = await db.execute(stmt) rows = result.all() def _to_bj(dt): if not dt: return None if dt.tzinfo is None: from datetime import timezone as dt_timezone dt = dt.replace(tzinfo=dt_timezone.utc).astimezone(BEIJING_TZ) else: dt = dt.astimezone(BEIJING_TZ) return dt raw_ids = list({r[2] for r in rows if r[2]}) name_map: dict[str, str] = {} if raw_ids: from app.services.mom_cache import get_display_names name_map = get_display_names(raw_ids) records: list[PersonHistoryRecord] = [] for row in rows: received_dt = _to_bj(row[4] or row[5]) # received_at or created_at completed_dt = _to_bj(row[6]) # completed_at（进行中为 None） if completed_dt: end_dt = completed_dt completed_str = completed_dt.strftime("%m-%d %H:%M") else: end_dt = now completed_str = None hours = round((end_dt - received_dt).total_seconds() / 3600, 1) if received_dt else 0.0 records.append(PersonHistoryRecord( task_id=str(row[0]), task_name=row[1] or "", assignee_id=row[2] or "", assignee_name=name_map.get(row[2] or "", row[2] or ""), product_sn=row[7] or "", external_serial=row[8] or None, material_name=row[9] or "", spec_model=row[10] or "", status=row[3] or "", received_at=received_dt.strftime("%m-%d %H:%M") if received_dt else None, completed_at=completed_str, duration_hours=hours, latest_valid_remark=row[11], record_count=row[12] or 0, )) return records # ============================================================ # 协同留言搜索（上帝视角 — 全厂） # ============================================================ async def search_product_messages( db: AsyncSession, keyword: str = "", skip: int = 0, limit: int = 50, ) -> ProductMessageList: """ 上帝视角 — 全厂所有产品的协同留言。 关联链: ProductMessage → Product → (material_name, serial_number) 搜索支持: SN码、物料名称、留言人 排序: created_at 倒序（最新在前） """ from app.models.message import ProductMessage from app.models.product import Product from app.core.time_utils import BEIJING_TZ # 基础查询 — 同时取出 16位追溯码 + 业务序列号 stmt = ( select(ProductMessage, Product.serial_number, Product.material_name, Product.external_serial) .join(Product, ProductMessage.product_id == Product.id) ) # 关键词搜索 if keyword and keyword.strip(): kw = f"%{keyword.strip()}%" stmt = stmt.where(or_( Product.serial_number.ilike(kw), Product.external_serial.ilike(kw), Product.material_name.ilike(kw), ProductMessage.operator_id.ilike(kw), ProductMessage.content.ilike(kw), )) # 总数 count_stmt = select(func.count()).select_from(stmt.subquery()) total = await db.scalar(count_stmt) or 0 # 分页 + 排序 stmt = stmt.order_by(ProductMessage.created_at.desc()).offset(skip).limit(limit) result = await db.execute(stmt) rows = result.all() # 收集 operator_id → 批量翻译中文姓名 raw_ids = list({row[0].operator_id for row in rows if row[0].operator_id}) name_map: dict[str, str] = {} if raw_ids: from app.services.mom_cache import get_display_names name_map = get_display_names(raw_ids) items: list[ProductMessageItem] = [] for msg, sn, mat_name, ext_sn in rows: t = msg.created_at if t: if t.tzinfo is None: # 旧数据（datetime.utcnow）：naive UTC → 转北京时间 from datetime import timezone as dt_timezone t = t.replace(tzinfo=dt_timezone.utc).astimezone(BEIJING_TZ) else: t = t.astimezone(BEIJING_TZ) time_str = t.isoformat() if t else "" items.append(ProductMessageItem( id=str(msg.id), content=msg.content, operator_name=name_map.get(msg.operator_id, msg.operator_id), product_sn=sn or "", external_serial=ext_sn or None, material_name=mat_name or "", created_at=time_str, )) return ProductMessageList(items=items, total=total)