mirror of
https://github.com/joaovitoriasilva/endurain.git
synced 2026-01-09 15:57:59 -05:00
f6e06fb3e6
Implements optional session idle and absolute timeout logic, including new environment variables for configuration. Adds last_activity_at to sessions, enforces timeouts on token refresh, and introduces a scheduler job to clean up idle sessions. Also introduces progressive lockout for failed logins and updates documentation and examples accordingly.
108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
# from apscheduler.schedulers.background import BackgroundScheduler
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
|
|
import strava.activity_utils as strava_activity_utils
|
|
import strava.utils as strava_utils
|
|
|
|
import garmin.activity_utils as garmin_activity_utils
|
|
import garmin.health_utils as garmin_health_utils
|
|
|
|
import password_reset_tokens.utils as password_reset_tokens_utils
|
|
|
|
import sign_up_tokens.utils as sign_up_tokens_utils
|
|
|
|
import session.utils as session_utils
|
|
|
|
import auth.oauth_state.utils as oauth_state_utils
|
|
|
|
import core.logger as core_logger
|
|
|
|
# scheduler = BackgroundScheduler()
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
|
|
def start_scheduler():
|
|
if not scheduler.running:
|
|
# Start the scheduler
|
|
scheduler.start()
|
|
|
|
add_scheduler_job(
|
|
strava_utils.refresh_strava_tokens,
|
|
"interval",
|
|
60,
|
|
[True],
|
|
"refresh Strava user tokens every 60 minutes",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
strava_activity_utils.retrieve_strava_users_activities_for_days,
|
|
"interval",
|
|
60,
|
|
[1, True],
|
|
"retrieve last day Strava users activities",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
garmin_activity_utils.retrieve_garminconnect_users_activities_for_days,
|
|
"interval",
|
|
60,
|
|
[1],
|
|
"retrieve last day Garmin Connect users activities",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
garmin_health_utils.retrieve_garminconnect_users_health_for_days,
|
|
"interval",
|
|
240,
|
|
[1],
|
|
"retrieve last day Garmin Connect users health data",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
password_reset_tokens_utils.delete_invalid_tokens_from_db,
|
|
"interval",
|
|
60,
|
|
[],
|
|
"delete invalid password reset tokens from the database",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
sign_up_tokens_utils.delete_invalid_tokens_from_db,
|
|
"interval",
|
|
60,
|
|
[],
|
|
"delete invalid sign-up tokens from the database",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
oauth_state_utils.delete_expired_oauth_states_from_db,
|
|
"interval",
|
|
5,
|
|
[],
|
|
"delete expired OAuth states from the database",
|
|
)
|
|
|
|
add_scheduler_job(
|
|
session_utils.cleanup_idle_sessions,
|
|
"interval",
|
|
15,
|
|
[],
|
|
"delete expired sessions from the database",
|
|
)
|
|
|
|
|
|
def add_scheduler_job(func, interval, minutes, args, description):
|
|
try:
|
|
core_logger.print_to_log(
|
|
f"Added scheduler job to {description} every {minutes} minutes"
|
|
)
|
|
scheduler.add_job(func, interval, minutes=minutes, args=args)
|
|
except Exception as e:
|
|
core_logger.print_to_log(
|
|
f"Failed to add scheduler job to {description}: {str(e)}", "error"
|
|
)
|
|
|
|
|
|
def stop_scheduler():
|
|
scheduler.shutdown()
|