This commit is contained in:
2026-03-12 12:32:29 +08:00
parent c0f9ddabbf
commit 470446fa6f
18 changed files with 591 additions and 142 deletions

View File

@@ -16,31 +16,42 @@ def parse_extracted_fields(
items = raw if isinstance(raw, list) else [raw]
records: list[TransactionRecord] = []
for item in items:
for idx, item in enumerate(items):
if not item or not item.get("amount"):
continue
raw_trade_time = item.get("trade_time")
try:
trade_time = datetime.fromisoformat(item["trade_time"])
except (ValueError, KeyError):
if isinstance(raw_trade_time, datetime):
trade_time = raw_trade_time
elif isinstance(raw_trade_time, str):
trade_time = datetime.fromisoformat(raw_trade_time)
else:
raise TypeError("trade_time is not str/datetime")
except (ValueError, KeyError, TypeError):
trade_time = datetime.now()
direction_str = item.get("direction", "out")
direction = Direction.in_ if direction_str == "in" else Direction.out
try:
amount = float(item.get("amount", 0))
confidence = float(item.get("confidence", 0.5))
except Exception:
raise
record = TransactionRecord(
case_id=case_id,
evidence_image_id=image_id,
source_app=source_app,
trade_time=trade_time,
amount=float(item.get("amount", 0)),
amount=amount,
direction=direction,
counterparty_name=str(item.get("counterparty_name", "")),
counterparty_account=str(item.get("counterparty_account", "")),
self_account_tail_no=str(item.get("self_account_tail_no", "")),
order_no=str(item.get("order_no", "")),
remark=str(item.get("remark", "")),
confidence=float(item.get("confidence", 0.5)),
confidence=confidence,
)
records.append(record)