diff --git a/contrib/scripts/kick_users.py b/contrib/scripts/kick_users.py
index 5dfaec3ad0..b4a14385d0 100755
--- a/contrib/scripts/kick_users.py
+++ b/contrib/scripts/kick_users.py
@@ -1,10 +1,16 @@
#!/usr/bin/env python
+from __future__ import print_function
from argparse import ArgumentParser
import json
import requests
import sys
import urllib
+try:
+ raw_input
+except NameError: # Python 3
+ raw_input = input
+
def _mkurl(template, kws):
for key in kws:
template = template.replace(key, kws[key])
@@ -13,7 +19,7 @@ def _mkurl(template, kws):
def main(hs, room_id, access_token, user_id_prefix, why):
if not why:
why = "Automated kick."
- print "Kicking members on %s in room %s matching %s" % (hs, room_id, user_id_prefix)
+ print("Kicking members on %s in room %s matching %s" % (hs, room_id, user_id_prefix))
room_state_url = _mkurl(
"$HS/_matrix/client/api/v1/rooms/$ROOM/state?access_token=$TOKEN",
{
@@ -22,13 +28,13 @@ def main(hs, room_id, access_token, user_id_prefix, why):
"$TOKEN": access_token
}
)
- print "Getting room state => %s" % room_state_url
+ print("Getting room state => %s" % room_state_url)
res = requests.get(room_state_url)
- print "HTTP %s" % res.status_code
+ print("HTTP %s" % res.status_code)
state_events = res.json()
if "error" in state_events:
- print "FATAL"
- print state_events
+ print("FATAL")
+ print(state_events)
return
kick_list = []
@@ -44,15 +50,15 @@ def main(hs, room_id, access_token, user_id_prefix, why):
kick_list.append(event["state_key"])
if len(kick_list) == 0:
- print "No user IDs match the prefix '%s'" % user_id_prefix
+ print("No user IDs match the prefix '%s'" % user_id_prefix)
return
- print "The following user IDs will be kicked from %s" % room_name
+ print("The following user IDs will be kicked from %s" % room_name)
for uid in kick_list:
- print uid
+ print(uid)
doit = raw_input("Continue? [Y]es\n")
if len(doit) > 0 and doit.lower() == 'y':
- print "Kicking members..."
+ print("Kicking members...")
# encode them all
kick_list = [urllib.quote(uid) for uid in kick_list]
for uid in kick_list:
@@ -69,14 +75,14 @@ def main(hs, room_id, access_token, user_id_prefix, why):
"membership": "leave",
"reason": why
}
- print "Kicking %s" % uid
+ print("Kicking %s" % uid)
res = requests.put(kick_url, data=json.dumps(kick_body))
if res.status_code != 200:
- print "ERROR: HTTP %s" % res.status_code
+ print("ERROR: HTTP %s" % res.status_code)
if res.json().get("error"):
- print "ERROR: JSON %s" % res.json()
-
-
+ print("ERROR: JSON %s" % res.json())
+
+
if __name__ == "__main__":
parser = ArgumentParser("Kick members in a room matching a certain user ID prefix.")
|
