1 files changed, 8 insertions, 0 deletions
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)