diff --git a/files/classes/images.py b/files/classes/images.py index 9b13187635..be0fc8f691 100644 --- a/files/classes/images.py +++ b/files/classes/images.py @@ -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" diff --git a/files/classes/submission.py b/files/classes/submission.py index 20bbe318b6..61b92ee8ed 100644 --- a/files/classes/submission.py +++ b/files/classes/submission.py @@ -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" diff --git a/files/routes/static.py b/files/routes/static.py index b0ae3f9a95..4d2c6977f3 100644 --- a/files/routes/static.py +++ b/files/routes/static.py @@ -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