sigsci-update-agent-mode-v0.1.py

import sys, requests, os, json

# Initial setup

endpoint = 'https://dashboard.signalsciences.net/api/v0'
email = os.environ.get('SIGSCI_EMAIL')
password = os.environ.get('SIGSCI_PASSWORD')

# Set onelogin corpName for URL endpoint
corpName = 'REPLACEME'

# Set onelogin siteName for URL endpoint
siteName = 'REPLACEME'

# Authenticate

auth = requests.post(
	endpoint + '/auth',
	data = {"email": email, "password": password}
)

if auth.status_code == 401:
	print 'Invalid login.'
	sys.exit()
elif auth.status_code != 200:
	print 'Unexpected status: %s response: %s' % (auth.status_code, auth.text)
	sys.exit()

parsed_response = auth.json()
token = parsed_response['token']

print token
# Fetch list of corps

headers = {
	'Content-type': 'application/json',
	'Authorization': 'Bearer %s' % token
}

# set agentStatus ['block', 'log', 'off']
agentStatus = {
	'agentLevel': 'off'
}

agentStatus = json.dumps(agentStatus)

corps = requests.patch(endpoint + '/corps/' + corpName + '/sites/' + siteName, headers=headers, data=agentStatus)
print corps.text