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)):
		#check for leading spaces
		if acl_list[i][:1] == " ":
			acl_list[i] = acl_list[i][1:len(acl_list[i])]
		#check for R subscriptions
		if acl_list[i][:2] == "R ":
			subscription = acl_list[i][2:len(acl_list[i])]
			subscription = "\"" + subscription + "\""
			subscribeList = subscribeList + subscription + ","
		#check for W subscriptions
		if acl_list[i][:2] == "W ":
			publish = acl_list[i][2:len(acl_list[i])]
			publish = "\"" + publish + "\""
			publishList = publishList + publish + ","
		#check for RW subscription
		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
#############################################################################