From 4979f43baba8c7377471bbf65e7b5c003bf65406 Mon Sep 17 00:00:00 2001 From: Satya Ortiz-Gagne <satya.ortiz-gagne@mila.quebec> Date: Mon, 10 Jun 2019 10:12:45 -0400 Subject: DAIDE - Implemented notifications and notification_managers - MAP = MapNameNotification - HLO = HelloNotification - SCO = SupplyCenterNotification - NOW = CurrentPositionNotification - MIS = MissingOrdersNotification - ORD = OrderResultNotification - TME = TimeToDeadlineNotification - CCD = PowerInCivilDisorderNotification - OUT = PowerIsEliminatedNotification - DRW = DrawNotification - FRM = MessageFromNotification - SLO = SoloNotification - SMR = SummaryNotification - OFF = TurnOffNotification --- diplomacy/server/users.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'diplomacy/server/users.py') diff --git a/diplomacy/server/users.py b/diplomacy/server/users.py index d1c8ca0..d63df3e 100644 --- a/diplomacy/server/users.py +++ b/diplomacy/server/users.py @@ -116,6 +116,10 @@ class Users(Jsonable): """ Return username of given token. """ return self.token_to_username[token] + def get_user(self, username): + """ Returns user linked to username """ + return self.users.get(username, None) + def get_connection_handler(self, token): """ Return connection handler associated to given token, or None if no handler currently associated. """ return self.token_to_connection_handler.get(token, None) @@ -145,6 +149,10 @@ class Users(Jsonable): self.users[username] = user return user + def replace_user(self, username, new_user): + """ Replaces user object with a new user """ + self.users[username] = new_user + def remove_user(self, username): """ Remove user related to given username. """ user = self.users.pop(username) -- cgit v1.2.3