x
/
craton
Code Issues Proposed changes

Merge "Fake Data: Add labels to devices"

This commit is contained in:
Jenkins 2017-03-06 18:39:11 +00:00 committed by Gerrit Code Review
parent 61a805b185 dfeb340ab5
commit c044bed5d0
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tools/generate_fake_data.py
View File

@ -142,6 +142,15 @@ class Inventory(object):
if container_obj.status_code != 201:
raise Exception(container_obj.text)
label_url = self.url + "/hosts/%s/labels" % container_obj.json()['id']
labels = {"labels": ["container"]}
resp = requests.put(label_url, headers=self.headers,
data=json.dumps(labels), verify=False)
if resp.status_code != 200:
print("Failed to create label for host %s" % name)
print(resp.text)
def create_device(self, host, device_type, parent=None, data=None):
region_url = self.url + "/hosts"
payload = {"region_id": self.region.get("id"),
@ -169,6 +178,15 @@ class Inventory(object):
if resp.status_code != 200:
print(resp.text)
label_url = self.url + "/hosts/%s/labels" % device_obj.json()["id"]
labels = {"labels": [device_type]}
resp = requests.put(label_url, headers=self.headers,
data=json.dumps(labels), verify=False)
if resp.status_code != 200:
print("Failed to create label for host %s" % host)
print(resp.text)
return device_obj.json()
def create_network(self, name, cidr, gateway, netmask, block_type):