fix ocr
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user