1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#!/usr/bin/env python
import sys
import httplib
import urllib
import base64
import json
import optparse
import getpass
import logging
from pprint import pprint
#from BeautifulSoup import BeautifulSoup
BASE_URL = "https://login.linaro.org:8443/crowd/rest/usermanagement/1"
AUTH = (None, None)
log = logging.getLogger()
class CrowdRequestError(Exception):
pass
def strip_html(html):
return html
html = BeautifulSoup(html, convertEntities=BeautifulSoup.HTML_ENTITIES)
if not html.body:
return html
m = ' '.join(html.body.findAll(text=True))
return m
def rest(uri, params):
c = httplib.HTTPSConnection("login.linaro.org", 8443)
auth = base64.encodestring('%s:%s' % AUTH)
headers = {"Authorization": "Basic " + auth}
url = uri + "?" + urllib.urlencode(params)
c.request("GET", "/crowd/rest/usermanagement/1" + url, headers=headers)
resp = c.getresponse()
if resp.status != 200:
data = resp.read()
log.error("Non-successful response code %d from Crowd: %s", resp.status, strip_html(data))
raise CrowdRequestError(resp, data)
data = json.load(resp)
return data
if __name__ == "__main__":
logging.basicConfig()
optparser = optparse.OptionParser(usage="%prog user|alises|group|usergroups|ismember")
optparser.add_option("-u", "--user", default="rest-test", help="Crowd username")
optparser.add_option("-p", "--passwd", help="Crowd password")
optparser.add_option("-P", "--ask-passwd", action="store_true", help="Ask Crowd password")
optparser.add_option("-n", "--nested", action="store_true", help="Process nested groups")
optparser.add_option("-r", "--raw", action="store_true", help="Show raw JSON response")
options, args = optparser.parse_args(sys.argv[1:])
if len(args) < 1:
optparser.error("Wrong number of arguments")
if options.ask_passwd:
options.passwd = getpass.getpass("Crowd password: ")
AUTH = (options.user, options.passwd)
if args[0] == "user":
pprint(rest("/user.json", {"username": args[1], "expand": "attributes"}))
elif args[0] == "aliases":
pprint(rest("/aliases.json", {"username": args[1]}))
elif args[0] == "group":
pprint(rest("/group.json", {"groupname": args[1], "expand": "attributes"}))
elif args[0] == "usergroups":
url = "/user/group/nested.json" if options.nested else "/user/group/direct.json"
data = rest(url, {"username": args[1]})
if options.raw:
pprint(data)
else:
names = [x["name"] for x in data["groups"]]
names.sort()
for n in names:
print n
elif args[0] == "ismember":
url = "/user/group/nested.json" if options.nested else "/user/group/direct.json"
try:
data = rest(url, {"username": args[1], "groupname": args[2]})
assert data["name"] == args[2]
print "yes"
except CrowdRequestError, e:
print e.args[1]
print "no"
|
