Add global url_set_query Jinja template function

[cheatdb.git] / app / utils.py

diff --git a/app/utils.py b/app/utils.py
index 5960fa66ca18a575de463f2c2138242ab7e8bb87..cc338b4b04ee37ea12fc4762477c51cec9df1f1a 100644 (file)
--- a/app/utils.py
+++ b/app/utils.py
@@ -23,6 +23,8 @@ from . import app
 import random, string, os, imghdr
 from urllib.parse import urljoin
 
+# These are given to Jinja in template_filters.py
+
 def abs_url_for(path, **kwargs):
        scheme = "https" if app.config["BASE_URL"][:5] == "https" else "http"
        return url_for(path, _external=True, _scheme=scheme, **kwargs)
@@ -30,6 +32,11 @@ def abs_url_for(path, **kwargs):
 def abs_url(path):
        return urljoin(app.config["BASE_URL"], path)
 
+def url_set_query(**kwargs):
+       args = dict(request.args)
+       args.update(kwargs)
+       return url_for(request.endpoint, **args)
+
 def get_int_or_abort(v, default=None):
        if v is None:
                return default