mirror of https://github.com/watcha-fr/synapse
commit
db8d4e8dd6
@ -0,0 +1,93 @@ |
||||
#!/usr/bin/env python |
||||
from argparse import ArgumentParser |
||||
import json |
||||
import requests |
||||
import sys |
||||
import urllib |
||||
|
||||
def _mkurl(template, kws): |
||||
for key in kws: |
||||
template = template.replace(key, kws[key]) |
||||
return template |
||||
|
||||
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) |
||||
room_state_url = _mkurl( |
||||
"$HS/_matrix/client/api/v1/rooms/$ROOM/state?access_token=$TOKEN", |
||||
{ |
||||
"$HS": hs, |
||||
"$ROOM": room_id, |
||||
"$TOKEN": access_token |
||||
} |
||||
) |
||||
print "Getting room state => %s" % room_state_url |
||||
res = requests.get(room_state_url) |
||||
print "HTTP %s" % res.status_code |
||||
state_events = res.json() |
||||
if "error" in state_events: |
||||
print "FATAL" |
||||
print state_events |
||||
return |
||||
|
||||
kick_list = [] |
||||
room_name = room_id |
||||
for event in state_events: |
||||
if not event["type"] == "m.room.member": |
||||
if event["type"] == "m.room.name": |
||||
room_name = event["content"].get("name") |
||||
continue |
||||
if not event["content"].get("membership") == "join": |
||||
continue |
||||
if event["state_key"].startswith(user_id_prefix): |
||||
kick_list.append(event["state_key"]) |
||||
|
||||
if len(kick_list) == 0: |
||||
print "No user IDs match the prefix '%s'" % user_id_prefix |
||||
return |
||||
|
||||
print "The following user IDs will be kicked from %s" % room_name |
||||
for uid in kick_list: |
||||
print uid |
||||
doit = raw_input("Continue? [Y]es\n") |
||||
if len(doit) > 0 and doit.lower() == 'y': |
||||
print "Kicking members..." |
||||
# encode them all |
||||
kick_list = [urllib.quote(uid) for uid in kick_list] |
||||
for uid in kick_list: |
||||
kick_url = _mkurl( |
||||
"$HS/_matrix/client/api/v1/rooms/$ROOM/state/m.room.member/$UID?access_token=$TOKEN", |
||||
{ |
||||
"$HS": hs, |
||||
"$UID": uid, |
||||
"$ROOM": room_id, |
||||
"$TOKEN": access_token |
||||
} |
||||
) |
||||
kick_body = { |
||||
"membership": "leave", |
||||
"reason": why |
||||
} |
||||
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 |
||||
if res.json().get("error"): |
||||
print "ERROR: JSON %s" % res.json() |
||||
|
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
parser = ArgumentParser("Kick members in a room matching a certain user ID prefix.") |
||||
parser.add_argument("-u","--user-id",help="The user ID prefix e.g. '@irc_'") |
||||
parser.add_argument("-t","--token",help="Your access_token") |
||||
parser.add_argument("-r","--room",help="The room ID to kick members in") |
||||
parser.add_argument("-s","--homeserver",help="The base HS url e.g. http://matrix.org") |
||||
parser.add_argument("-w","--why",help="Reason for the kick. Optional.") |
||||
args = parser.parse_args() |
||||
if not args.room or not args.token or not args.user_id or not args.homeserver: |
||||
parser.print_help() |
||||
sys.exit(1) |
||||
else: |
||||
main(args.homeserver, args.room, args.token, args.user_id, args.why) |
Loading…
Reference in new issue