skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

dffd

Browse source
This commit is contained in:
Aevann1 2021-08-19 07:40:23 +02:00
parent 8017c95d27
commit f2adcf0997
3 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
files/classes/images.py
View file

@ -4,6 +4,7 @@ from files.__main__ import Base
from os import environ
site = environ.get("DOMAIN").strip()
site_name = environ.get("SITE_NAME").strip()
class Image(Base):
__tablename__ = "images"
@ -15,7 +16,7 @@ class Image(Base):
@property
def path(self): return f"/assets/images/{site}/cover.png"
def path(self): return f"/assets/images/{site_name}/cover.png"

3
files/classes/submission.py
View file

@ -10,6 +10,7 @@ from .flags import *
from os import environ
site = environ.get("DOMAIN").strip()
site_name = environ.get("SITE_NAME").strip()
class SubmissionAux(Base):
@ -203,7 +204,7 @@ class Submission(Base, Stndrd, Age_times, Scores, Fuzzing):
@lazy
def thumb_url(self):
if self.over_18: return f"https://{site}/assets/images/nsfw.png"
elif not self.url: return f"https://{site}/assets/images/{site}/default_thumb_text.png"
elif not self.url: return f"https://{site}/assets/images/{site_name}/default_thumb_text.png"
elif self.thumburl: return self.thumburl
elif "youtu.be" in self.domain or "youtube.com" in self.domain: return f"https://{site}/assets/images/default_thumb_yt.png"
else: return f"https://{site}/assets/images/default_thumb_link.png"

3
files/routes/static.py
View file

@ -3,6 +3,7 @@ from files.__main__ import app, limiter
from files.helpers.alerts import *
site = environ.get("DOMAIN").strip()
site_name = environ.get("SITE_NAME").strip()
@app.get("/stats")
@auth_desired
@ -97,7 +98,7 @@ def index():
@app.get("/assets/favicon.ico")
def favicon():
return send_file(f"./assets/images/{site}/favicon.png")
return send_file(f"./assets/images/{site_name}/favicon.png")
@app.get("/api")
@auth_desired