fix: Improve stale IP flagging logic to support forced rescan
This commit is contained in:
@@ -12,21 +12,34 @@ TASK_CONFIG = {
|
|||||||
"run_when_loaded": True,
|
"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():
|
def main():
|
||||||
|
global FORCE_IP_RESCAN
|
||||||
|
|
||||||
app_logger = get_app_logger()
|
app_logger = get_app_logger()
|
||||||
db = get_database()
|
db = get_database()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
count = db.flag_stale_ips_for_reevaluation()
|
if FORCE_IP_RESCAN:
|
||||||
if count > 0:
|
count = db.flag_all_ips_for_reevaluation()
|
||||||
|
FORCE_IP_RESCAN = False
|
||||||
app_logger.info(
|
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:
|
else:
|
||||||
app_logger.debug(
|
count = db.flag_stale_ips_for_reevaluation()
|
||||||
"[Background Task] flag-stale-ips: No stale IPs found to flag"
|
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:
|
except Exception as e:
|
||||||
app_logger.error(
|
app_logger.error(
|
||||||
f"[Background Task] flag-stale-ips: Error flagging stale IPs: {e}"
|
f"[Background Task] flag-stale-ips: Error flagging stale IPs: {e}"
|
||||||
|
|||||||
Reference in New Issue
Block a user