def logout_action(msg): resp = RespClass() try: cur.execute(f"SELECT login FROM users WHERE id = {user_id[notified_socket]}") result = cur.fetchall() if len(result) == 1: cur.execute(f"UPDATE users SET islogged = 0 WHERE id = {user_id[notified_socket]}") mydb.commit() user_id[notified_socket] = None resp.logout(True, msg) notified_socket.send(resp.get_prepared_response()) except KeyError: print("Client is not logged") resp.get_chats(False, "You are not logged", None) notified_socket.send(resp.get_prepared_response())
def get_chats(cur, user_id, notified_socket): response = RespClass() try: cur.execute( f"SELECT chat FROM chatmember WHERE user = {user_id[notified_socket]} order by chat" ) result = cur.fetchall() if len(result) == 0: response.get_chats(True, "You are not added to any chat", None) notified_socket.send(response.get_prepared_response()) else: chats = [] for row in result: chats.append(row[0]) response.get_chats(True, None, chats) notified_socket.send(response.get_prepared_response()) except KeyError: print("Client is not logged") response.get_chats(False, "You are not logged", None) notified_socket.send(response.get_prepared_response())