fu
This commit is contained in:
+28
-15
@@ -24,18 +24,31 @@ class DatabaseService:
|
||||
cursorclass=pymysql.cursors.DictCursor)
|
||||
logger.info("Connected to database.")
|
||||
|
||||
def get_user_badges(self) -> List[Dict]:
|
||||
with self._conn.cursor() as cursor:
|
||||
cursor.execute(
|
||||
"""SELECT u.user_id, u.user_name, s.seat_id, upp.picture, tic.ticket_id
|
||||
FROM users AS u
|
||||
LEFT JOIN seats AS s
|
||||
ON u.user_id = s.`user`
|
||||
LEFT JOIN user_profile_picture AS upp
|
||||
ON u.user_id = upp.user_id
|
||||
INNER JOIN tickets AS tic
|
||||
ON u.user_id = tic.user;"""
|
||||
)
|
||||
user_badges = cursor.fetchall()
|
||||
logger.info(f"Got {len(user_badges)} user badges from database.")
|
||||
return user_badges
|
||||
def get_user_badges(self, user_id: Optional[int] = None) -> List[Dict]:
|
||||
if user_id is None:
|
||||
with self._conn.cursor() as cursor:
|
||||
cursor.execute(
|
||||
"""SELECT u.user_id, u.user_name, s.seat_id, upp.picture, tic.ticket_id
|
||||
FROM users AS u
|
||||
LEFT JOIN seats AS s
|
||||
ON u.user_id = s.`user`
|
||||
LEFT JOIN user_profile_picture AS upp
|
||||
ON u.user_id = upp.user_id
|
||||
INNER JOIN tickets AS tic
|
||||
ON u.user_id = tic.user;"""
|
||||
)
|
||||
user_badges = cursor.fetchall()
|
||||
else:
|
||||
with self._conn.cursor() as cursor:
|
||||
cursor.execute(
|
||||
"""SELECT u.user_id, u.user_name, s.seat_id, upp.picture
|
||||
FROM users AS u
|
||||
LEFT JOIN seats AS s
|
||||
ON u.user_id = s.`user`
|
||||
LEFT JOIN user_profile_picture AS upp
|
||||
ON u.user_id = upp.user_id
|
||||
WHERE u.user_id = %s;""", (user_id,)
|
||||
)
|
||||
user_badges = cursor.fetchall()
|
||||
logger.info(f"Got {len(user_badges)} user badges from database.")
|
||||
return user_badges
|
||||
|
||||
Reference in New Issue
Block a user