product

db/models.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python3
+"""
+Data models for ACH file processing.
+Represents database records and transactions.
+"""
+
+from dataclasses import dataclass, asdict
+from datetime import date, datetime
+from decimal import Decimal
+from typing import Optional
+
+
+@dataclass
+class TransactionRecord:
+    """Represents a transaction record for ach_api_log table."""
+    narration: str
+    status: str
+    bankcode: str
+    jrnl_id: str
+    tran_date: date
+    cbs_acct: str
+    tran_amt: Decimal
+    txnind: str
+
+    def to_dict(self):
+        """Convert to dictionary for database insertion."""
+        return {
+            'narration': self.narration,
+            'status': self.status,
+            'bankcode': self.bankcode,
+            'jrnl_id': self.jrnl_id,
+            'tran_date': self.tran_date,
+            'cbs_acct': self.cbs_acct,
+            'tran_amt': self.tran_amt,
+            'TXNIND': self.txnind,
+        }
+
+
+@dataclass
+class ProcessedFile:
+    """Represents a processed file record for ach_processed_files table."""
+    filename: str
+    bankcode: str
+    file_path: str
+    transaction_count: int
+    status: str = 'SUCCESS'
+    error_message: Optional[str] = None
+    processed_at: Optional[datetime] = None
+
+    def to_dict(self):
+        """Convert to dictionary for database insertion."""
+        return {
+            'filename': self.filename,
+            'bankcode': self.bankcode,
+            'file_path': self.file_path,
+            'transaction_count': self.transaction_count,
+            'status': self.status,
+            'error_message': self.error_message,
+            'processed_at': self.processed_at or datetime.now(),
+        }