diff --git a/src/tasks/flag_stale_ips.py b/src/tasks/flag_stale_ips.py index 3bf6f82..0428e15 100644 --- a/src/tasks/flag_stale_ips.py +++ b/src/tasks/flag_stale_ips.py @@ -12,21 +12,34 @@ TASK_CONFIG = { "run_when_loaded": True, } +# Set to True to force all IPs to be flagged for reevaluation on next run. +# Resets to False automatically after execution. +FORCE_IP_RESCAN = False + def main(): + global FORCE_IP_RESCAN + app_logger = get_app_logger() db = get_database() try: - count = db.flag_stale_ips_for_reevaluation() - if count > 0: + if FORCE_IP_RESCAN: + count = db.flag_all_ips_for_reevaluation() + FORCE_IP_RESCAN = False app_logger.info( - f"[Background Task] flag-stale-ips: Flagged {count} stale IPs for reevaluation" + f"[Background Task] flag-stale-ips: FORCE RESCAN - Flagged {count} IPs for reevaluation" ) else: - app_logger.debug( - "[Background Task] flag-stale-ips: No stale IPs found to flag" - ) + count = db.flag_stale_ips_for_reevaluation() + if count > 0: + app_logger.info( + f"[Background Task] flag-stale-ips: Flagged {count} stale IPs for reevaluation" + ) + else: + app_logger.debug( + "[Background Task] flag-stale-ips: No stale IPs found to flag" + ) except Exception as e: app_logger.error( f"[Background Task] flag-stale-ips: Error flagging stale IPs: {e}"