Source code for abilian.web.filters

# coding: utf-8
"""Add a few specific filters to Jinja2."""
import re
from calendar import timegm
from datetime import date, datetime
from functools import wraps
from typing import Any, Callable, Optional, Union

import bleach
import dateutil.parser
import flask_babel as babel
from babel.dates import DateTimePattern, format_timedelta, parse_pattern
from deprecated import deprecated
from jinja2 import Environment, escape, evalcontextfilter
from jinja2.nodes import EvalContext
from markupsafe import Markup
from pytz import utc
from werkzeug.routing import BuildError

from abilian.core.util import local_dt, slugify, utc_dt

from .util import url_for


[docs]def autoescape(filter_func: Callable) -> Callable: """Decorator to autoescape result from filters.""" @evalcontextfilter @wraps(filter_func) def _autoescape( eval_ctx: EvalContext, *args: str, **kwargs: Any ) -> Union[Markup, str]: result = filter_func(*args, **kwargs) if eval_ctx.autoescape: result = Markup(result) return result return _autoescape
[docs]@autoescape def nl2br(value: str) -> Markup: """Replace newlines with <br />.""" result = escape(value).replace("\n", Markup("<br />\n")) return result
_PARAGRAPH_RE = re.compile(r"(?:\r\n|\r|\n){2,}")
[docs]@autoescape def paragraphs(value: str) -> str: """Blank lines delimitates paragraphs.""" result = "\n\n".join( "<p>{}</p>".format(p.strip().replace("\n", Markup("<br />\n"))) for p in _PARAGRAPH_RE.split(escape(value)) ) return result
[docs]def labelize(s: str) -> str: return " ".join([w.capitalize() for w in s.split("_")])
[docs]def filesize(d: Union[int, str]) -> Markup: if not isinstance(d, int): d = int(d) if d < 1000: s = f"{d:d}&nbsp;B" elif d < int(1e4): s = "%.1f&nbsp;kB" % (d / 1e3) elif d < int(1e6): s = "%.0f&nbsp;kB" % (d / 1e3) elif d < int(1e7): s = "%.1f&nbsp;MB" % (d / 1e6) elif d < int(1e9): s = "%.0f&nbsp;MB" % (d / 1e6) elif d < int(1e10): s = "%.1f&nbsp;GB" % (d / 1e9) else: s = "%.0f&nbsp;GB" % (d / 1e9) return Markup(s)
[docs]def roughsize(size: int, above: int = 20, mod: int = 10) -> str: """6 -> '6' 15 -> '15' 134 -> '130+'.""" if size < above: return str(size) return f"{(size - size % mod):d}+"
[docs]def datetimeparse(s) -> Optional[datetime]: """Parse a string date time to a datetime object. Suitable for dates serialized with .isoformat() :return: None, or an aware datetime instance, tz=UTC. """ try: dt = dateutil.parser.parse(s) except ValueError: return None return utc_dt(dt)
[docs]def age( dt: Optional[datetime], now: Optional[datetime] = None, add_direction: bool = True, date_threshold: Optional[Any] = None, ) -> str: """ :param dt: :class:`datetime<datetime>` instance to format :param now: :class:`datetime<datetime>` instance to compare to `dt` :param add_direction: if `True`, will add "in" or "ago" (example for `en` locale) to time difference `dt - now`, i.e "in 9 min." or " 9min. ago" :param date_threshold: above threshold, will use a formated date instead of elapsed time indication. Supported values: "day". """ # Fail silently for now XXX if not dt: return "" if not now: now = datetime.utcnow() locale = babel.get_locale() dt = utc_dt(dt) now = utc_dt(now) delta = dt - now if date_threshold is not None: dy, dw, dd = dt_cal = dt.isocalendar() ny, nw, nd = now_cal = now.isocalendar() if dt_cal != now_cal: # not same day remove_year = dy == ny date_fmt = locale.date_formats["long"].pattern time_fmt = locale.time_formats["short"].pattern fmt = locale.datetime_formats["medium"] if remove_year: date_fmt = date_fmt.replace("y", "").strip() # remove leading or trailing spaces, comma, etc... date_fmt = re.sub("^[^A-Za-z]*|[^A-Za-z]*$", "", date_fmt) fmt = fmt.format(time_fmt, date_fmt) return babel.format_datetime(dt, format=fmt) # don't use (flask.ext.)babel.format_timedelta: as of Flask-Babel 0.9 it # doesn't support "threshold" arg. return format_timedelta( delta, locale=locale, granularity="minute", threshold=0.9, add_direction=add_direction, )
[docs]def date_age(dt: Optional[datetime], now: Optional[datetime] = None) -> str: # Fail silently for now XXX if not dt: return "" formatted_date = babel.format_datetime(dt, format="yyyy-MM-dd HH:mm") return f"{formatted_date} ({age(dt, now)})"
[docs]@deprecated def date_fmt(value, format="EE, d MMMM y"): """ @deprecated: use flask_babel's dateformat filter instead. """ if isinstance(value, date): return babel.format_date(value, format) else: return babel.format_date(local_dt(value), format)
[docs]def babel2datepicker(pattern: DateTimePattern) -> str: """Convert date format from babel (http://babel.pocoo.org/docs/dates/#date- fields)) to a format understood by bootstrap-datepicker.""" if not isinstance(pattern, DateTimePattern): pattern = parse_pattern(pattern) map_fmt = { # days "d": "dd", "dd": "dd", "EEE": "D", "EEEE": "DD", "EEEEE": "D", # narrow name => short name # months "M": "mm", "MM": "mm", "MMM": "M", "MMMM": "MM", # years "y": "yyyy", "yy": "yyyy", "yyy": "yyyy", "yyyy": "yyyy", # time picker format # hours "h": "%I", "hh": "%I", "H": "%H", "HH": "%H", # minutes, "m": "%M", "mm": "%M", # seconds "s": "%S", "ss": "%S", # am/pm "a": "%p", } return pattern.format % map_fmt
# Doesn't work yet. TZ issues.
[docs]def to_timestamp(dt): utc_datetime = dt.astimezone(utc) return timegm(utc_datetime.timetuple()) + utc_datetime.microsecond / 1e6
[docs]def abbrev(s: str, max_size: int) -> str: if len(s) <= max_size: return s else: h = max_size // 2 - 1 return s[0:h] + "..." + s[-h:]
[docs]def bool2check(val, true="\u2713", false=""): """Filter value as boolean and show check mark (✓) or nothing.""" return true if val else false
[docs]@autoescape def linkify(s: str) -> Markup: return Markup(bleach.linkify(s))
[docs]def obj_to_url(obj): """Find url for obj using :func:`url_for`, return empty string is not found. :func:`url_for` is also provided in jinja context, the filtering version is forgiving when `obj` has no default view set. """ try: return url_for(obj) except BuildError: return ""
[docs]def init_filters(env: Environment) -> None: env.filters["nl2br"] = nl2br env.filters["paragraphs"] = paragraphs env.filters["date_age"] = date_age env.filters["datetimeparse"] = datetimeparse env.filters["age"] = age env.filters["date"] = date_fmt env.filters["babel2datepicker"] = babel2datepicker env.filters["to_timestamp"] = to_timestamp env.filters["url_for"] = obj_to_url env.filters["abbrev"] = abbrev env.filters["filesize"] = filesize env.filters["roughsize"] = roughsize env.filters["labelize"] = labelize env.filters["linkify"] = linkify env.filters["toslug"] = slugify env.filters["bool2check"] = bool2check