diff --git a/src/database.py b/src/database.py index 140660d..789aa29 100644 --- a/src/database.py +++ b/src/database.py @@ -358,9 +358,7 @@ class DatabaseManager: session = self.session try: sanitized_ip = sanitize_ip(ip) - ip_stats = ( - session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() - ) + ip_stats = session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() if not ip_stats: now = datetime.now() @@ -406,9 +404,7 @@ class DatabaseManager: session = self.session try: sanitized_ip = sanitize_ip(ip) - ip_stats = ( - session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() - ) + ip_stats = session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() if not ip_stats or ip_stats.ban_timestamp is None: return False @@ -445,9 +441,7 @@ class DatabaseManager: session = self.session try: sanitized_ip = sanitize_ip(ip) - ip_stats = ( - session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() - ) + ip_stats = session.query(IpStats).filter(IpStats.ip == sanitized_ip).first() if not ip_stats: return { diff --git a/src/migrations/runner.py b/src/migrations/runner.py index 0c5c67d..7a74267 100644 --- a/src/migrations/runner.py +++ b/src/migrations/runner.py @@ -59,9 +59,7 @@ def _migrate_ban_state_columns(cursor) -> List[str]: } for col_name, col_type in columns.items(): if not _column_exists(cursor, "ip_stats", col_name): - cursor.execute( - f"ALTER TABLE ip_stats ADD COLUMN {col_name} {col_type}" - ) + cursor.execute(f"ALTER TABLE ip_stats ADD COLUMN {col_name} {col_type}") added.append(col_name) return added diff --git a/src/tasks/analyze_ips.py b/src/tasks/analyze_ips.py index 81ad1d9..12ef5e4 100644 --- a/src/tasks/analyze_ips.py +++ b/src/tasks/analyze_ips.py @@ -98,7 +98,9 @@ def main(): ips_to_analyze = set(db_manager.get_ips_needing_reevaluation()) if not ips_to_analyze: - app_logger.debug("[Background Task] analyze-ips: No IPs need reevaluation, skipping") + app_logger.debug( + "[Background Task] analyze-ips: No IPs need reevaluation, skipping" + ) return for ip in ips_to_analyze: diff --git a/src/tracker.py b/src/tracker.py index bd0c0ee..c52cf24 100644 --- a/src/tracker.py +++ b/src/tracker.py @@ -435,4 +435,3 @@ class AccessTracker: stats["credential_attempts"] = self.db.get_credential_attempts(limit=50) return stats -