Update invite script to use new API

Author: Grazfather

purge/invite_all_users.py

@@ -1,33 +1,35 @@
 #!/usr/bin/env python3
 
-from slackclient import SlackClient
+import time
+
+from slack_sdk import WebClient
+from slack_sdk.errors import SlackApiError
 
 # Slack API Token
-SLACK_TOKEN  = ""
+SLACK_TOKEN  = "xoxb-..."
 
 # Channel to invite users too
-DEST_CHANNEL = "general"
+DEST_CHANNEL = "channelname"
 
 def channel_id_by_name(client, name):
-    """ Fetch channel ID for a given channel name. """
+    """Fetch channel ID for a given channel name."""
 
     output = client.api_call("channels.list")
-    channels = output['channels']
+    channels = output["channels"]
 
-    channel_id = ''
     for channel in channels:
-        if channel['name'] == name:
-            return channel['id']
+        if channel["name"] == name:
+            return channel["id"]
 
     return None
 
 def get_all_users(client):
-    """ Fetch all users in the team. Includes deleted/deactivated users. """
+    """Fetch all users in the team. Includes deleted/deactivated users."""
 
     output = client.api_call("users.list")
-    return output['members']
+    return output["members"]
 
-sc = SlackClient(SLACK_TOKEN)
+sc = WebClient(SLACK_TOKEN)
 
 channel_id = channel_id_by_name(sc, DEST_CHANNEL)
 
@@ -39,11 +41,26 @@ def get_all_users(client):
 members = get_all_users(sc)
 print("[*] Found {} members.".format(len(members)))
 
+# Join the channel, so we can invite to it
+sc.api_call("conversations.join", json={"channel": channel_id})
+
 # Invite to channel in groups of 30
 # Slack limits channel invitations to 30 members per API call.
 print("[*] Inviting users.")
-member_ids = [member['id'] for member in members]
+member_ids = [member["id"] for member in members]
 groups = [member_ids[n:n+30] for n in range(0, len(member_ids), 30)]
 
 for group in groups:
-    sc.api_call("conversations.invite", channel=channel_id, users=','.join(group))
+    try:
+        sc.api_call("conversations.invite", json={"channel": channel_id, "users":",".join(group)})
+    except SlackApiError as e:
+        # Technically we can have up to 30 errors, and some might be bad
+        # e.response.data["errors"] to check them individually
+        if "ratelimited" in str(e):
+            time.sleep(1)
+        elif "cant_invite_self" in str(e):
+            continue
+        elif "already_in_channel" in str(e):
+            continue
+        # TODO: Should we bail otherwise?
+        continue