zWaveApi.py

import urllib2
import urllib
import httplib
import json

class zWaveApi(object):
    '''
    zWave zWay Automation API access
        
    This is a library for Python 2.7

    from https://github.com/topcats/raspberry-zwave-python

    Methods:
        getDevices       - Gets a full List of zWave Devices
        setDeviceCommand - Sends a command to a zWave Device
    
    Init Variables:
        zUsername - Username with access to API
        zPassword - password for API
        zBaseUrl  - full url for the ZAutomation api path
    '''
    zlogin_cookie = ''
    zlogin_username = ''
    zlogin_password = ''
    zlogin_apibaseURL = 'http://localhost:8083/ZAutomation/api/v1/'
    zlogin_server = ''
    zlogin_serverpath = ''

    def __init__(self, zUsername, zPassword, zBaseUrl):
        self.zlogin_username = zUsername
        self.zlogin_password = zPassword
        self.zlogin_apibaseURL = zBaseUrl
        self.strServer = self.zlogin_apibaseURL.replace('http://', '')
        self.strServer = self.strServer[:self.strServer.index('/')]
        self.strServerPath = self.zlogin_apibaseURL.replace('http://'+self.strServer,'')

    def DoLogin(self):
        '''Does the initial authentication before anything else
        Is run only when needed, you do not need to run directly'''
        post_login = urllib.urlencode({'form': 'true', 'login': self.zlogin_username, 'password': self.zlogin_password, 'keepme': 'false', 'default_ui': 1})

        webheaders = {"Content-type": "application/x-www-form-urlencoded", "Accept": "text/plain"}
        webconn = httplib.HTTPConnection(self.strServer)
        webconn.request("POST", self.strServerPath+'login', post_login, webheaders)
        webresponse = webconn.getresponse()
        webresponseCookie = webresponse.getheader('set-cookie')#['ZWAYSession']
        webresponseCookie = webresponseCookie.replace(' ','')
        webresponseCookie = webresponseCookie[webresponseCookie.index('ZWAYSession='):].replace('ZWAYSession=','')
        webresponseCookie = webresponseCookie[:webresponseCookie.index(';')]
        self.zlogin_cookie = webresponseCookie
        webconn.close()
        return webresponse.status


    def getDevices(self):
        '''Get a list of all zWave Devices
        Return a JSON object of Devices'''
        #Double check login
        if (self.zlogin_cookie == ''):
            self.DoLogin()

        webheaders = {'Content-type': 'application/json', "Accept": "application/json", "Cookie": "ZWAYSession="+self.zlogin_cookie}
        webconn = httplib.HTTPConnection(self.strServer)
        webconn.request("GET", self.strServerPath+'devices?since=0', '', webheaders)
        webresponse = webconn.getresponse()
        webdata = webresponse.read()
        webconn.close()
        json_obj = json.loads(webdata)
        return json_obj['data']['devices']


    def setDeviceCommand(self, deviceid, newcommand):
        '''Send a Command to a zWave Device

        Init Variables:
            deviceid   - The unique zWave device ID
            newcommand - The Require command for the device
                         This depends on the device type, see below
        
        switchMultilevel : on / off / min / max / exact?level=40 / increase / decrease / update
        switchBinary     : on / off / update
        toggleButton     : on

        Returns either 0 or 1 if successful'''

        #Double check login
        if (self.zlogin_cookie == ''):
            self.DoLogin()

        #Check inputs
        if (deviceid == ''):
            return 0
        if (newcommand == ''):
            return 0

        #Do Command
        webheaders = {'Content-type': 'application/json', "Accept": "application/json", "Cookie": "ZWAYSession="+self.zlogin_cookie}
        webconn = httplib.HTTPConnection(self.strServer)
        webconn.request("GET", self.strServerPath+'devices/'+deviceid+'/command/'+newcommand, '', webheaders)
        webresponse = webconn.getresponse()
        webdata = webresponse.read()
        webconn.close()

        #Check Response
        json_obj = json.loads(webdata)
        if (json_obj['code'] == 200):
            return 1
        else:
            return 0