diff options
author | Martin Herkt <lachs0r@srsfckn.biz> | 2017-01-01 20:26:09 +0100 |
---|---|---|
committer | Martin Herkt <lachs0r@srsfckn.biz> | 2017-01-01 20:26:09 +0100 |
commit | 0c5d8690cc4dad7abff112982504f5c3f254daef (patch) | |
tree | 6baf64adec416f09173bb84fcf4eded06d674acf /fhost.py | |
parent | fix text/ MIME charset workaround for real (diff) | |
download | hex-zero-0c5d8690cc4dad7abff112982504f5c3f254daef.tar.gz hex-zero-0c5d8690cc4dad7abff112982504f5c3f254daef.tar.bz2 hex-zero-0c5d8690cc4dad7abff112982504f5c3f254daef.zip |
use function to format host URL
Diffstat (limited to 'fhost.py')
-rwxr-xr-x | fhost.py | 16 |
1 files changed, 11 insertions, 5 deletions
@@ -94,6 +94,15 @@ def getpath(fn): def geturl(p): return url_for("get", path=p, _external=True) + "\n" +def fhost_url(scheme=None): + if not scheme: + return url_for(".fhost", _external=True).rstrip("/") + else: + return url_for(".fhost", _external=True, _scheme=scheme).rstrip("/") + +def is_fhost_url(url): + return url.startswith(fhost_url()) or url.startswith(fhost_url("https")) + def shorten(url): if len(url) > app.config["MAX_URL_LENGTH"]: abort(414) @@ -171,10 +180,7 @@ def store_file(f, addr): return geturl(sf.getname()) def store_url(url, addr): - fhost_url = url_for(".fhost", _external=True).rstrip("/") - fhost_url_https = url_for(".fhost", _external=True, _scheme="https").rstrip("/") - - if url.startswith(fhost_url) or url.startswith(fhost_url_https): + if is_fhost_url(url): return segfault(508) r = requests.get(url, stream=True, verify=False) @@ -317,7 +323,7 @@ IRC on Freenode, or send an email to lachs0r@(this domain). Please allow up to 24 hours for a response. </pre> -""".format(url_for(".fhost", _external=True).rstrip("/"), +""".format(fhost_url(), maxsize, str(maxsizehalf).rjust(27), str(maxsizenum).rjust(27), maxsizeunit.rjust(54), ", ".join(app.config["FHOST_MIME_BLACKLIST"])) |