mirror of
https://github.com/ansible/awx.git
synced 2026-01-11 10:00:01 -03:30
Some initial hammering of the python API client layer into shape.
This commit is contained in:
parent
0bb36d41d7
commit
b7133ca35a
@ -2,9 +2,13 @@ from django.conf import settings
|
||||
from django.conf.urls import *
|
||||
from tastypie.api import Api
|
||||
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.register(Organizations())
|
||||
v1_api.register(Projects())
|
||||
v1_api.register(Users())
|
||||
|
||||
|
||||
urlpatterns = patterns('',
|
||||
|
||||
127
lib/cli/main.py
127
lib/cli/main.py
@ -7,19 +7,15 @@ 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")
|
||||
PARAMS = dict(format='json')
|
||||
HEADERS = { 'Content-Type' : 'application/json' }
|
||||
AUTH = HTTPBasicAuth(username, password)
|
||||
|
||||
# TODO: error handling/output/etc
|
||||
# TODO: format into actual command line
|
||||
|
||||
PARAMS = {
|
||||
'format' : 'json',
|
||||
}
|
||||
HEADERS = {
|
||||
'Content-Type' : 'application/json'
|
||||
}
|
||||
AUTH = HTTPBasicAuth(username, password)
|
||||
# =============================================================
|
||||
# wrappers around URL request functions
|
||||
|
||||
def get(url_seg, expect=200):
|
||||
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)
|
||||
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):
|
||||
|
||||
def __init__(self):
|
||||
|
||||
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.meta = self.data['meta']
|
||||
self.objects = self.data['objects']
|
||||
@ -60,6 +67,14 @@ class Collection(object):
|
||||
for x in self.objects:
|
||||
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):
|
||||
|
||||
def __init__(self, data):
|
||||
@ -69,29 +84,93 @@ class Entry(object):
|
||||
def __repr__(self):
|
||||
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):
|
||||
|
||||
def name(self):
|
||||
return "Organizations"
|
||||
|
||||
def base_url(self):
|
||||
return "organizations/"
|
||||
|
||||
class Projects(Collection):
|
||||
|
||||
def name(self):
|
||||
return "Projects"
|
||||
|
||||
def base_url(self):
|
||||
return "projects/"
|
||||
|
||||
|
||||
#(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 "---"
|
||||
|
||||
print "*** adding an org"
|
||||
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
|
||||
|
||||
last_org = list(Organizations())[-1]
|
||||
Organizations().print_display()
|
||||
|
||||
print "*** adding a project"
|
||||
projects = Projects()
|
||||
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:
|
||||
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, ...)
|
||||
BASE_DIR = os.path.dirname(os.path.dirname(__file__))
|
||||
|
||||
DEBUG = True
|
||||
DEBUG = False
|
||||
TEMPLATE_DEBUG = DEBUG
|
||||
|
||||
ADMINS = (
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user