mirror of
https://github.com/tubearchivist/tubearchivist-frontend.git
synced 2024-11-04 19:30:13 +00:00
ScheduleBuilder config class for celery beat
This commit is contained in:
parent
012e28d4ef
commit
b2861e0369
@ -8,6 +8,7 @@ Functionality:
|
|||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
from celery.schedules import crontab
|
||||||
from home.src.helper import RedisArchivist
|
from home.src.helper import RedisArchivist
|
||||||
|
|
||||||
|
|
||||||
@ -150,3 +151,58 @@ class AppConfig:
|
|||||||
|
|
||||||
if needs_update:
|
if needs_update:
|
||||||
RedisArchivist().set_message("config", redis_config, expire=False)
|
RedisArchivist().set_message("config", redis_config, expire=False)
|
||||||
|
|
||||||
|
|
||||||
|
class ScheduleBuilder:
|
||||||
|
"""build schedule dicts for beat"""
|
||||||
|
|
||||||
|
SCHEDULES = [
|
||||||
|
"update_subscribed",
|
||||||
|
"download_pending",
|
||||||
|
"check_reindex",
|
||||||
|
"thumbnail_check",
|
||||||
|
"run_backup",
|
||||||
|
]
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.config = AppConfig().config
|
||||||
|
|
||||||
|
def update_schedule_conf(self, form_post):
|
||||||
|
"""process form post"""
|
||||||
|
print("processing form, restart container for changes to take effect")
|
||||||
|
redis_config = self.config
|
||||||
|
for key, value in form_post.items():
|
||||||
|
if key in self.SCHEDULES and value[0]:
|
||||||
|
print(f"change schedule for {key} to {value[0]}")
|
||||||
|
if value[0] == "0":
|
||||||
|
to_write = False
|
||||||
|
else:
|
||||||
|
keys = ["minute", "hour", "day_of_week"]
|
||||||
|
to_write = dict(zip(keys, value[0].split()))
|
||||||
|
redis_config["scheduler"][key] = to_write
|
||||||
|
RedisArchivist().set_message("config", redis_config, expire=False)
|
||||||
|
|
||||||
|
def build_schedule(self):
|
||||||
|
"""build schedule dict as expected by app.conf.beat_schedule"""
|
||||||
|
schedule_dict = {}
|
||||||
|
|
||||||
|
for schedule_item in self.SCHEDULES:
|
||||||
|
item_conf = self.config["scheduler"][schedule_item]
|
||||||
|
if not item_conf:
|
||||||
|
continue
|
||||||
|
|
||||||
|
minute = item_conf["minute"]
|
||||||
|
hour = item_conf["hour"]
|
||||||
|
day_of_week = item_conf["day_of_week"]
|
||||||
|
schedule_name = f"schedule_{schedule_item}"
|
||||||
|
to_add = {
|
||||||
|
schedule_name: {
|
||||||
|
"task": schedule_item,
|
||||||
|
"schedule": crontab(
|
||||||
|
minute=minute, hour=hour, day_of_week=day_of_week
|
||||||
|
),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
schedule_dict.update(to_add)
|
||||||
|
|
||||||
|
return schedule_dict
|
||||||
|
@ -21,12 +21,13 @@ from home.forms import (
|
|||||||
ChannelSearchForm,
|
ChannelSearchForm,
|
||||||
CustomAuthForm,
|
CustomAuthForm,
|
||||||
PlaylistSearchForm,
|
PlaylistSearchForm,
|
||||||
|
SchedulerSettingsForm,
|
||||||
SubscribeToChannelForm,
|
SubscribeToChannelForm,
|
||||||
SubscribeToPlaylistForm,
|
SubscribeToPlaylistForm,
|
||||||
UserSettingsForm,
|
UserSettingsForm,
|
||||||
VideoSearchForm,
|
VideoSearchForm,
|
||||||
)
|
)
|
||||||
from home.src.config import AppConfig
|
from home.src.config import AppConfig, ScheduleBuilder
|
||||||
from home.src.frontend import PostData
|
from home.src.frontend import PostData
|
||||||
from home.src.helper import RedisArchivist, UrlListParser
|
from home.src.helper import RedisArchivist, UrlListParser
|
||||||
from home.src.index import YoutubePlaylist
|
from home.src.index import YoutubePlaylist
|
||||||
@ -889,6 +890,7 @@ class SettingsView(View):
|
|||||||
|
|
||||||
user_form = UserSettingsForm()
|
user_form = UserSettingsForm()
|
||||||
app_form = ApplicationSettingsForm()
|
app_form = ApplicationSettingsForm()
|
||||||
|
scheduler_form = SchedulerSettingsForm()
|
||||||
|
|
||||||
context = {
|
context = {
|
||||||
"title": "Settings",
|
"title": "Settings",
|
||||||
@ -896,6 +898,7 @@ class SettingsView(View):
|
|||||||
"colors": colors,
|
"colors": colors,
|
||||||
"user_form": user_form,
|
"user_form": user_form,
|
||||||
"app_form": app_form,
|
"app_form": app_form,
|
||||||
|
"scheduler_form": scheduler_form,
|
||||||
}
|
}
|
||||||
|
|
||||||
return render(request, "home/settings.html", context)
|
return render(request, "home/settings.html", context)
|
||||||
@ -916,6 +919,10 @@ class SettingsView(View):
|
|||||||
elif "user-settings" in form_post:
|
elif "user-settings" in form_post:
|
||||||
del form_post["user-settings"]
|
del form_post["user-settings"]
|
||||||
config_handler.set_user_config(form_post, request.user.id)
|
config_handler.set_user_config(form_post, request.user.id)
|
||||||
|
elif "scheduler-settings" in form_post:
|
||||||
|
del form_post["scheduler-settings"]
|
||||||
|
print(form_post)
|
||||||
|
ScheduleBuilder().update_schedule_conf(form_post)
|
||||||
|
|
||||||
return redirect("settings", permanent=True)
|
return redirect("settings", permanent=True)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user