selfprivacy-rest-api/selfprivacy_api/utils/huey.py

25 lines
524 B
Python
Raw Normal View History

"""MiniHuey singleton."""
2024-01-19 14:06:07 +00:00
from os import environ
from huey import RedisHuey
2024-01-19 14:06:07 +00:00
from selfprivacy_api.utils.redis_pool import RedisPool
2024-01-19 14:06:07 +00:00
HUEY_DATABASE_NUMBER = 10
2024-01-19 14:06:07 +00:00
def immediate() -> bool:
if environ.get("HUEY_QUEUES_FOR_TESTS"):
return False
if environ.get("TEST_MODE"):
return True
return False
2024-01-19 14:06:07 +00:00
# Singleton instance containing the huey database.
huey = RedisHuey(
"selfprivacy-api",
2024-01-19 14:06:07 +00:00
url=RedisPool.connection_url(dbnumber=HUEY_DATABASE_NUMBER),
immediate=immediate(),
utc=True,
)