48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import threading
|
|
|
|
from modeltranslation import settings
|
|
|
|
from ._typing import AutoPopulate
|
|
|
|
|
|
class ModelTranslationThreadLocal(threading.local):
|
|
"""Holds thread-local data for modeltranslation."""
|
|
|
|
auto_populate: AutoPopulate | None = None
|
|
enable_fallbacks: bool | None = None
|
|
|
|
|
|
_mt_thread_context = ModelTranslationThreadLocal()
|
|
|
|
|
|
def set_auto_populate(value: AutoPopulate | None) -> None:
|
|
"""Set the auto_populate for the current thread."""
|
|
_mt_thread_context.auto_populate = value
|
|
|
|
|
|
def set_enable_fallbacks(value: bool | None) -> None:
|
|
"""Set the enable_fallbacks for the current thread."""
|
|
_mt_thread_context.enable_fallbacks = value
|
|
|
|
|
|
def auto_populate_mode() -> AutoPopulate:
|
|
"""Return the auto_populate mode for the current thread."""
|
|
auto_populate = _mt_thread_context.auto_populate
|
|
|
|
if auto_populate is not None:
|
|
return auto_populate
|
|
|
|
return settings.AUTO_POPULATE
|
|
|
|
|
|
def fallbacks_enabled() -> bool:
|
|
"""Return whether fallbacks are enabled for the current thread."""
|
|
enable_fallbacks = _mt_thread_context.enable_fallbacks
|
|
|
|
if enable_fallbacks is not None:
|
|
return enable_fallbacks
|
|
|
|
return settings.ENABLE_FALLBACKS
|