From e609e7574d87a7311d43475ea652f86962bc75ec Mon Sep 17 00:00:00 2001
From: Aevann1 <Aevann1@protonmail.com>
Date: Mon, 26 Jul 2021 03:34:58 +0200
Subject: [PATCH] gffg

---
 drama/classes/user.py  | 11 ++++-------
 drama/routes/admin.py  |  2 +-
 drama/routes/static.py |  2 +-
 3 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/drama/classes/user.py b/drama/classes/user.py
index 53e1bac07e..471607b1fc 100644
--- a/drama/classes/user.py
+++ b/drama/classes/user.py
@@ -75,8 +75,8 @@ class User(Base, Stndrd, Age_times):
 	#	primaryjoin="and_(Notification.user_id==User.id, Notification.read==False)")
 
 	referred_by = Column(Integer)
-	is_banned = Column(Integer, default=0)
-	unban_utc = Column(Integer, default=0)
+	is_banned = Column(Integer, default=None)
+	unban_utc = Column(Integer, default=None)
 	ban_reason = Column(String, default="")
 	feed_nonce = Column(Integer, default=0)
 	login_nonce = Column(Integer, default=0)
@@ -94,7 +94,6 @@ class User(Base, Stndrd, Age_times):
 	hide_bot = Column(Boolean, default=False)
 	is_private = Column(Boolean, default=False)
 	read_announcement_utc = Column(Integer, default=0)
-	unban_utc = Column(Integer, default=0)
 	filter_nsfw = Column(Boolean, default=False)
 	stored_subscriber_count = Column(Integer, default=0)
 	defaultsortingcomments = Column(String, default="top")
@@ -586,7 +585,7 @@ class User(Base, Stndrd, Age_times):
 	def json_core(self):
 
 		now = int(time.time())
-		if self.is_banned and (self.unban_utc == 0 or now < self.unban_utc):
+		if self.is_banned and (not self.unban_utc or now < self.unban_utc):
 			return {'username': self.username,
 					'permalink': self.permalink,
 					'is_banned': True,
@@ -618,7 +617,6 @@ class User(Base, Stndrd, Age_times):
 			self.unban_utc = ban_time
 
 		else:
-			self.unban_utc = 0
 			if self.has_banner:
 				self.del_banner()
 			if self.profileurl:
@@ -646,8 +644,7 @@ class User(Base, Stndrd, Age_times):
 
 	@property
 	def is_suspended(self):
-		return (self.is_banned and (self.unban_utc ==
-									0 or self.unban_utc > time.time()))
+		return (self.is_banned and (not self.unban_utc or self.unban_utc > time.time()))
 
 	@property
 	def is_blocking(self):
diff --git a/drama/routes/admin.py b/drama/routes/admin.py
index 17bd14e70b..e3f55fc68c 100644
--- a/drama/routes/admin.py
+++ b/drama/routes/admin.py
@@ -234,7 +234,7 @@ def participation_stats(v):
 
 	data = {"valid_users": g.db.query(User).count(),
 			"private_users": g.db.query(User).filter_by(is_private=False).count(),
-			"banned_users": g.db.query(User).filter(User.is_banned > 0, User.unban_utc == 0).count(),
+			"banned_users": g.db.query(User).filter(User.is_banned > 0, User.unban_utc == None).count(),
 			"verified_users": g.db.query(User).filter_by(is_activated=True).count(),
 			"signups_last_24h": g.db.query(User).filter(User.created_utc > day).count(),
 			"total_posts": g.db.query(Submission).count(),
diff --git a/drama/routes/static.py b/drama/routes/static.py
index f1be69e322..695ead7865 100644
--- a/drama/routes/static.py
+++ b/drama/routes/static.py
@@ -171,7 +171,7 @@ def blocks(v):
 def banned(v):
 	if v and v.is_banned and not v.unban_utc: return render_template("seized.html")
 
-	users = [x for x in g.db.query(User).filter(User.is_banned != 0, User.unban_utc == 0).all()]
+	users = [x for x in g.db.query(User).filter(User.is_banned is not None, User.unban_utc is None).all()]
 	return render_template("banned.html", v=v, users=users)
 
 @app.route("/formatting", methods=["GET"])