Update models.py

models.py

@@ -175,12 +175,23 @@ def initialize_database():
             FOREIGN KEY (heir_id) REFERENCES heirs(id)
         );
 
+        CREATE TABLE IF NOT EXISTS login_attempts (
+            id TEXT PRIMARY KEY,
+            username TEXT NOT NULL,
+            attempt_time TEXT NOT NULL,
+            ip_address TEXT,
+            success BOOLEAN DEFAULT FALSE,
+            FOREIGN KEY (username) REFERENCES users(username)
+        );
+
         CREATE INDEX IF NOT EXISTS idx_users_username ON users(username);
         CREATE INDEX IF NOT EXISTS idx_users_email ON users(email);
         CREATE INDEX IF NOT EXISTS idx_social_accounts_user ON social_accounts(user_id);
         CREATE INDEX IF NOT EXISTS idx_heirs_user ON heirs(user_id);
         CREATE INDEX IF NOT EXISTS idx_audit_logs_user ON audit_logs(user_id);
         CREATE INDEX IF NOT EXISTS idx_activity_logs_account ON activity_logs(account_id);
+        CREATE INDEX IF NOT EXISTS idx_login_attempts_username ON login_attempts(username);
+        CREATE INDEX IF NOT EXISTS idx_login_attempts_time ON login_attempts(attempt_time);
     ''')
     
     conn.commit()