mirror of
https://github.com/ZwareBear/awx.git
synced 2026-03-20 07:43:35 -05:00
Some initial hammering of the python API client layer into shape.
This commit is contained in:
@@ -2,9 +2,13 @@ from django.conf import settings
|
|||||||
from django.conf.urls import *
|
from django.conf.urls import *
|
||||||
from tastypie.api import Api
|
from tastypie.api import Api
|
||||||
from lib.api.resources import Organizations
|
from lib.api.resources import Organizations
|
||||||
|
from lib.api.resources import Projects
|
||||||
|
from lib.api.resources import Users
|
||||||
|
|
||||||
v1_api = Api(api_name='v1')
|
v1_api = Api(api_name='v1')
|
||||||
v1_api.register(Organizations())
|
v1_api.register(Organizations())
|
||||||
|
v1_api.register(Projects())
|
||||||
|
v1_api.register(Users())
|
||||||
|
|
||||||
|
|
||||||
urlpatterns = patterns('',
|
urlpatterns = patterns('',
|
||||||
|
|||||||
127
lib/cli/main.py
127
lib/cli/main.py
@@ -7,19 +7,15 @@ import json
|
|||||||
# this is temporary
|
# this is temporary
|
||||||
username = os.getenv("ACOM_USER","admin")
|
username = os.getenv("ACOM_USER","admin")
|
||||||
password = os.getenv("ACOM_PASS","admin")
|
password = os.getenv("ACOM_PASS","admin")
|
||||||
print "USER=%s" % username
|
|
||||||
server = os.getenv("ACOM_SERVER","http://127.0.0.1:8000")
|
server = os.getenv("ACOM_SERVER","http://127.0.0.1:8000")
|
||||||
|
PARAMS = dict(format='json')
|
||||||
|
HEADERS = { 'Content-Type' : 'application/json' }
|
||||||
|
AUTH = HTTPBasicAuth(username, password)
|
||||||
|
|
||||||
# TODO: error handling/output/etc
|
|
||||||
# TODO: format into actual command line
|
# TODO: format into actual command line
|
||||||
|
|
||||||
PARAMS = {
|
# =============================================================
|
||||||
'format' : 'json',
|
# wrappers around URL request functions
|
||||||
}
|
|
||||||
HEADERS = {
|
|
||||||
'Content-Type' : 'application/json'
|
|
||||||
}
|
|
||||||
AUTH = HTTPBasicAuth(username, password)
|
|
||||||
|
|
||||||
def get(url_seg, expect=200):
|
def get(url_seg, expect=200):
|
||||||
resp = requests.get("%s/api/v1/%s" % (server, url_seg), auth=AUTH)
|
resp = requests.get("%s/api/v1/%s" % (server, url_seg), auth=AUTH)
|
||||||
@@ -33,15 +29,26 @@ def post(url_seg, data, expect=201):
|
|||||||
assert "POST: Expecting %s got %s: %s" % (expect, resp.status_code, resp.text)
|
assert "POST: Expecting %s got %s: %s" % (expect, resp.status_code, resp.text)
|
||||||
return resp
|
return resp
|
||||||
|
|
||||||
|
def update_item(item, data, expect=200):
|
||||||
|
resp = requests.put("%s/%s" % (server, item.resource_uri), auth=AUTH, data=data, headers=HEADERS)
|
||||||
|
if resp.status_code != expect:
|
||||||
|
assert "PUT: Expecting %s got %s: %s" % (expect, resp.status_code, resp.text)
|
||||||
|
return resp
|
||||||
|
|
||||||
|
def delete(url_seg, expect=204):
|
||||||
|
print "DELETING: %s" % url_seg
|
||||||
|
resp = requests.delete("%s/%s" % (server, url_seg), auth=AUTH, headers=HEADERS)
|
||||||
|
if resp.status_code != expect:
|
||||||
|
assert "DELETE: Expecting %s got %s: %s" % (expect, resp.status_code, resp.text)
|
||||||
|
return resp
|
||||||
|
|
||||||
|
# =============================================================
|
||||||
|
|
||||||
class Collection(object):
|
class Collection(object):
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
||||||
self.response = get(self.base_url())
|
self.response = get(self.base_url())
|
||||||
|
|
||||||
print self.response.text
|
|
||||||
print self.response.status_code
|
|
||||||
# TODO: error handling on non-200
|
|
||||||
self.data = json.loads(self.response.text)
|
self.data = json.loads(self.response.text)
|
||||||
self.meta = self.data['meta']
|
self.meta = self.data['meta']
|
||||||
self.objects = self.data['objects']
|
self.objects = self.data['objects']
|
||||||
@@ -60,6 +67,14 @@ class Collection(object):
|
|||||||
for x in self.objects:
|
for x in self.objects:
|
||||||
yield Entry(x)
|
yield Entry(x)
|
||||||
|
|
||||||
|
def print_display(self):
|
||||||
|
# TODO: paginate/flexible headers
|
||||||
|
print "%s" % self.name()
|
||||||
|
print "==="
|
||||||
|
for x in self.objects:
|
||||||
|
print Entry(x).display()
|
||||||
|
print ""
|
||||||
|
|
||||||
class Entry(object):
|
class Entry(object):
|
||||||
|
|
||||||
def __init__(self, data):
|
def __init__(self, data):
|
||||||
@@ -69,29 +84,93 @@ class Entry(object):
|
|||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return repr(self.data)
|
return repr(self.data)
|
||||||
|
|
||||||
|
def update(self, data):
|
||||||
|
json_data = json.dumps(data)
|
||||||
|
return update_item(self, json_data)
|
||||||
|
|
||||||
|
def delete(self):
|
||||||
|
''' deletes the resource '''
|
||||||
|
return delete(self.resource_uri)
|
||||||
|
|
||||||
|
def deactivate(self):
|
||||||
|
return self.update(dict(active=False))
|
||||||
|
|
||||||
|
def activate(self):
|
||||||
|
return self.update(dict(active=True))
|
||||||
|
|
||||||
|
def delete_membership(self, parent, field):
|
||||||
|
'''
|
||||||
|
removes this object from a parent collection:
|
||||||
|
example project.delete_membership(org, 'projects')
|
||||||
|
'''
|
||||||
|
parent_uri = parent.resource_uri
|
||||||
|
uri = "/".join(parent_uri, "/", field, self.data['id'])
|
||||||
|
return delete(uri)
|
||||||
|
|
||||||
|
def display(self):
|
||||||
|
keyz = self.data.keys()
|
||||||
|
keyz.sort()
|
||||||
|
lines = []
|
||||||
|
lines.append(" %20s" % self.resource_uri)
|
||||||
|
lines.append(" %20s | %s" % ('name', self.data['name']))
|
||||||
|
lines.append(" %20s | %s" % ('description', self.data['description']))
|
||||||
|
lines.append(" %20s | %s" % ('creation_date', self.data['creation_date']))
|
||||||
|
for key in keyz:
|
||||||
|
if key not in [ 'name', 'description', 'id', 'resource_uri', 'creation_date' ]:
|
||||||
|
lines.append(" %20s | %s" % (key, self.data[key]))
|
||||||
|
return "\n".join(lines)
|
||||||
|
|
||||||
|
def print_display(self):
|
||||||
|
print self.display()
|
||||||
|
|
||||||
class Organizations(Collection):
|
class Organizations(Collection):
|
||||||
|
|
||||||
|
def name(self):
|
||||||
|
return "Organizations"
|
||||||
|
|
||||||
def base_url(self):
|
def base_url(self):
|
||||||
return "organizations/"
|
return "organizations/"
|
||||||
|
|
||||||
|
class Projects(Collection):
|
||||||
|
|
||||||
|
def name(self):
|
||||||
|
return "Projects"
|
||||||
|
|
||||||
|
def base_url(self):
|
||||||
|
return "projects/"
|
||||||
|
|
||||||
|
|
||||||
#(Epdb) got.text
|
#(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/"}]}'
|
#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:
|
try:
|
||||||
print "---"
|
|
||||||
|
print "*** adding an org"
|
||||||
orgs = Organizations()
|
orgs = Organizations()
|
||||||
for x in orgs:
|
|
||||||
print x
|
|
||||||
print "---"
|
|
||||||
orgs.add(dict(description="new org?", name="new org"))
|
orgs.add(dict(description="new org?", name="new org"))
|
||||||
print "---"
|
last_org = list(Organizations())[-1]
|
||||||
|
Organizations().print_display()
|
||||||
print "---"
|
|
||||||
orgs = Organizations()
|
print "*** adding a project"
|
||||||
for x in orgs:
|
projects = Projects()
|
||||||
print x
|
projects.add(dict(description="new project?", name="new project"))
|
||||||
|
last_project = list(Projects())[-1]
|
||||||
|
|
||||||
|
last_project.update(dict(description="new project!!!!", name="new project!!!!", organization=last_org.resource_uri))
|
||||||
|
Organizations().print_display()
|
||||||
|
|
||||||
|
print "*** showing the orgs"
|
||||||
|
Organizations().print_display()
|
||||||
|
|
||||||
|
print "*** showing the projects"
|
||||||
|
Projects().print_display()
|
||||||
|
|
||||||
|
print "*** deleting all projects"
|
||||||
|
[x.delete() for x in projects]
|
||||||
|
|
||||||
|
print "*** showing the projects"
|
||||||
|
Projects().print_display()
|
||||||
|
|
||||||
except requests.exceptions.ConnectionError:
|
except requests.exceptions.ConnectionError:
|
||||||
print "connect failed"
|
print "connect failed"
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ for setting in dir(global_settings):
|
|||||||
# Build paths inside the project like this: os.path.join(BASE_DIR, ...)
|
# Build paths inside the project like this: os.path.join(BASE_DIR, ...)
|
||||||
BASE_DIR = os.path.dirname(os.path.dirname(__file__))
|
BASE_DIR = os.path.dirname(os.path.dirname(__file__))
|
||||||
|
|
||||||
DEBUG = True
|
DEBUG = False
|
||||||
TEMPLATE_DEBUG = DEBUG
|
TEMPLATE_DEBUG = DEBUG
|
||||||
|
|
||||||
ADMINS = (
|
ADMINS = (
|
||||||
|
|||||||
Reference in New Issue
Block a user