[verified] Implement account valuation history for issue 12

jarca_ledger_mcp/application/imports_service.py

@@ -31,6 +31,13 @@ from .shared import (
     transaction_preview,
 )
 
+
+def _record_import_valuation_candidate(conn: Any, account_id: str, balance: Any, basis_date: str, source_hash: str) -> None:
+    if balance is None or not basis_date:
+        return
+    from .account_valuation_service import record_account_valuation_candidate
+    record_account_valuation_candidate(conn, account_id, basis_date, balance, "KRW", "import", source_hash, f"import-balance:{account_id}:{basis_date}:{source_hash}")
+
 def preview_ledger_csv_import(file_path: str, source: str = "generic", db_path: str | None = None) -> dict[str, Any]:
     path=Path(file_path).expanduser()
     if not path.exists():
@@ -211,10 +218,9 @@ def import_kb_bank_pdf(file_path: str, password: str, account_id: str = "", skip
                     raise
             conn.execute("UPDATE import_batches SET row_count=? WHERE id=?", (inserted, batch_id))
             if transactions:
-                conn.execute(
-                    "UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?",
-                    (transactions[-1].balance_after, ts, ts, linked_account_id),
-                )
+                conn.execute("UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?", (transactions[-1].balance_after, ts, ts, linked_account_id))
+            if transactions and transactions[-1].balance_after is not None:
+                _record_import_valuation_candidate(conn, linked_account_id, transactions[-1].balance_after, transactions[-1].transaction_date, source_hash)
     except ValueError as e:
         return fail("validation_error", str(e))
     except Exception as e:
@@ -677,10 +683,9 @@ def import_toss_bank_pdf(file_path: str, account_id: str = "", skip_duplicates:
                     raise
             conn.execute("UPDATE import_batches SET row_count=? WHERE id=?", (inserted, batch_id))
             if transactions:
-                conn.execute(
-                    "UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?",
-                    (transactions[-1].balance_after, ts, ts, linked_account_id),
-                )
+                conn.execute("UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?", (transactions[-1].balance_after, ts, ts, linked_account_id))
+            if transactions and transactions[-1].balance_after is not None:
+                _record_import_valuation_candidate(conn, linked_account_id, transactions[-1].balance_after, transactions[-1].transaction_date, source_hash)
     except ValueError as e:
         return fail("validation_error", str(e))
     except Exception as e:
@@ -902,10 +907,9 @@ def import_toss_securities_pdf(file_path: str, account_id: str = "", skip_duplic
                     raise
             conn.execute("UPDATE import_batches SET row_count=? WHERE id=?", (inserted, batch_id))
             if transactions:
-                conn.execute(
-                    "UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?",
-                    (transactions[-1].balance_after, ts, ts, linked_account_id),
-                )
+                conn.execute("UPDATE accounts SET last_balance=?, last_synced_at=?, updated_at=? WHERE id=?", (transactions[-1].balance_after, ts, ts, linked_account_id))
+            if transactions and transactions[-1].balance_after is not None:
+                _record_import_valuation_candidate(conn, linked_account_id, transactions[-1].balance_after, transactions[-1].transaction_date, source_hash)
     except ValueError as e:
         return fail("validation_error", str(e))
     except Exception as e: