Handle callable initial_channels (#68)

PythonistaGuild · Jan 21, 2020 · 7c9738b · 7c9738b
2 parents 258b90a + ab6b9e4
commit 7c9738b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/twitchio/websocket.py b/twitchio/websocket.py
@@ -75,6 +75,13 @@ def __init__(self, **kwargs):
         self.nick = kwargs.get('nick').lower()
         self.modes = kwargs.pop('modes', ("commands", "tags", "membership"))
         self._initial_channels = kwargs.get('initial_channels')
+
+        if callable(self._initial_channels):
+            _temp_initial_channels = self._initial_channels()
+            if isinstance(_temp_initial_channels, (list, tuple)):
+                self._initial_channels = _temp_initial_channels
+            else:
+                self._initial_channels = [_temp_initial_channels]
 
         self._last_ping = 0