def on_log(self, username): self.logged = True Interaction.clients.append(self) self.username = username line = Data.users[Data.users['username'] == username] self.friends = line['friends'].values[0] self.index = line.index[0] Interaction.manage_friends(self) self.look_for_fr_request() Data.send_stats(username, self.conn)
def add_friend(self, username): other_index = Data.users[Data.users['username'] == username].index[0] # add friend to data Data.users.loc[self.index, 'friends'].append(other_index) Data.users.loc[other_index, 'friends'].append(self.index) # remove other from demands Data.users.loc[self.index, 'demands'].remove(other_index) store_user_data() Interaction.manage_friends(self) # if other is connected, update his friends if Interaction.is_connected(username): other_client = Interaction.get_client(username) Interaction.manage_friends(other_client)