You've already forked snikket-web-portal
20abe4b903
It was found during testing that some user agents cache aggressively even between switches of the UI language. To properly indicate that the pages actually depend on that information, we add the correct Vary header. Fixes #106.
109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
import base64
|
||
import itertools
|
||
import math
|
||
import secrets
|
||
import typing
|
||
|
||
import quart.flask_patch # noqa:F401
|
||
from quart import (
|
||
current_app,
|
||
request,
|
||
g,
|
||
)
|
||
|
||
import flask_babel
|
||
import flask_wtf
|
||
from flask_babel import _
|
||
|
||
from . import prosodyclient
|
||
|
||
|
||
client = prosodyclient.ProsodyClient()
|
||
client.default_login_redirect = "main.login"
|
||
|
||
babel = flask_babel.Babel()
|
||
|
||
|
||
BYTE_UNIT_SCALE_MAP = [
|
||
"B",
|
||
"kiB",
|
||
"MiB",
|
||
"GiB",
|
||
"TiB",
|
||
]
|
||
|
||
|
||
@babel.localeselector # type:ignore
|
||
def selected_locale() -> str:
|
||
# Needs mypy ignore because this is a free-for-all object and has no
|
||
# publicly known attributes.
|
||
g.language_header_accessed = True # type: ignore
|
||
selected = request.accept_languages.best_match(
|
||
current_app.config['LANGUAGES']
|
||
) or current_app.config['LANGUAGES'][0]
|
||
return selected
|
||
|
||
|
||
def flatten(a: typing.Iterable, levels: int = 1) -> typing.Iterable:
|
||
for i in range(levels):
|
||
a = itertools.chain(*a)
|
||
return a
|
||
|
||
|
||
def circle_name(c: typing.Any) -> str:
|
||
if c.id_ == "default" and c.name == "default":
|
||
return _("Main")
|
||
return c.name
|
||
|
||
|
||
def format_bytes(n: float) -> str:
|
||
try:
|
||
scale = max(math.floor(math.log(n, 1024)), 0)
|
||
except ValueError:
|
||
scale = 0
|
||
try:
|
||
unit = BYTE_UNIT_SCALE_MAP[scale]
|
||
factor = 1024**scale
|
||
except IndexError:
|
||
unit = "TiB"
|
||
factor = 1024**4
|
||
if factor > 1:
|
||
return "{:.1f} {}".format(n / factor, unit)
|
||
return "{} {}".format(n, unit)
|
||
|
||
|
||
def add_vary_language_header(resp: quart.Response) -> quart.Response:
|
||
if getattr(g, "language_header_accessed", False):
|
||
resp.vary.add("Accept-Language")
|
||
return resp
|
||
|
||
|
||
def init_templating(app: quart.Quart) -> None:
|
||
app.template_filter("repr")(repr)
|
||
app.template_filter("format_datetime")(flask_babel.format_datetime)
|
||
app.template_filter("format_date")(flask_babel.format_date)
|
||
app.template_filter("format_time")(flask_babel.format_time)
|
||
app.template_filter("format_timedelta")(flask_babel.format_timedelta)
|
||
app.template_filter("format_percent")(flask_babel.format_percent)
|
||
app.template_filter("format_bytes")(format_bytes)
|
||
app.template_filter("flatten")(flatten)
|
||
app.template_filter("circle_name")(circle_name)
|
||
app.after_request(add_vary_language_header)
|
||
|
||
|
||
def generate_error_id() -> str:
|
||
return base64.b32encode(secrets.token_bytes(8)).decode(
|
||
"ascii"
|
||
).rstrip("=")
|
||
|
||
|
||
class BaseForm(flask_wtf.FlaskForm): # type:ignore
|
||
def __init__(self, *args: typing.Any, **kwargs: typing.Any):
|
||
meta = kwargs["meta"] = dict(kwargs.get("meta", {}))
|
||
if "locales" not in meta:
|
||
locale = flask_babel.get_locale()
|
||
if locale:
|
||
meta["locales"] = [str(locale)]
|
||
|
||
super().__init__(*args, **kwargs)
|