feat: add sample jinja filter

upciti · Feb 20, 2022 · c1770c2 · c1770c2
1 parent 35796ff
commit c1770c2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/wakemebot/plugin.py b/src/wakemebot/plugin.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+from random import sample
 from typing import Any, Dict, List, Optional
 
 from jinja2 import Environment
@@ -13,7 +14,10 @@
 from .apt import parse_repository
 from .badges import Badge
 
+filters = {"sample": sample}
+
 environment = Environment()
+environment.filters.update(filters)
 
 
 class WakeMeDocPlugin(BasePlugin):
@@ -101,6 +105,7 @@ def on_files(self, files: Files, config: Config) -> Files:
 
     def on_env(self, env: Environment, config: Config, files: Files) -> Environment:
         env.globals.update({"repository": self.repository, "packages": self.packages})
+        env.filters.update(filters)
         return env
 
     def on_page_context(