#!/usr/bin/env python
import itertools
import subprocess
import requests
import json
import sys
def ssh_cmd(extra):
cmd = ['ssh', '-p', '29418', 'gerrit.wikimedia.org', '--', 'gerrit']
return subprocess.check_output(cmd + extra)
def chunks(l, n=100):
"""Yield successive n-sized chunks from l."""
for i in range(0, len(l), n):
yield l[i:i + n]
def trust_accounts(accounts):
accounts_cmd = ['set-members']
accounts_cmd += [
str(item) for sublist in
list(zip(itertools.repeat('--add'), accounts))
for item in sublist
]
accounts_cmd += ['Trusted-Users']
return ssh_cmd(accounts_cmd)
class GerritActiveAccounts(object):
_gerrit_shit = ")]}'"
gerrit = 'https://gerrit.wikimedia.org/r'
def __init__(self):
self.url_fmt = '/accounts/?q=is:active&start={}'
self.active = []
def make_url(self, start=0):
base = self.gerrit + self.url_fmt
return base.format(start)
def get_json(self, start=0):
r = requests.get(self.make_url(start))
r.raise_for_status()
return json.loads(r.text[len(self._gerrit_shit):])
def get_users(self, start=0):
for entry in self.get_json(start):
for k, v in entry.items():
if k == '_account_id':
self.active.append(int(v))
if k == '_more_accounts':
return self.get_users(start=self.active[-1])
return set(self.active)
class GerritTrustedAccounts(object):
TRUSTED_USERS = 'Trusted-Users'
def __init__(self):
self.users = []
def get_users(self):
for row in ssh_cmd(['ls-members', self.TRUSTED_USERS]).splitlines():
if row.strip().startswith(b'id'):
continue
self.users.append(int(row.split(b'\t')[0]))
return set(self.users)
def main():
ga = GerritActiveAccounts()
active_accounts = ga.get_users()
ta = GerritTrustedAccounts()
trusted_accounts = ta.get_users()
accounts_to_add = active_accounts - trusted_accounts
for chunk in chunks(list(accounts_to_add)):
print(trust_accounts(chunk))
if __name__ == '__main__':
main()