Initial commit: site monorepo with API, web, and infra.
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
влад
@@ -0,0 +1,151 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from datetime import UTC, datetime
|
||||
from typing import Any
|
||||
|
||||
from sqlalchemy import select
|
||||
|
||||
from app.core.database import session_scope
|
||||
from app.modules.zootech.report_models import ZootechFeedAlert, ZootechLoadingReport, ZootechUnloadingReport
|
||||
|
||||
REPORT_TABLE_MAP = {
|
||||
"loading_report": ZootechLoadingReport,
|
||||
"unloading_report": ZootechUnloadingReport,
|
||||
"feed_alert": ZootechFeedAlert,
|
||||
}
|
||||
|
||||
|
||||
def _write_payload_json(row: Any, extra: dict[str, Any]) -> None:
|
||||
if not extra:
|
||||
return
|
||||
row.payload_json = json.dumps(extra, ensure_ascii=False)
|
||||
|
||||
|
||||
def repair_report_payloads_from_event_log(enterprise_id: str) -> int:
|
||||
"""Re-apply latest sync event payload per report row (one-time repair helper)."""
|
||||
from app.modules.sync.models import SyncEventLog
|
||||
|
||||
repaired = 0
|
||||
events: list[tuple[str, str, str, int, str, str | None]] = []
|
||||
with session_scope() as db:
|
||||
rows = list(
|
||||
db.scalars(
|
||||
select(SyncEventLog)
|
||||
.where(
|
||||
SyncEventLog.enterprise_id == enterprise_id,
|
||||
SyncEventLog.table_name.in_(("loading_report", "unloading_report", "feed_alert")),
|
||||
)
|
||||
.order_by(SyncEventLog.record_id.asc(), SyncEventLog.received_at.desc())
|
||||
)
|
||||
)
|
||||
latest_by_record: dict[tuple[str, str], SyncEventLog] = {}
|
||||
for row in rows:
|
||||
key = (row.table_name, row.record_id)
|
||||
if key not in latest_by_record:
|
||||
latest_by_record[key] = row
|
||||
for (table_name, record_id), event in latest_by_record.items():
|
||||
events.append(
|
||||
(
|
||||
table_name,
|
||||
record_id,
|
||||
event.payload_json or "{}",
|
||||
int(event.version or 1),
|
||||
str(event.content_hash or ""),
|
||||
event.origin_site_id,
|
||||
)
|
||||
)
|
||||
|
||||
for table_name, record_id, payload_json, version, content_hash, origin_site_id in events:
|
||||
try:
|
||||
payload = json.loads(payload_json)
|
||||
except json.JSONDecodeError:
|
||||
continue
|
||||
if not isinstance(payload, dict):
|
||||
continue
|
||||
apply_report_change(
|
||||
enterprise_id,
|
||||
table_name,
|
||||
record_id,
|
||||
"upsert",
|
||||
payload,
|
||||
version,
|
||||
content_hash,
|
||||
farm_hub_id=origin_site_id,
|
||||
)
|
||||
repaired += 1
|
||||
return repaired
|
||||
|
||||
|
||||
def apply_report_change(
|
||||
enterprise_id: str,
|
||||
table_name: str,
|
||||
record_id: str,
|
||||
action: str,
|
||||
payload: dict[str, Any],
|
||||
version: int,
|
||||
content_hash: str,
|
||||
farm_hub_id: str | None = None,
|
||||
) -> None:
|
||||
model = REPORT_TABLE_MAP.get(table_name)
|
||||
if not model:
|
||||
return
|
||||
with session_scope() as db:
|
||||
row = db.scalar(
|
||||
select(model).where(model.enterprise_id == enterprise_id, model.id == record_id)
|
||||
)
|
||||
if action == "delete":
|
||||
if row:
|
||||
row.is_deleted = True
|
||||
row.version = version
|
||||
row.content_hash = content_hash
|
||||
row.updated_at = datetime.now(UTC)
|
||||
return
|
||||
data = dict(payload)
|
||||
data["id"] = record_id
|
||||
data["enterprise_id"] = enterprise_id
|
||||
data["farm_hub_id"] = farm_hub_id or data.get("farm_hub_id")
|
||||
data["version"] = version
|
||||
data["content_hash"] = content_hash
|
||||
data["is_deleted"] = False
|
||||
if table_name == "feed_alert":
|
||||
if data.get("event_type") and not data.get("alert_type"):
|
||||
data["alert_type"] = str(data["event_type"])
|
||||
if data.get("detail") and not data.get("message"):
|
||||
data["message"] = str(data["detail"])
|
||||
if row:
|
||||
allowed = {c.key for c in model.__table__.columns}
|
||||
extra = {k: v for k, v in data.items() if k not in allowed}
|
||||
for key, value in data.items():
|
||||
if key in allowed and key not in ("enterprise_id", "created_at"):
|
||||
setattr(row, key, value)
|
||||
if extra and "payload_json" in allowed:
|
||||
_write_payload_json(row, extra)
|
||||
# #region agent log
|
||||
try:
|
||||
import pathlib
|
||||
_log_path = pathlib.Path("/Users/vlad/Documents/wesp new (1)/.cursor/debug-785e22.log")
|
||||
_log_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with _log_path.open("a", encoding="utf-8") as _lf:
|
||||
_lf.write(json.dumps({"sessionId":"785e22","hypothesisId":"D","location":"report_apply.py:update","message":"report payload update","data":{"table":table_name,"record_id":record_id,"extra_keys":sorted(extra.keys()),"components_len":len(extra.get("components") or []) if isinstance(extra.get("components"), list) else None,"groups_len":len(extra.get("unloading_groups") or []) if isinstance(extra.get("unloading_groups"), list) else None},"timestamp":int(datetime.now(UTC).timestamp()*1000)}, ensure_ascii=False) + "\n")
|
||||
except Exception:
|
||||
pass
|
||||
# #endregion
|
||||
row.updated_at = datetime.now(UTC)
|
||||
else:
|
||||
allowed = {c.key for c in model.__table__.columns}
|
||||
filtered = {k: v for k, v in data.items() if k in allowed}
|
||||
extra = {k: v for k, v in data.items() if k not in allowed}
|
||||
if extra and "payload_json" in allowed:
|
||||
filtered["payload_json"] = json.dumps(extra, ensure_ascii=False)
|
||||
db.add(model(**filtered))
|
||||
# #region agent log
|
||||
try:
|
||||
import pathlib
|
||||
_log_path = pathlib.Path("/Users/vlad/Documents/wesp new (1)/.cursor/debug-785e22.log")
|
||||
_log_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with _log_path.open("a", encoding="utf-8") as _lf:
|
||||
_lf.write(json.dumps({"sessionId":"785e22","hypothesisId":"D","location":"report_apply.py:insert","message":"report payload insert","data":{"table":table_name,"record_id":record_id,"components_len":len(extra.get("components") or []) if isinstance(extra.get("components"), list) else None},"timestamp":int(datetime.now(UTC).timestamp()*1000)}, ensure_ascii=False) + "\n")
|
||||
except Exception:
|
||||
pass
|
||||
# #endregion
|
||||
Reference in New Issue
Block a user