From d3354b757205cf2dfa211b400d1851f34f902def Mon Sep 17 00:00:00 2001
From: Aevann1 <Aevann3@gmail.com>
Date: Sun, 22 Aug 2021 20:25:50 +0200
Subject: [PATCH] fdfd

---
 files/classes/award.py | 44 ++++++++++++++++++++++++++++--------------
 1 file changed, 29 insertions(+), 15 deletions(-)

diff --git a/files/classes/award.py b/files/classes/award.py
index 95b0ad20a..7ae073b0f 100644
--- a/files/classes/award.py
+++ b/files/classes/award.py
@@ -1,23 +1,37 @@
 from sqlalchemy import *
 from sqlalchemy.orm import relationship
 from files.__main__ import Base
+from os import environ
 
-AWARDS = {
-	"ban": {
-		"kind": "ban",
-		"title": "One-Day Ban",
-		"description": "Ban the author for a day.",
-		"icon": "fas fa-gavel",
-		"color": "text-danger"
-	},
-	"shit": {
-		"kind": "shit",
-		"title": "Shitpost",
-		"description": "Let OP know how much their post sucks ass. Flies will swarm their idiotic post. (flies only work on posts lol!!)",
-		"icon": "fas fa-poop",
-		"color": "text-black-50"
+site_name = environ.get("SITE_NAME").strip()
+
+if site_name = "Drama":
+	AWARDS = {
+		"ban": {
+			"kind": "ban",
+			"title": "One-Day Ban",
+			"description": "Ban the author for a day.",
+			"icon": "fas fa-gavel",
+			"color": "text-danger"
+		},
+		"shit": {
+			"kind": "shit",
+			"title": "Shitpost",
+			"description": "Let OP know how much their post sucks ass. Flies will swarm their idiotic post. (flies only work on posts lol!!)",
+			"icon": "fas fa-poop",
+			"color": "text-black-50"
+		}
+	}
+else:
+	AWARDS = {
+		"shit": {
+			"kind": "shit",
+			"title": "Shitpost",
+			"description": "Let OP know how much their post sucks ass. Flies will swarm their idiotic post. (flies only work on posts lol!!)",
+			"icon": "fas fa-poop",
+			"color": "text-black-50"
+		}
 	}
-}
 
 
 class AwardRelationship(Base):