50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
import itertools
|
|
import json
|
|
|
|
from django.utils.http import urlquote
|
|
from jinja2.ext import Extension
|
|
from mptt.utils import get_cached_trees
|
|
from statici18n.templatetags.statici18n import inlinei18n
|
|
|
|
from judge.highlight_code import highlight_code
|
|
from judge.user_translations import gettext
|
|
from . import (
|
|
camo,
|
|
chat,
|
|
datetime,
|
|
filesize,
|
|
gravatar,
|
|
language,
|
|
markdown,
|
|
rating,
|
|
reference,
|
|
render,
|
|
social,
|
|
spaceless,
|
|
submission,
|
|
timedelta,
|
|
)
|
|
from . import registry
|
|
|
|
registry.function("str", str)
|
|
registry.filter("str", str)
|
|
registry.filter("json", json.dumps)
|
|
registry.filter("highlight", highlight_code)
|
|
registry.filter("urlquote", urlquote)
|
|
registry.filter("roundfloat", round)
|
|
registry.function("inlinei18n", inlinei18n)
|
|
registry.function("mptt_tree", get_cached_trees)
|
|
registry.function("user_trans", gettext)
|
|
|
|
|
|
@registry.function
|
|
def counter(start=1):
|
|
return itertools.count(start).__next__
|
|
|
|
|
|
class DMOJExtension(Extension):
|
|
def __init__(self, env):
|
|
super(DMOJExtension, self).__init__(env)
|
|
env.globals.update(registry.globals)
|
|
env.filters.update(registry.filters)
|
|
env.tests.update(registry.tests)
|