Automatic Device Discovery

inputDiscovery.py

import json
import sys
import urllib2

'''
Auto-authenticate Syslog TCP inputs.
Usage: python inputs.py -u user -p pass -s subdomain
'''

state = None
params = {}
for i in range(len(sys.argv)):
   arg = sys.argv[i]

   if state:
       params[state] = arg
       state = None

   if arg == '--username' or arg == '-u':
       state = 'username'

   if arg == '--password' or arg == '-p':
       state = 'password'

   if arg == '--subdomain' or arg == '-s':
       state = 'subdomain'

url = 'https://%s.loggly.com/api/inputs' % params['subdomain']
password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
password_mgr.add_password(None, url, params['username'], params['password'])

handler = urllib2.HTTPBasicAuthHandler(password_mgr)

opener = urllib2.build_opener(handler)

opener.open(url)

urllib2.install_opener(opener)

inputs = json.loads(urllib2.urlopen(url).read())

for input in inputs:
   if input['service']['name'] == 'syslogtcp':
       url = 'https://%s.loggly.com/api/inputs/%d/adddevice' % \
           (params['subdomain'], input['id'])
       response = urllib2.urlopen(url, {}).read()
       print response