fix: 强化报工归属回填脚本

scripts/migrate_report_allocations.py

@@ -9,9 +9,9 @@ ROOT = Path(__file__).resolve().parents[1]
 sys.path.insert(0, str(ROOT))
 
 from app.database import SessionLocal, engine  # noqa: E402
-from app.models import ProductionReport, WorkSession  # noqa: E402
+from app.models import AttendancePoint, ProductionReport, WorkSession  # noqa: E402
 from app.services.report_allocations import refresh_report_allocations  # noqa: E402
-from app.services.work_schedule import get_work_schedule_config  # noqa: E402
+from app.services.work_schedule import DEFAULT_WORK_SCHEDULE_CONFIG, WorkScheduleConfig  # noqa: E402
 
 
 TABLE_NAME = "production_report_allocations"
@@ -378,33 +378,60 @@ def verify_schema(conn) -> None:
         raise RuntimeError(f"{TABLE_NAME} missing foreign keys: {', '.join(missing_foreign_keys)}")
 
 
-def _backfill() -> tuple[int, int]:
+def _read_work_schedule_config(db, attendance_point_name: str | None) -> WorkScheduleConfig:
+    point_name = str(attendance_point_name or "").strip()
+    if not point_name:
+        return DEFAULT_WORK_SCHEDULE_CONFIG
+
+    point = db.get(AttendancePoint, point_name)
+    if point is None:
+        return DEFAULT_WORK_SCHEDULE_CONFIG
+
+    return WorkScheduleConfig(
+        day_start=point.day_start or DEFAULT_WORK_SCHEDULE_CONFIG.day_start,
+        day_end=point.day_end or DEFAULT_WORK_SCHEDULE_CONFIG.day_end,
+        lunch_start=point.lunch_start or DEFAULT_WORK_SCHEDULE_CONFIG.lunch_start,
+        lunch_end=point.lunch_end or DEFAULT_WORK_SCHEDULE_CONFIG.lunch_end,
+        dinner_start=point.dinner_start or DEFAULT_WORK_SCHEDULE_CONFIG.dinner_start,
+        dinner_end=point.dinner_end or DEFAULT_WORK_SCHEDULE_CONFIG.dinner_end,
+        overtime_start=point.overtime_start or DEFAULT_WORK_SCHEDULE_CONFIG.overtime_start,
+        overtime_end=point.overtime_end or DEFAULT_WORK_SCHEDULE_CONFIG.overtime_end,
+        night_start=point.night_start or DEFAULT_WORK_SCHEDULE_CONFIG.night_start,
+        night_end=point.night_end or DEFAULT_WORK_SCHEDULE_CONFIG.night_end,
+    )
+
+
+def _backfill(batch_size: int = 200) -> tuple[int, int]:
     processed = 0
     skipped = 0
+    last_id = 0
 
+    with SessionLocal() as db:
+        while True:
             query = (
                 select(ProductionReport)
+                .where(ProductionReport.id > last_id)
                 .options(
                     selectinload(ProductionReport.items),
                     selectinload(ProductionReport.session).selectinload(WorkSession.devices),
                 )
                 .order_by(ProductionReport.id.asc())
+                .limit(batch_size)
             )
-
-    with SessionLocal() as db:
             reports = db.scalars(query).all()
+            if not reports:
+                break
+
             for report in reports:
                 if report is None:
                     skipped += 1
                     continue
 
-            schedule = get_work_schedule_config(db, report.attendance_point_name)
+                last_id = report.id
+                schedule = _read_work_schedule_config(db, report.attendance_point_name)
                 refresh_report_allocations(db, report, schedule=schedule, commit=False)
                 processed += 1
 
-            if processed % 200 == 0:
-                db.commit()
-
             db.commit()
 
     return processed, skipped