use function to format host URL

This commit is contained in:
Martin Herkt 2017-01-01 20:26:09 +01:00
parent 6de22ca8fa
commit 0c5d8690cc
No known key found for this signature in database
GPG key ID: C24B9CD04DC6AE7F

View file

@ -94,6 +94,15 @@ def getpath(fn):
def geturl(p): def geturl(p):
return url_for("get", path=p, _external=True) + "\n" 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): def shorten(url):
if len(url) > app.config["MAX_URL_LENGTH"]: if len(url) > app.config["MAX_URL_LENGTH"]:
abort(414) abort(414)
@ -171,10 +180,7 @@ def store_file(f, addr):
return geturl(sf.getname()) return geturl(sf.getname())
def store_url(url, addr): def store_url(url, addr):
fhost_url = url_for(".fhost", _external=True).rstrip("/") if is_fhost_url(url):
fhost_url_https = url_for(".fhost", _external=True, _scheme="https").rstrip("/")
if url.startswith(fhost_url) or url.startswith(fhost_url_https):
return segfault(508) return segfault(508)
r = requests.get(url, stream=True, verify=False) 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. Please allow up to 24 hours for a response.
</pre> </pre>
""".format(url_for(".fhost", _external=True).rstrip("/"), """.format(fhost_url(),
maxsize, str(maxsizehalf).rjust(27), str(maxsizenum).rjust(27), maxsize, str(maxsizehalf).rjust(27), str(maxsizenum).rjust(27),
maxsizeunit.rjust(54), maxsizeunit.rjust(54),
", ".join(app.config["FHOST_MIME_BLACKLIST"])) ", ".join(app.config["FHOST_MIME_BLACKLIST"]))