mirror of
https://github.com/ansible/awx.git
synced 2026-01-11 10:00:01 -03:30
105 lines
2.6 KiB
Python
105 lines
2.6 KiB
Python
import os
|
|
import requests
|
|
from requests.auth import HTTPBasicAuth
|
|
import sys
|
|
import json
|
|
|
|
# this is temporary
|
|
username = os.getenv("ACOM_USER","admin")
|
|
password = os.getenv("ACOM_PASS","admin")
|
|
print "USER=%s" % username
|
|
server = os.getenv("ACOM_SERVER","http://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 = HTTPBasicAuth(username, password)
|
|
|
|
def get(url_seg):
|
|
resp = requests.get("%s/api/v1/%s" % (server, url_seg), auth=AUTH)
|
|
return resp
|
|
|
|
def post(url_seg, data):
|
|
resp = requests.post("%s/api/v1/%s" % (server, url_seg), auth=AUTH, data=data, headers=HEADERS)
|
|
return resp
|
|
|
|
class Collection(object):
|
|
|
|
def __init__(self):
|
|
|
|
self.response = get(self.base_url())
|
|
|
|
print self.response.text
|
|
print self.response.status_code
|
|
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 base_url(self):
|
|
return exceptions.NotImplementedError()
|
|
|
|
def add(self, data):
|
|
# TODO: error handling
|
|
json_data = json.dumps(data)
|
|
response = post(self.base_url(), 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)
|
|
|
|
def __repr__(self):
|
|
return repr(self.data)
|
|
|
|
class Organizations(Collection):
|
|
|
|
def base_url(self):
|
|
return "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()
|
|
for x in orgs:
|
|
print x
|
|
print "---"
|
|
orgs.add(dict(description="new org?", name="new org"))
|
|
print "---"
|
|
|
|
print "---"
|
|
orgs = Organizations()
|
|
for x in orgs:
|
|
print x
|
|
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print "connect failed"
|
|
sys.exit(1)
|
|
|