mirror of
https://github.com/ansible/awx.git
synced 2026-01-11 01:57:35 -03:30
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
import hammock
|
|
import os
|
|
import requests
|
|
import sys
|
|
import json
|
|
|
|
# this is temporary
|
|
username = os.getenv("ACOM_USER","admin")
|
|
password = os.getenv("ACOM_PASS","admin")
|
|
server = os.getenv("ACOM_SERVER","127.0.0.1:8000")
|
|
|
|
# TODO: error handling/output/etc
|
|
# TODO: format into actual command line
|
|
|
|
PARAMS = {
|
|
'format' : 'json',
|
|
}
|
|
HEADERS = {
|
|
'Content-Type' : 'application/json'
|
|
}
|
|
AUTH = (username, password)
|
|
|
|
handle = hammock.Hammock("http://%s/api/v1" % server, auth=AUTH, append_slash=True, params=PARAMS, headers=HEADERS)
|
|
|
|
class Collection(object):
|
|
|
|
def __init__(self, handle):
|
|
self.handle = handle
|
|
self.response = self.accessor().GET(auth=AUTH, headers=HEADERS)
|
|
assert self.response.status_code == 200
|
|
# TODO: error handling on non-200
|
|
print "RESPONSE=%s" % self.response.text
|
|
self.data = json.loads(self.response.text)
|
|
self.meta = self.data['meta']
|
|
self.objects = self.data['objects']
|
|
self.meta = self.data['meta']
|
|
self.objects = self.data['objects']
|
|
|
|
def accessor(self):
|
|
return exceptions.NotImplementedError()
|
|
|
|
def add(self, data):
|
|
# TODO: error handling
|
|
json_data = json.dumps(data)
|
|
response = self.accessor().POST(data=json_data)
|
|
print response.status_code
|
|
print response.text
|
|
assert response.status_code == 201
|
|
# FIXME: error handling
|
|
data2 = response.text
|
|
print data2
|
|
return Entry(data)
|
|
|
|
def __iter__(self):
|
|
for x in self.objects:
|
|
yield Entry(x)
|
|
|
|
class Entry(object):
|
|
|
|
def __init__(self, data):
|
|
self.data = data
|
|
self.resource_uri = data.get('resource_uri', None)
|
|
print "LOADING"
|
|
self.accessor = hammock.Hammock(self.resource_uri, auth=AUTH, append_slash=True, params=PARAMS, headers=HEADERS)
|
|
|
|
def __repr__(self):
|
|
return repr(self.data)
|
|
|
|
class Organizations(Collection):
|
|
|
|
def accessor(self):
|
|
return self.handle.organizations
|
|
|
|
#(Epdb) got.text
|
|
#u'{"meta": {"limit": 20, "next": null, "offset": 0, "previous": null, "total_count": 1}, "objects": [{"active": true, "creation_date": "2013-03-15", "description": "testorg!", "id": 1, "name": "testorg", "resource_uri": "/api/v1/organizations/1/"}]}'
|
|
#
|
|
|
|
try:
|
|
print "---"
|
|
orgs = Organizations(handle)
|
|
for x in orgs:
|
|
print x
|
|
print "---"
|
|
orgs.add(dict(description="new org?", name="new org"))
|
|
print "---"
|
|
|
|
print "---"
|
|
orgs = Organizations(handle)
|
|
for x in orgs:
|
|
print x
|
|
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print "connect failed"
|
|
sys.exit(1)
|
|
|