From 542d10ebe08fb3c8feae6ae931b8ff2516c3a05d Mon Sep 17 00:00:00 2001 From: tcely Date: Tue, 24 Mar 2026 15:38:59 -0400 Subject: [PATCH] Allow a different tasks directory for `sqlite_tasks` --- tubesync/common/huey.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tubesync/common/huey.py b/tubesync/common/huey.py index 85028c19..5e768ccd 100644 --- a/tubesync/common/huey.py +++ b/tubesync/common/huey.py @@ -8,6 +8,7 @@ from huey import ( ) from huey.api import TaskLock from huey.storage import SqliteStorage as huey_SqliteStorage +from pathlib import Path from .timestamp import datetime_to_timestamp, timestamp_to_datetime @@ -214,7 +215,7 @@ def h_q_reset_tasks(q, /, *, maint_func=None): return maint_result -def sqlite_tasks(key, /, prefix=None, thread=None, workers=None): +def sqlite_tasks(key, /, prefix=None, thread=None, workers=None, *, tasks_dir=None): name_fmt = 'huey_{}' if prefix: name_fmt = f'huey_{prefix}_' + '{}' @@ -239,7 +240,7 @@ def sqlite_tasks(key, /, prefix=None, thread=None, workers=None): utc=True, compression=True, connection=dict( - filename=f'/config/tasks/{name}.db', + filename=str(Path(tasks_dir or '/config/tasks') / f'{name}.db'), fsync=True, isolation_level='IMMEDIATE', # _create_connection sets this to None strict_fifo=True,