if platform.system() == "Darwin":
import multiprocessing
try:
multiprocessing.set_start_method("fork")
except RuntimeError:
# sometimes it has already been set by
# other libs
pass
redis_host = None
if settings.cache_url:
redis_host = settings.cache_url.host
huey = PriorityRedisHuey("opennem.scheduler", host=redis_host)
# export tasks
@huey.periodic_task(crontab(minute="*/5"), priority=90)
@huey.lock_task("schedule_live_tasks")
def schedule_live_tasks() -> None:
if settings.workers_run:
export_power(priority=PriorityType.live)
export_flows()
@huey.periodic_task(crontab(minute="*/30"), priority=90)
@huey.lock_task("schedule_custom_tasks")
def schedule_custom_tasks() -> None:
if settings.workers_run:
from huey import PriorityRedisHuey
huey = PriorityRedisHuey('aurora', host='redis')