x
/
craton
Code Issues Proposed changes

Fake Data: Add labels to devices 

Adds labels to device and container creation
such that we can use --label query to filter
by labels.

Change-Id: Icd47cbaca0067fad66e0f3b75150653b6a5bde3d
This commit is contained in:
Sulochan Acharya 2017-03-06 15:13:19 +00:00
parent 8570377429
commit dfeb340ab5
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):