Boostrap stream customization (API may change)

This commit is contained in:
Thomas Sileo 2022-11-22 20:30:35 +01:00
parent bd065446bf
commit 2cc4eda143
3 changed files with 73 additions and 4 deletions

View file

@ -16,6 +16,8 @@ from loguru import logger
from mistletoe import markdown # type: ignore
from app.customization import _CUSTOM_ROUTES
from app.customization import _StreamVisibilityCallback
from app.customization import default_stream_visibility_callback
from app.utils.emoji import _load_emojis
from app.utils.version import get_version_commit
@ -262,3 +264,14 @@ def verify_csrf_token(
def hmac_sha256() -> hmac.HMAC:
return hmac.new(CONFIG.secret.encode(), digestmod=hashlib.sha256)
stream_visibility_callback: _StreamVisibilityCallback
try:
from data.stream import ( # type: ignore # noqa: F401, E501
custom_stream_visibility_callback,
)
stream_visibility_callback = custom_stream_visibility_callback
except ImportError:
stream_visibility_callback = default_stream_visibility_callback