From 70157c8c27cc29910dba8d8500503b19c0e158ca Mon Sep 17 00:00:00 2001
From: Aevann1 <Aevann3@gmail.com>
Date: Thu, 19 Aug 2021 07:40:23 +0200
Subject: [PATCH] dffd

---
 files/classes/images.py     | 3 ++-
 files/classes/submission.py | 3 ++-
 files/routes/static.py      | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/files/classes/images.py b/files/classes/images.py
index 9b1318763..be0fc8f69 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 20bbe318b..61b92ee8e 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 b0ae3f9a9..4d2c6977f 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