06/03 Optimize app

app/models/audit_log.py

@@ -0,0 +1,15 @@
+from app.extensions import db
+from datetime import datetime
+
+
+class AuditLog(db.Model):
+    __tablename__ = 'audit_logs'
+
+    id          = db.Column(db.Integer, primary_key=True)
+    timestamp   = db.Column(db.DateTime, default=datetime.utcnow, index=True)
+    action      = db.Column(db.String(64), nullable=False, index=True)
+    description = db.Column(db.String(255), nullable=True)
+    ip_address  = db.Column(db.String(45), nullable=True)   # IPv4 or IPv6
+
+    def __repr__(self):
+        return f'<AuditLog {self.action} @ {self.timestamp}>'

app/models/schwab_connection.py

@@ -1,4 +1,5 @@
 from app.extensions import db
+from app.utils.crypto import EncryptedText
 from datetime import datetime
 
 
@@ -7,8 +8,8 @@ class SchwabConnection(db.Model):
     __tablename__ = 'schwab_connections'
 
     id               = db.Column(db.Integer, primary_key=True)
-    access_token              = db.Column(db.Text, nullable=False)
-    refresh_token             = db.Column(db.Text, nullable=False)
+    access_token              = db.Column(EncryptedText, nullable=False)
+    refresh_token             = db.Column(EncryptedText, nullable=False)
     token_expires_at          = db.Column(db.DateTime, nullable=False)   # access token expiry
     refresh_token_expires_at  = db.Column(db.DateTime, nullable=True)    # refresh token expiry (7 days)
     is_active                 = db.Column(db.Boolean, default=True)

app/models/teller_enrollment.py

@@ -1,4 +1,5 @@
 from app.extensions import db
+from app.utils.crypto import EncryptedText
 from datetime import datetime
 
 
@@ -11,7 +12,7 @@ class TellerEnrollment(db.Model):
 
     id = db.Column(db.Integer, primary_key=True)
     enrollment_id = db.Column(db.String(64), unique=True, nullable=False, index=True)
-    access_token = db.Column(db.String(128), nullable=False)
+    access_token = db.Column(EncryptedText, nullable=False)
     institution_name = db.Column(db.String(100), nullable=True)
     user_id = db.Column(db.String(64), nullable=True)  # Teller user ID
     is_active = db.Column(db.Boolean, default=True)