blob: 6a3161abab8022c4627f74384c8e7ca77c2a18e5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
"""Jinja2 extensions."""
import json
import string
import uuid
from secrets import choice
from jinja2.ext import Extension
from slugify import slugify as pyslugify
class JsonifyExtension(Extension):
"""Jinja2 extension to convert a Python object to JSON."""
def __init__(self, environment):
"""Initialize the extension with the given environment."""
super().__init__(environment)
def jsonify(obj):
return json.dumps(obj, sort_keys=True, indent=4)
environment.filters['jsonify'] = jsonify
class RandomStringExtension(Extension):
"""Jinja2 extension to create a random string."""
def __init__(self, environment):
"""Jinja2 Extension Constructor."""
super().__init__(environment)
def random_ascii_string(length, punctuation=False):
if punctuation:
corpus = "".join((string.ascii_letters, string.punctuation))
else:
corpus = string.ascii_letters
return "".join(choice(corpus) for _ in range(length))
environment.globals.update(random_ascii_string=random_ascii_string)
class SlugifyExtension(Extension):
"""Jinja2 Extension to slugify string."""
def __init__(self, environment):
"""Jinja2 Extension constructor."""
super().__init__(environment)
def slugify(value, **kwargs):
"""Slugifies the value."""
return pyslugify(value, **kwargs)
environment.filters['slugify'] = slugify
class UUIDExtension(Extension):
"""Jinja2 Extension to generate uuid4 string."""
def __init__(self, environment):
"""Jinja2 Extension constructor."""
super().__init__(environment)
def uuid4():
"""Generate UUID4."""
return str(uuid.uuid4())
environment.globals.update(uuid4=uuid4)
|