Move presence change to a SlackTeam method

1 files changed, 9 insertions, 6 deletions

diff --git a/wee_slack.py b/wee_slack.py
index d92bff5..4d78fc2 100644
--- a/wee_slack.py
+++ b/wee_slack.py
@@ -1081,6 +1081,14 @@ class SlackTeam(object):
             dbg("Unexpected error: {}\nSent: {}".format(sys.exc_info()[0], data))
             self.set_connected()
 
+    def update_member_presence(self, user, presence):
+        user.presence = presence
+
+        for c in self.channels:
+            c = self.channels[c]
+            if user.id in c.members:
+                c.update_nicklist(user.id)
+
 
 class SlackChannel(object):
     """
@@ -2154,12 +2162,7 @@ def process_presence_change(message_json, eventrouter, **kwargs):
     if "user" in kwargs:
         user = kwargs["user"]
         team = kwargs["team"]
-        user.presence = message_json["presence"]
-
-        for c in team.channels:
-            c = team.channels[c]
-            if user.id in c.members:
-                c.update_nicklist(user.id)
+        team.update_member_presence(user, message_json["presence"])
 
 
 def process_pref_change(message_json, eventrouter, **kwargs):