You've already forked snikket-web-portal
b4e6ee8943
Previously, that would raise a ValueError (math domain error), because log(0) is undefined.
98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
import base64
|
||
import itertools
|
||
import math
|
||
import secrets
|
||
import typing
|
||
|
||
import quart.flask_patch # noqa:F401
|
||
from quart import (
|
||
current_app,
|
||
request,
|
||
)
|
||
|
||
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:
|
||
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 ValueError:
|
||
unit = "TiB"
|
||
factor = 1024**4
|
||
if factor > 1:
|
||
return "{:.1f} {}".format(n / factor, unit)
|
||
return "{} {}".format(n, unit)
|
||
|
||
|
||
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)
|
||
|
||
|
||
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)
|