Page History

import requests
import base64

#HTTP call method
def httpCall(method,url,headers,data):
	if method == "POST":
		resp = requests.post(url, headers=headers, data=data)
	if method == "GET":
		resp = requests.get(url, headers=headers, data=data)
	if method == "PUT":
		resp = requests.put(url, headers=headers, data=data)
	
	#check for failures on httpCall	
	if resp.status_code != 200:
		print resp.json()
		print data
	
	return resp.json()

#Set URL for you Chariot installation
#chariotUrl = "http://localhost:8080"

#Create Chariot username:password dictionary for your MQTT Distributor users
#userpassword = {"admin":"password","test2":"password2","test3":"password3"}

#############################################################################
#Accept Chariot EULA
url = chariotUrl + "/eula"
headers["accept"] = "application/json;api-version=1.0"
headers["content-type"] = "application/json"
data = "{\"isAccepted\": true}"
http_response = httpCall("POST",url,headers,data)

#Chariot login using admin user to get bearer token valid for 90 minutes
url = chariotUrl + "/login"
headers["accept"] = "application/json;api-version=1.0"
headers["Authorization"] = "Basic " + base64.b64encode("admin:password")
data = {}
http_response = httpCall("POST",url,headers,data)
accessToken = http_response["access_token"]

#Read Distributor users
users = system.cirruslink.distributor.readConfig("Users")

#Create new format Access Control Lists (ACLs)
for i in range(len(users)):
	subscribeList = ""
	publishList = ""	
	userName = users[i]["Username"]
	password = list(userpassword.values())[list(userpassword.keys()).index(userName)]
	acls = users[i]["ACLs"]
	acl_list = acls.split(",")
	for i in range(len(acl_list)):
		if acl_list[i][:2] == "R ":
			subscription = acl_list[i][2:len(acl_list[i])]
			subscription = "\"" + subscription + "\""
			subscribeList = subscribeList + subscription + ","
		if acl_list[i][:2] == "W ":
			publish = acl_list[i][2:len(acl_list[i])]
			publish = "\"" + publish + "\""
			publishList = publishList + publish + ","
		if acl_list[i][:2] == "RW":
			subscription = acl_list[i][3:len(acl_list[i])]
			subscription = "\"" + subscription + "\""
			subscribeList = subscribeList + subscription + ","
			publish = acl_list[i][3:len(acl_list[i])]
			publish = "\"" + publish + "\""
			publishList = publishList + publish + ","
				
	#create Chariot user
	url = chariotUrl + "/mqttusers"
	headers["accept"] = "application/json;api-version=1.0"
	headers["content-type"] = "application/json"
	headers["Authorization"] = "Bearer " + accessToken
	data = "[{\"username\":\"" + userName + "\"," + "\"password\":\"" + password + "\"," + "\"acl\":{\"publishTopics\":[" + publishList[:-1] +"],\"subscribeTopics\":[" + subscribeList[:-1] + "]}}]"
	http_response = httpCall("POST",url,headers,data)


#Read Chariot MQtt Users
url = chariotUrl + "/mqttusers"
headers["accept"] = "application/json;api-version=1.0"
headers["Authorization"] = "Bearer " + accessToken
data = {}
chariotMqttUsers = httpCall("GET",url,headers,data)
print chariotMqttUsers
#############################################################################