mirror of
https://github.com/maxdorninger/MediaManager.git
synced 2026-02-20 07:59:50 -05:00
a39e0d204a
This PR enables the ruff rule for return type annotations (ANN), and adds the ty package for type checking.
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
|
|
import media_manager.database
|
|
from media_manager.config import MediaManagerConfig
|
|
from media_manager.movies.service import (
|
|
auto_download_all_approved_movie_requests,
|
|
import_all_movie_torrents,
|
|
update_all_movies_metadata,
|
|
)
|
|
from media_manager.tv.service import (
|
|
auto_download_all_approved_season_requests,
|
|
import_all_show_torrents,
|
|
update_all_non_ended_shows_metadata,
|
|
)
|
|
|
|
|
|
def setup_scheduler(config: MediaManagerConfig) -> BackgroundScheduler:
|
|
from media_manager.database import init_engine
|
|
|
|
init_engine(config.database)
|
|
jobstores = {"default": SQLAlchemyJobStore(engine=media_manager.database.engine)}
|
|
scheduler = BackgroundScheduler(jobstores=jobstores)
|
|
every_15_minutes_trigger = CronTrigger(minute="*/15", hour="*")
|
|
daily_trigger = CronTrigger(hour=0, minute=0, jitter=60 * 60 * 24 * 2)
|
|
weekly_trigger = CronTrigger(
|
|
day_of_week="mon", hour=0, minute=0, jitter=60 * 60 * 24 * 2
|
|
)
|
|
scheduler.add_job(
|
|
import_all_movie_torrents,
|
|
every_15_minutes_trigger,
|
|
id="import_all_movie_torrents",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.add_job(
|
|
import_all_show_torrents,
|
|
every_15_minutes_trigger,
|
|
id="import_all_show_torrents",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.add_job(
|
|
auto_download_all_approved_season_requests,
|
|
daily_trigger,
|
|
id="auto_download_all_approved_season_requests",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.add_job(
|
|
auto_download_all_approved_movie_requests,
|
|
daily_trigger,
|
|
id="auto_download_all_approved_movie_requests",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.add_job(
|
|
update_all_movies_metadata,
|
|
weekly_trigger,
|
|
id="update_all_movies_metadata",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.add_job(
|
|
update_all_non_ended_shows_metadata,
|
|
weekly_trigger,
|
|
id="update_all_non_ended_shows_metadata",
|
|
replace_existing=True,
|
|
)
|
|
scheduler.start()
|
|
return scheduler
|