From cd9952ff8adf8778e2782125353e881da0001311 Mon Sep 17 00:00:00 2001 From: Aevann Date: Mon, 1 Apr 2024 05:14:43 +0200 Subject: [PATCH] fix april fool in WPD --- files/classes/user.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/files/classes/user.py b/files/classes/user.py index a3a69e2fb..db272291c 100644 --- a/files/classes/user.py +++ b/files/classes/user.py @@ -1312,7 +1312,12 @@ class User(Base): def user_name(self): random.seed(self.id) uid = random.choice(USER_IDS) - self = g.db.query(User).options(load_only(User.username, User.earlylife)).filter_by(id=uid).one() + + to_load = [User.username] + if SITE_NAME == 'rDrama': + to_load.append(User.earlylife) + + self = g.db.query(User).options(load_only(*to_load)).filter_by(id=uid).one() if self.earlylife: expiry = int(self.earlylife - time.time()) if expiry > 86400: @@ -1328,7 +1333,7 @@ class User(Base): def switched(self): random.seed(self.id) uid = random.choice(USER_IDS) - return g.db.query(User).options(load_only(User.username, User.earlylife)).filter_by(id=uid).one() + return g.db.query(User).filter_by(id=uid).one() @property @lazy