From 6afd8d64abb4f6e912b5c1214081ee8d3e7a0a54 Mon Sep 17 00:00:00 2001 From: Aevann Date: Wed, 13 Sep 2023 23:55:59 +0300 Subject: [PATCH] add every_fri cron flag --- files/helpers/cron.py | 7 ++++--- files/routes/__init__.py | 2 +- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/files/helpers/cron.py b/files/helpers/cron.py index af3cf5df0..fe2b4bb1e 100644 --- a/files/helpers/cron.py +++ b/files/helpers/cron.py @@ -27,7 +27,7 @@ from files.cli import app, db_session, g CRON_CACHE_TIMEOUT = 172800 -def cron_fn(every_5m, every_1d): +def cron_fn(every_5m, every_1d, every_fri): with app.app_context(): g.db = db_session() g.v = None @@ -86,8 +86,9 @@ def cron_fn(every_5m, every_1d): @app.cli.command('cron', help='Run scheduled tasks.') @click.option('--every-5m', is_flag=True, help='Call every 5 minutes.') @click.option('--every-1d', is_flag=True, help='Call every 1 day.') -def cron(every_5m, every_1d): - cron_fn(every_5m, every_1d) +@click.option('--every-fri', is_flag=True, help='Call every Friday.') +def cron(every_5m, every_1d, every_fri): + cron_fn(every_5m, every_1d, every_fri) def _grant_one_year_badges(): one_year_ago = int(time.time()) - 364 * 86400 diff --git a/files/routes/__init__.py b/files/routes/__init__.py index 872292efb..7a652fc6d 100644 --- a/files/routes/__init__.py +++ b/files/routes/__init__.py @@ -56,5 +56,5 @@ if IS_LOCALHOST: if "cron" not in argv: from files.helpers.cron import cron_fn print('Starting cron tasks!', flush=True) - gevent.spawn(cron_fn, True, False) + gevent.spawn(cron_fn, True, False, False) print('Cron tasks Finished!', flush=True)