File: Add is_nsfw property

This commit is contained in:
Mia Herkt 2022-12-13 21:51:39 +01:00
parent b1ed63c401
commit 6055a50948
Signed by untrusted user: mia
GPG Key ID: 72E154B8622EC191
1 changed files with 5 additions and 1 deletions

View File

@ -149,13 +149,17 @@ class File(db.Model):
self.expiration = expiration
self.mgmt_token = mgmt_token
@property
def is_nsfw(self) -> bool:
return self.nsfw_score and self.nsfw_score > app.config["NSFW_THRESHOLD"]
def getname(self):
return u"{0}{1}".format(su.enbase(self.id), self.ext)
def geturl(self):
n = self.getname()
if self.nsfw_score and self.nsfw_score > app.config["NSFW_THRESHOLD"]:
if self.is_nsfw:
return url_for("get", path=n, secret=self.secret, _external=True, _anchor="nsfw") + "\n"
else:
return url_for("get", path=n, secret=self.secret, _external=True) + "\n"