def setrecipe(controldbpath, channelname, recipe, startstage=0):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set controlrecipe=\'' + recipe + '\' where name=\'' + channelname + '\'')
sqlitequery(controldbpath,
'update channels set recipestage=\'' + str(startstage) + '\' where name=\'' + channelname + '\'')
def updatelightsfromdb(database, table, CS):
import pilib
# get settings from database
query = 'select status from \'' + table + '\' where interface=\'SPI' + str(CS) + '\''
query2 = 'select name from \'' + table + '\' where interface=\'SPI' + str(CS) + '\''
# print(query)
# print(query2)
statuses = dblib.sqlitequery(database, query)
names = dblib.sqlitequery(database, query2)
# print(statuses)
# print(names)
d = {}
for status, name in zip(statuses, names):
d[name[0]] = status[0]
# print(d)
try:
setarray = [[d['SPI_RGB1_R'], d['SPI_RGB1_G'], d['SPI_RGB1_B']],
[d['SPI_RGB2_R'], d['SPI_RGB2_G'], d['SPI_RGB2_B']],
[d['SPI_RGB3_R'], d['SPI_RGB3_G'], d['SPI_RGB3_B']],
[d['SPI_RGB4_R'], d['SPI_RGB4_G'], d['SPI_RGB4_B']],
d['SPI_SC_R'], d['SPI_SC_G'], d['SPI_SC_B'],
d['SPI_SC_Y']]
# print(setarray)
except KeyError:
pass
# print('key error on indicator keys')
else:
setspilights(setarray, CS)
def updatelightsfromdb(database, table, CS):
import pilib
# get settings from database
query = 'select status from \'' + table + '\' where interface=\'SPI' + str(
CS) + '\''
query2 = 'select name from \'' + table + '\' where interface=\'SPI' + str(
CS) + '\''
# print(query)
# print(query2)
statuses = dblib.sqlitequery(database, query)
names = dblib.sqlitequery(database, query2)
# print(statuses)
# print(names)
d = {}
for status, name in zip(statuses, names):
d[name[0]] = status[0]
# print(d)
try:
setarray = [[d['SPI_RGB1_R'], d['SPI_RGB1_G'], d['SPI_RGB1_B']],
[d['SPI_RGB2_R'], d['SPI_RGB2_G'], d['SPI_RGB2_B']],
[d['SPI_RGB3_R'], d['SPI_RGB3_G'], d['SPI_RGB3_B']],
[d['SPI_RGB4_R'], d['SPI_RGB4_G'], d['SPI_RGB4_B']],
d['SPI_SC_R'], d['SPI_SC_G'], d['SPI_SC_B'], d['SPI_SC_Y']]
# print(setarray)
except KeyError:
pass
# print('key error on indicator keys')
else:
setspilights(setarray, CS)
def readhardwarefileintoversions():
from iiutilities import utility
from cupid import pilib
from iiutilities import dblib
devicefile = '/var/wwwsafe/devicedata'
try:
file = open(devicefile)
lines = file.readlines()
devicedict = {}
for line in lines:
split = line.split(':')
try:
devicedict[split[0].strip()] = split[1].strip()
except:
utility.log(pilib.dirs.logs.system, 'Device data parse error',
1, pilib.loglevels.system)
dblib.sqlitequery(
pilib.dirs.dbs.system,
dblib.makesqliteinsert('versions',
['hardware', devicedict['hardware']],
['item', 'version']))
except:
utility.log(pilib.dirs.logs.system,
'Error opening devicedata file to parse', 1,
pilib.loglevels.system)
def setmode(controldbpath, channelname, mode):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath, 'update channels set mode=\'' + mode + '\' where name=\'' + channelname + '\'')
# set action to 0 if we switch to manual
if mode == 'manual':
setaction(controldbpath, channelname, 0)
def runsendhandler(ser):
from iiutilities import dblib, datalib
from iiutilities import utility
# print('looking for message to send')
motes_db = pilib.dbs.motes
try:
last_queued_message = motes_db.get_first_time_row(
'queued', 'queuedtime')
except:
import traceback
utility.log(
pilib.dirs.logs.serial,
'Error getting queued message : {}'.format(traceback.format_exc()),
1, pilib.loglevels.serial)
else:
try:
utility.log(
pilib.dirs.logs.serial,
'Sending message : {}'.format(last_queued_message['message']),
3, pilib.loglevels.serial)
ser.write(last_queued_message['message'].encode())
# sendserialmessage(ser, lastqueuedmessage['message'])
except:
utility.log(pilib.dirs.logs.serial, 'Error sending message', 1, 1)
else:
utility.log(pilib.dirs.logs.serial, 'Success sending message', 1,
1)
conditionnames = ['queuedtime', 'message']
conditionvalues = [
last_queued_message['queuedtime'],
last_queued_message['message']
]
delquery = dblib.makedeletesinglevaluequery(
'queued', {
'conditionnames': conditionnames,
'conditionvalues': conditionvalues
})
dblib.sqlitequery(pilib.dirs.dbs.motes, delquery)
dblib.sqliteinsertsingle(pilib.dirs.dbs.motes, 'sent', [
last_queued_message['queuedtime'],
datalib.gettimestring(), last_queued_message['message']
])
dblib.size_sqlite_table(pilib.dirs.dbs.motes, 'sent', 1000)
return
def readhardwarefileintoversions():
from iiutilities import utility
from cupid import pilib
from iiutilities import dblib
devicefile = '/var/wwwsafe/devicedata'
try:
file = open(devicefile)
lines = file.readlines()
devicedict = {}
for line in lines:
split = line.split(':')
try:
devicedict[split[0].strip()] = split[1].strip()
except:
utility.log(pilib.dirs.logs.system, 'Device data parse error', 1, pilib.loglevels.system)
dblib.sqlitequery(pilib.dirs.dbs.system,
dblib.makesqliteinsert('versions', ['hardware', devicedict['hardware']], ['item', 'version']))
except:
utility.log(pilib.dirs.logs.system, 'Error opening devicedata file to parse', 1,
pilib.loglevels.system)
def set_action(self, action):
from pilib import dirs
from iiutilities.dblib import sqlitequery
sqlitequery(dirs.dbs.control,
'update channels set action = \'' + str(action) + '\' where name = \'' + channel.name + '\'')
def getsetpoint(controldbpath, channelname):
from iiutilities.dblib import sqlitequery
currentsetpoint = \
sqlitequery(controldbpath, 'select setpointvalue from channels where name=\'' + channelname + '\'')[0][0]
return currentsetpoint
def getaction(controldbpath, channelname):
from iiutilities.dblib import sqlitequery
action = sqlitequery(controldbpath, 'select action from channels where name=\'' + channelname + '\'')[0][0]
return action
def setcontrolvalue(controldbpath, channelname, controlvalue):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set controlvalue = \'' + str(controlvalue) + '\' where name = \'' + channelname + '\'')
def setcontrolinput(controldbpath, channelname, inputid):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set controlinput = \'' + inputid + '\' where name = \'' + channelname + '\'')
def updateowfsdevices(busdevices, myProxy=None, debug=False):
from cupid import pilib
from iiutilities import dblib
from iiutilities import datalib
from iiutilities import utility
# get defaults
defaults = pilib.dirs.dbs.control.read_table('defaults')
default_dict={}
for default_item in defaults:
default_dict[default_item['valuename']] = default_item['value']
# get current entries
previnputs = pilib.dirs.dbs.control.read_table('inputs')
# Make list of IDs for easy indexing
previnputids = []
for input in previnputs:
previnputids.append(input['id'])
# Iterate over devices. Determine if values exist for polltime, frequency.
# If so, update the device. If not, use defaults.
# Then determine whether we should update value or not (Read temperature)
for index, device in enumerate(busdevices):
# print(device.__dict__)
if device.sensorid in previnputids:
try:
newpollfreq = float(previnputs[previnputids.index(device.sensorid)]['pollfreq'])
except ValueError:
device.pollfreq = float(default_dict['inputpollfreq'])
else:
if newpollfreq >= 0:
device.pollfreq = float(previnputs[previnputids.index(device.sensorid)]['pollfreq'])
else:
device.pollfreq = float(default_dict['inputpollfreq'])
device.ontime = previnputs[previnputids.index(device.sensorid)]['ontime']
device.offtime = previnputs[previnputids.index(device.sensorid)]['offtime']
device.polltime = previnputs[previnputids.index(device.sensorid)]['polltime']
device.value = previnputs[previnputids.index(device.sensorid)]['value']
device.log_options = previnputs[previnputids.index(device.sensorid)]['log_options']
else:
device.pollfreq = float(default_dict['inputpollfreq'])
device.ontime = ''
device.offtime = ''
device.polltime = ''
device.value = ''
"""
We're going to set a name because calling things by their ids is getting
a bit ridiculous, but we can't have empty name fields if we rely on them
being there. They need to be unique, so we'll name them by type and increment them
Not really sure why this is conditional?
"""
if device.type in ['DS18B20', 'DS1825']:
# Get name if one exists
name = dblib.sqlitedatumquery(pilib.dirs.dbs.control, 'select name from ioinfo where id=\'' + device.sensorid + '\'')
# If doesn't exist, check to see if proposed name exists. If it doesn't, add it.
# If it does, keep trying.
if name == '':
for rangeindex in range(100):
# check to see if name exists
name = device.type + '-' + str(int(index + 1))
# print(name)
foundid = dblib.sqlitedatumquery(pilib.dirs.dbs.control, 'select id from ioinfo where name=\'' + name + '\'')
# print('foundid' + foundid)
if foundid:
pass
else:
dblib.sqlitequery(pilib.dirs.dbs.control, dblib.makesqliteinsert('ioinfo', valuelist=[device.sensorid, name],
valuenames=['id', 'name']))
break
device.name = name
device.time_since_last = datalib.timestringtoseconds(datalib.gettimestring()) - datalib.timestringtoseconds(device.polltime, defaulttozero=True)
# Is it time to read temperature?
if device.time_since_last > device.pollfreq:
utility.log(pilib.dirs.logs.io, 'reading temperature [' + device.name + '][' + device.id + ']' , 9, pilib.loglevels.io)
device.readprop('temperature', myProxy)
device.polltime = datalib.gettimestring()
device.value = device.temperature.decode('utf-8')
else:
utility.log(pilib.dirs.logs.io, 'not time to poll', 9, pilib.loglevels.io, )
# print('not time to poll')
device.unit = 'F'
# We update the device and send them back for other purposes.
busdevices[index] = device
return busdevices
def getmode(controldbpath, channelname):
from iiutilities.dblib import sqlitequery
mode = sqlitequery(controldbpath, 'select mode from channels where name=\'' + channelname + '\'')[0][0]
return mode
def setalgorithm(controldbpath, channelname, algorithm):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set controlalgorithm=\'' + algorithm + '\' where name=\'' + channelname + '\'')
def set_setpoint(self, setpointvalue):
from pilib import dirs
from iiutilities.dblib import sqlitequery
sqlitequery(dirs.dbs.control, 'update channels set setpointvalue=\'' + str(setpointvalue) + '\' where name=\'' + channel.name + '\'')
def runalgorithm(controldbpath, recipedbpath, channelname):
from iiutilities.datalib import timestringtoseconds, gettimestring
from iiutilities.dblib import sqlitequery, datarowtodict
from iiutilities import dblib
import time
message = ''
# get our details of our channel
# controldb = dblib.sqliteDatabase(controldbpath)
# controldb.read_table('channels', condition="name='{}'".format(channelname), queue=True)
channeldata = sqlitequery(controldbpath, 'select * from channels where name=' + "'" + channelname + "'")[0]
channeldict = datarowtodict(controldbpath, 'channels', channeldata)
# check to see if we are running a recipe
controlrecipename = channeldict['controlrecipe']
if controlrecipename and controlrecipename != 'none':
# Get recipe details
# If recipes get too big, we'll just get
# a couple stages. For now, we make a
# dictionary array
#print('we are in recipe ' + controlrecipename)
#print(dirs.dbs.session)
recipedata = sqlitequery(recipedbpath, 'select * from \'' + controlrecipename + '\'')
recipedictarray = []
for stage in recipedata:
recipedict = datarowtodict(recipedbpath, controlrecipename, stage)
recipedictarray.append(recipedict)
# get current stage
currentstagenumber = int(channeldict['recipestage'])
#print('current stage is ' + str(currentstagenumber) )
# Get data for current stage
stagefound = False
for stage in recipedictarray:
if int(stage['stagenumber']) == currentstagenumber:
currentstage = stage
stagefound = True
break
if stagefound:
#print("stage found")
pass
else:
print('error. stage not found.')
# Check to see if we need to move to next stage
currenttime = time.time()
#print('Time')
#print(currenttime)
#print(gettimestring(currenttime))
if currentstagenumber == 0 or currenttime - timestringtoseconds(channeldict['recipestagestarttime']) > int(
currentstage['stagelength']):
print('stage time expired for stage ' + str(currentstagenumber) + '. Checking on stage advance. ')
# Advance stage if there is another stage. Otherwise
# update channel to be off a recipe. We assume explicitly
# that the stages are sequential integers.
nextstagenumber = currentstagenumber + 1
# look for next stage
stagefound = False
for stage in recipedictarray:
if int(stage['stagenumber']) == nextstagenumber:
nextstage = stage
stagefound = True
break
if stagefound:
print(' Next stage was found. Setting next stage. ')
if currentstagenumber == 0:
print("Stagenumber is 0. Setting recipe start time. ")
# Set recipe start time
sqlitequery(controldbpath, 'update channels set recipestarttime=\'' + gettimestring(
currenttime) + '\' where name=\'' + channelname + '\'')
# Set stage to new stage number
sqlitequery(controldbpath, 'update channels set recipestage=\'' + str(
nextstagenumber) + '\' where name=\'' + channelname + '\'')
# Set setpointvalue
sqlitequery(controldbpath, 'update channels set setpointvalue=\'' + str(
nextstage['setpointvalue']) + '\' where name=\'' + channelname + '\'')
# Set stage start time to now
sqlitequery(controldbpath, 'update channels set recipestagestarttime=\'' + gettimestring(
currenttime) + '\' where name=\'' + channelname + '\'')
# Set new controlalgorithm
sqlitequery(controldbpath, 'update channels set controlalgorithm=\'' + nextstage[
'controlalgorithm'] + '\' where name=\'' + channelname + '\'')
else:
# Take channel off recipe
sqlitequery(controldbpath,
'update channels set controlrecipe=\'none\' where name=\'' + channelname + '\'')
sqlitequery(controldbpath, 'update channels set recipestate=\'0\' where name=\'' + channelname + '\'')
sqlitequery(controldbpath,
'update channels set recipestage=\'0\' where name=\'' + channelname + '\'')
# if lengthmode is setpoint
# get current stage
# check stage start against stage length
# and current time
# move to next stage if time and revise setpoint
# adjust setpoint based on stage
# set action based on setpoint
else:
# make sure we're not on recipe and on stage 0
sqlitequery(controldbpath,
'update channels set controlrecipe=\'none\' where name=\'' + channelname + '\'')
sqlitequery(controldbpath, 'update channels set recipestate=\'0\' where name=\'' + channelname + '\'')
sqlitequery(controldbpath,
'update channels set recipestage=\'0\' where name=\'' + channelname + '\'')
algorithm = channeldict['controlalgorithm']
setpointvalue = float(channeldict['setpointvalue'])
controlvalue = float(channeldict['controlvalue'])
algorithmrows = sqlitequery(controldbpath, 'select * from controlalgorithms where name=' + "'" + algorithm + "'")
algorithmrow = algorithmrows[0]
algorithm = datarowtodict(controldbpath, 'controlalgorithms', algorithmrow)
algtype = algorithm['type']
if algtype == 'on/off with deadband':
#print(type)
deadbandhigh = algorithm['deadbandhigh']
deadbandlow = algorithm['deadbandlow']
if setpointvalue > (controlvalue + deadbandhigh):
action = 100
elif setpointvalue < (controlvalue - deadbandlow):
action = -100
else:
action = 0
#print('setpoint' + str(setpoint))
#print('controlvalue' + str(controlvalue))
#print(action)
#print(message)
return [action, message]
def updateowfsdevices(busdevices, myProxy=None, debug=False):
from cupid import pilib
from iiutilities import dblib
from iiutilities import datalib
from iiutilities import utility
# get defaults
defaults = pilib.dirs.dbs.control.read_table('defaults')
default_dict = {}
for default_item in defaults:
default_dict[default_item['valuename']] = default_item['value']
# get current entries
previnputs = pilib.dirs.dbs.control.read_table('inputs')
# Make list of IDs for easy indexing
previnputids = []
for input in previnputs:
previnputids.append(input['id'])
# Iterate over devices. Determine if values exist for polltime, frequency.
# If so, update the device. If not, use defaults.
# Then determine whether we should update value or not (Read temperature)
for index, device in enumerate(busdevices):
# print(device.__dict__)
if device.sensorid in previnputids:
try:
newpollfreq = float(previnputs[previnputids.index(
device.sensorid)]['pollfreq'])
except ValueError:
device.pollfreq = float(default_dict['inputpollfreq'])
else:
if newpollfreq >= 0:
device.pollfreq = float(previnputs[previnputids.index(
device.sensorid)]['pollfreq'])
else:
device.pollfreq = float(default_dict['inputpollfreq'])
device.ontime = previnputs[previnputids.index(
device.sensorid)]['ontime']
device.offtime = previnputs[previnputids.index(
device.sensorid)]['offtime']
device.polltime = previnputs[previnputids.index(
device.sensorid)]['polltime']
device.value = previnputs[previnputids.index(
device.sensorid)]['value']
device.log_options = previnputs[previnputids.index(
device.sensorid)]['log_options']
else:
device.pollfreq = float(default_dict['inputpollfreq'])
device.ontime = ''
device.offtime = ''
device.polltime = ''
device.value = ''
"""
We're going to set a name because calling things by their ids is getting
a bit ridiculous, but we can't have empty name fields if we rely on them
being there. They need to be unique, so we'll name them by type and increment them
Not really sure why this is conditional?
"""
if device.type in ['DS18B20', 'DS1825']:
# Get name if one exists
name = dblib.sqlitedatumquery(
pilib.dirs.dbs.control,
'select name from ioinfo where id=\'' + device.sensorid + '\'')
# If doesn't exist, check to see if proposed name exists. If it doesn't, add it.
# If it does, keep trying.
if name == '':
for rangeindex in range(100):
# check to see if name exists
name = device.type + '-' + str(int(index + 1))
# print(name)
foundid = dblib.sqlitedatumquery(
pilib.dirs.dbs.control,
'select id from ioinfo where name=\'' + name + '\'')
# print('foundid' + foundid)
if foundid:
pass
else:
dblib.sqlitequery(
pilib.dirs.dbs.control,
dblib.makesqliteinsert(
'ioinfo',
valuelist=[device.sensorid, name],
valuenames=['id', 'name']))
break
device.name = name
device.time_since_last = datalib.timestringtoseconds(
datalib.gettimestring()) - datalib.timestringtoseconds(
device.polltime, defaulttozero=True)
# Is it time to read temperature?
if device.time_since_last > device.pollfreq:
utility.log(
pilib.dirs.logs.io, 'reading temperature [' + device.name +
'][' + device.id + ']', 9, pilib.loglevels.io)
device.readprop('temperature', myProxy)
device.polltime = datalib.gettimestring()
device.value = device.temperature.decode('utf-8')
else:
utility.log(
pilib.dirs.logs.io,
'not time to poll',
9,
pilib.loglevels.io,
)
# print('not time to poll')
device.unit = 'F'
# We update the device and send them back for other purposes.
busdevices[index] = device
return busdevices
def application(environ, start_response):
import cgi
import json
import os, sys, inspect
# Set top folder to allow import of modules
top_folder = os.path.split(
os.path.realpath(
os.path.abspath(
os.path.split(inspect.getfile(inspect.currentframe()))[0])))[0]
if top_folder not in sys.path:
sys.path.insert(0, top_folder)
from cupid import pilib, controllib
from iiutilities import dblib, utility, datalib
# post_env = environ.copy()
# post_env['QUERY_STRING'] = ''
# post = cgi.FieldStorage(
# fp=environ['wsgi.input'],
# environ=post_env,
# keep_blank_values=True
# )
#
# formname=post.getvalue('name')
# output = {}
# output['message'] = 'Output Message: '
# for k in post.keys():
# d[k] = post.getvalue(k)
try:
request_body_size = int(environ.get('CONTENT_LENGTH', 0))
except ValueError:
request_body_size = 0
request_body = environ['wsgi.input'].read(request_body_size)
post = json.loads(request_body.decode('utf-8'))
output = {}
output['message'] = ''
status = '200 OK'
wsgiauth = True
authverified = False
if wsgiauth:
# Verfiy that session login information is legit: hashed password, with salt and username, match
# hash stored in database.
import hashlib
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.users)
if 'username' in post and post['username']:
output['message'] += 'Session user is ' + post['username'] + '. '
else:
output['message'] += 'No session user found. '
post['username'] = ''
if post['username']:
try:
condition = "name='" + post['username'] + "'"
user_data = safe_database.read_table_row(
'users', condition=condition)[0]
except:
output[
'message'] += 'Error in user sqlite query for session user "' + post[
'username'] + '". '
output[
'message'] += 'Condition: ' + condition + '. Path: ' + pilib.dirs.dbs.safe
user_data = {'accesskeywords': 'demo', 'admin': False}
else:
# Get session hpass to verify credentials
hashedpassword = post['hpass']
hname = hashlib.new('sha1')
hname.update(post['username'])
hashedname = hname.hexdigest()
hentry = hashlib.new('md5')
hentry.update(hashedname + pilib.salt + hashedpassword)
hashedentry = hentry.hexdigest()
if hashedentry == user_data['password']:
# successful auth
output['message'] += 'Password verified. '
authverified = True
# TODO: implement usermeta
else:
# Demo status
authverified = True
user_data = {'authlevel': 0}
else:
output['message'] += 'WSGI authorization not enabled. '
if authverified or not wsgiauth:
output['authorized'] = True
try:
action = post['action']
except KeyError:
output['message'] = 'no action in request'
action = ''
else:
output['message'] += '{} action keyword found'.format(action)
if output['authorized'] and action:
output['action_allowed'] = pilib.check_action_auths(
action, user_data['authlevel'])
else:
output['action_allowed'] = False
if output['authorized'] and output['action_allowed']:
output['message'] += 'Found action. '
if action == 'testdbvn':
from iiutilities.dblib import dbvntovalue
try:
output['data'] = dbvntovalue(post['dbvn'])
except:
output['message'] += 'Error in dbvn evaluation. '
output['data'] = 'error'
else:
output['message'] += 'Seems to have worked out. '
elif action == 'testlogical':
from iiutilities.datalib import evaldbvnformula
try:
output['data'] = evaldbvnformula(post['logical'])
except:
output['message'] += 'Error in logical evaluation. '
output['data'] = 'error'
else:
output['message'] += 'Seems to have worked out. '
elif action == 'testmodule':
output['message'] += 'Testing module: '
if 'modulename' in post:
import cupid.cupidunittests
output['message'] += post['modulename']
output['data'] = cupid.cupidunittests.testmodule(
post['modulename'])
else:
output['message'] += 'Modulename not found. '
elif action == 'testfunction':
output['message'] += 'Testing function: '
if 'testname' in post:
import cupid.cupidunittests
output['message'] += post['testname']
# output['data'] = cupid.tests.testfunction(d['testname'])
output['data'] = cupid.cupidunittests.testfunction(
post['testname'])
# output['data'] = str(cupid.tests.testfunction('systemstatus'))
else:
output['message'] += 'Testname not found. '
elif action == 'modifychannelalarm':
controllib.handle_modify_channel_alarm(post, output)
from cupid.actions import processactions
# process only this action.
processactions(name=post['actionname'])
elif action == 'modifychannel':
controllib.handle_modify_channel(post, output)
elif action == 'getalarmscount':
control_db = dblib.sqliteDatabase(pilib.dirs.dbs.control)
actions = control_db.read_table('actions')
output['data'] = {
'totalalarms': len(actions),
'channelalarms': 0,
'activealarms': 0,
'activechannelalarms': 0
}
for action in actions:
if action['conditiontype'] == 'channel':
output['data']['channelalarms'] += 1
if action['active']:
output['data']['activechannelalarms'] += 1
if action['active']:
output['data']['activealarms'] += 1
elif action == 'copy_log_to_archive':
pilib.app_copy_log_to_archive(post, output)
elif action == 'getlogscount':
logtablenames = dblib.sqliteDatabase(
pilib.dirs.dbs.log).get_table_names()
output['data'] = {'logscount': len(logtablenames)}
elif action == 'test_action':
output['message'] += 'Testing action. '
controldb = dblib.sqliteDatabase(pilib.dirs.dbs.control)
actiondict = controldb.read_table('actions',
condition='"name"=\'' +
post['actionname'] + "'")[0]
from cupid.actions import action
test_action = action(actiondict)
test_action.test()
elif action == 'update_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
safe_database.set_single_value('wireless', 'password',
post['password'],
"SSID='" + post['ssid'] + "'")
elif action == 'add_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
insert = {'SSID': post['ssid'], 'auto': 1, 'priority': 1}
if 'password' in post:
insert['password'] = post['password']
safe_database.insert('wireless', insert)
elif action == 'delete_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
safe_database.delete('wireless', "SSID='" + post['ssid'] + "'")
# elif action == 'dump':
# # this has to go.
# if 'database' in d:
# dbpath = pilib.dbnametopath(d['database'])
# if dbpath:
# if 'tablelist' in d and 'outputfile' in d:
# dbpath = pilib.dbnametopath(d['database'])
# dblib.sqlitedatadump(dbpath, d['tablelist'], d['outputfile'])
# output['message'] = 'data dumped'
# elif 'tablename' in d and 'outputfile' in d:
# dblib.sqlitedatadump(dbpath, [d['tablename']], d['outputfile'])
# output['message'] = 'data dumped. '
# else:
# output['message'] += 'keys not present for dump. '
# else:
# output['message'] += 'keys not present for dump. '
# else:
# output['message'] += 'keys not present for dump. '
elif action in ['userdelete', 'useradd', 'usermodify']:
"""
This needs to be consolidate with the other useradd, modify algorithm written already.
Probably do this when we update the user permissions interface.
"""
# Ensure that we are authorized for this action
if action == 'userdelete':
try:
dblib.sqlitequery(
pilib.dirs.dbs.users,
"delete from users where name='" +
post['usertodelete'] + "'")
except:
output['message'] += 'Error in delete query. '
else:
output['message'] += 'Successful delete query. '
elif action == 'usermodify':
if 'usertomodify' in post:
querylist = []
if 'newpass' in post:
from pilib import salt
# Get session hpass to verify credentials
hashedpassword = post['newpass']
hname = hashlib.new('sha1')
hname.update(post['usertomodify'])
hashedname = hname.hexdigest()
hentry = hashlib.new('md5')
hentry.update(hashedname + salt + hashedpassword)
hashedentry = hentry.hexdigest()
querylist.append('update users set password='******'" +
post['usertomodify'] + "'")
if 'newemail' in post:
querylist.append("update users set email='" +
post['newemail'] + "' where name='" +
post['usertomodify'] + "'")
if 'newauthlevel' in post:
querylist.append("update users set authlevel='" +
post['newauthlevel'] +
"' where name='" +
post['usertomodify'] + "'")
try:
dblib.sqlitemultquery(pilib.dirs.dbs.users, querylist)
except:
output[
'message'] += 'Error in modify/add query: ' + ",".join(
querylist)
else:
output[
'message'] += 'Successful modify/add query. ' + ",".join(
querylist)
else:
output['message'] += 'Need usertomodify in query. '
elif action == 'useradd':
try:
username = post['newusername']
except:
username = '******'
try:
newemail = post['newemail']
except:
newemail = '*****@*****.**'
try:
newauthlevel = post['newauthlevel']
except:
newauthlevel = 0
query = "insert into users values(NULL,'" + username + "','','" + newemail + "',''," + str(
newauthlevel) + ")"
try:
dblib.sqlitequery(pilib.dirs.dbs.users, query)
except:
output[
'message'] += "Error in useradd sqlite query: " + query + ' . '
else:
output['message'] += "Successful query: " + query + ' . '
elif action == 'getfiletext':
try:
filepath = post['filepath']
if 'numlines' in post:
numlines = int(post['numlines'])
else:
numlines = 9999
output['message'] += 'Using numlines: ' + str(
numlines) + ' for read action. '
if 'startposition' in post:
startposition = post['startposition']
else:
startposition = 'end'
output[
'message'] += 'Reading from position ' + startposition + '. '
except KeyError:
output[
'message'] += 'Sufficient keys for action getfile text do not exist. '
except:
output['message'] += 'Uncaught error in getfiletext. '
else:
try:
file = open(filepath)
lines = file.readlines()
except:
output[
'message'] += 'Error reading file in getfiletext action. '
else:
output['data'] = []
if startposition == 'end':
try:
output['data'] = datalib.tail(file, numlines)[0]
except:
output['message'] += 'Error in tail read. '
else:
linecount = 0
for line in lines:
linecount += 1
if linecount > numlines:
break
else:
output['data'].append(line)
elif action == 'getmbtcpdata':
try:
clientIP = post['clientIP']
register = post['register']
length = post['length']
except KeyError:
output[
'message'] += 'Sufficient keys do not exist for the command. Requires clientIP, register, and length. '
else:
from iiutilities.netfun import readMBcodedaddresses
# try:
output['response'] = readMBcodedaddresses(
clientIP, int(register), int(length))
elif action == 'queuemessage':
output['message'] += 'Queue message. '
if 'message' in post:
try:
dblib.sqliteinsertsingle(
pilib.dirs.dbs.motes, 'queuedmessages',
[datalib.gettimestring(), post['message']])
except:
import traceback
exc_type, exc_value, exc_traceback = sys.exc_info()
output[
'message'] += 'Error in queue insert query: {}. '.format(
traceback.format_exc())
else:
output['message'] += 'Message insert successful'
else:
output['message'] += 'No message present. '
elif action == 'setsystemflag' and 'systemflag' in post:
database = pilib.dirs.dbs.system
dblib.setsinglevalue(database, 'systemflags', 'value', 1,
"name=\'" + post['systemflag'] + "'")
elif action == 'rundaemon':
from cupiddaemon import rundaemon
rundaemon()
# TODO: Eliminate this scary thing.
elif action == 'setvalue':
utility.log(pilib.dirs.logs.control, "Setting value in wsgi", 1, 1)
# we use the auxiliary 'setsinglecontrolvalue' to add additional actions to update
if all(k in post
for k in ('database', 'table', 'valuename', 'value')):
dbpath = pilib.dbnametopath(post['database'])
if dbpath:
output[
'message'] += 'Carrying out setvalue for value ' + post[
'valuename'] + ' on ' + post[
'table'] + ' in ' + dbpath
if 'condition' in post:
pilib.setsinglecontrolvalue(dbpath, post['table'],
post['valuename'],
post['value'],
post['condition'])
elif 'index' in post:
condition = 'rowid= ' + post['index']
pilib.setsinglecontrolvalue(dbpath, post['table'],
post['valuename'],
post['value'], condition)
else:
pilib.setsinglecontrolvalue(dbpath, post['table'],
post['valuename'],
post['value'])
else:
output[
'message'] += 'Problem translating dbpath from friendly name: ' + post[
'database']
else:
output['message'] += 'Insufficient data for setvalue '
elif action == 'updateioinfo':
if all(k in post for k in ['database', 'ioid', 'value']):
query = dblib.makesqliteinsert('ioinfo',
[post['ioid'], post['value']],
['id', 'name'])
try:
dblib.sqlitequery(pilib.dirs.dbs.control, query)
except:
output[
'message'] += 'Error in updateioinfo query execution: ' + query + '. into database: ' + pilib.dirs.dbs.control
output['message'] += 'ioid: ' + post['ioid'] + ' . '
else:
output['message'] += 'Executed updateioinfo query. '
else:
output[
'message'] += 'Insufficient data for updateioinfo query ! '
# TODO: properly incorporate and test channel class functions here, and then sub it.
elif action == 'modify_channel':
controllib.app_modify_channel(post, output)
elif action == 'deletechannelbyname' and 'database' in post and 'channelname' in post:
dbpath = pilib.dbnametopath(post['database'])
dblib.sqlitequery(
dbpath, 'delete channelname from channels where name=\"' +
post['channelname'] + '\"')
elif action == 'updatecameraimage':
output['message'] += 'Take camera image keyword. '
import cupid.camera
if 'width' in post:
width = post['width']
else:
width = 800
try:
values = cupid.camera.takesnap(width=width)
except:
output['message'] += 'Error taking image. '
else:
output['message'] += 'Appears successful. Path : ' + values[
'imagepath'] + '. Timestamp : ' + values['timestamp'] + '. '
output['data'] = values
elif action == 'getcurrentcamtimestamp':
output['message'] += 'getcurrentcamtimestamp keyword found. '
try:
with open('/var/www/webcam/images/current.jpg.timestamp') as f:
data = f.read()
except:
output['message'] += 'Error reading file as requested. '
else:
output['data'] = data
else:
output[
'message'] += 'Action keyword present(' + action + '), but not handled. '
else:
output[
'message'] += 'Authentication unsuccessful or action not authorized.'
status = '401 Not Authorized'
foutput = json.dumps(output, indent=1)
response_headers = [('Content-type', 'application/json')]
start_response(status, response_headers)
return [foutput]
def onact(self):
from iiutilities import dblib, datalib, utility
from cupid import pilib
if self.actiontype == 'email':
# process email action
self.statusmsg += 'Processing email alert. '
email = self.actiondetail
# Special messages
if self.conditiontype == 'channel':
message = 'Channel alarm for ' + self.name + ' is active with value of ' + str(self.value) + '. '
if 'PV_low' in self.actiondatadict:
message += 'Low alarm: ' + str(self.actiondatadict['PV_low'] + '. ')
if 'PV_high' in self.actiondatadict:
message += 'High alarm: ' + str(self.actiondatadict['PV_high'] + '. ')
elif self.conditiontype == 'value':
# message = 'Alert is active for ' + self.name + '. Criterion ' + self.variablename + ' in ' + self.tablename + ' has value ' + str(self.variablevalue) + ' with a criterion of ' + str(self.criterion) + ' with an operator of ' + self.operator + '. This alarm status has been on since ' + self.ontime + '.'
message = 'Alert for alarm ' + self.name + ' . On time of ' + self.ontime + '. Current time of ' \
+ datalib.gettimestring()
message += ' Value: ' + str(self.value) + self.actiondatadict['operator'] + str(self.actiondatadict['criterion'])
else:
# message = 'Alert is active for ' + self.name + '. Criterion ' + self.variablename + ' in ' + self.tablename + ' has value ' + str(self.variablevalue) + ' with a criterion of ' + str(self.criterion) + ' with an operator of ' + self.operator + '. This alarm status has been on since ' + self.ontime + '.'
message = 'Alert for alarm ' + self.name + ' . On time of ' + self.ontime + '. Current time of ' \
+ datalib.gettimestring()
import socket
hostname = socket.gethostname()
subject = 'CuPID ' + hostname + ' Alert : Alarm On - ' + self.name
try:
actionmail = utility.gmail(message=message, subject=subject, recipient=email)
actionmail.send()
except:
self.statusmsg += 'Error sending email. '
else:
self.statusmsg += 'Mail sent. '
elif self.actiontype == 'indicator':
# process indicator action
self.statusmsg += 'Processing indicator on action. '
indicatorname = self.actiondetail
dblib.sqlitequery(pilib.dirs.dbs.control, 'update indicators set status=1 where name = \'' + indicatorname + '\'')
elif self.actiontype == 'output':
self.statusmsg += 'Processing output on action. '
dblib.setsinglevalue(pilib.dirs.dbs.control, 'outputs', 'value', '1', condition='"id"=\'' + self.actiondetail + "'")
elif self.actiontype == 'mote_command':
settings = {
'no_duplicates':True,
'retries':5
}
settings.update(self.actiondatadict)
self.statusmsg += 'Processing command on action. '
destination = self.actiondatadict['destination']
message = self.actiondatadict['message']
command_id = '{}_{}'.format(destination, message)
command = {'queuedtime': datalib.gettimestring(), 'destination': destination,
'status': 'new', 'message': message, 'commandid':command_id,
}
command['options'] = json.dumps({'retries':settings['retries']})
insert = True
if settings['no_duplicates']:
# Check to see if commands exist with our command id.
condition = "commandid='{}'".format(command_id)
matching_commands = pilib.dbs.motes.read_table('commands',condition=condition)
if matching_commands:
self.statusmsg += '{} matching commands are already queued. Not inserting. '.format(len(matching_commands))
insert = False
if insert:
self.statusmsg += 'Inserting command. '
pilib.dbs.motes.settings['quiet'] = False
pilib.dbs.motes.insert('commands',command)
# This should be the generic handler that we migrate to
elif self.actiontype == 'setvalue':
# to set a value, we need at minimum:
# dbname, tablename, valuename, setmethod and either:
# setmethod = increment, incrementvalue=1
# setmethod = value
dbvndict = datalib.parsedbvn(self.actiondetail)
dbpath = pilib.dbnametopath(dbvndict['dbname'])
# Special set formula?
if 'setvalueformula' in self.actiondatadict:
# Stuff that we don't know yet.
dblib.setsinglevalue(dbpath, dbvndict['tablename'], dbvndict['valuename'], 'formulastuff here', dbvndict['condition'])
else:
""" TODO: Fix this hack. We cannot currently single quote in the database entry because it breaks the reinsert.
So for now, we have to add quotes on either side of the string literal before executing the sqlite query. """
if dbvndict['condition']:
querycondition = dbvndict['condition'].split('=')[0] + "='" + dbvndict['condition'].split('=')[1] + "'"
# print('FIXED CONDITION')
# print(querycondition)
else:
querycondition = None
dblib.setsinglevalue(dbpath, dbvndict['tablename'], dbvndict['valuename'], '1', querycondition)
def setnegout(controldbpath, channelname, outputname):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set negativeoutput=\'' + outputname + '\' where name=\'' + channelname + '\'')
def setchanneloutputsenabled(controldbpath, channelname, newstatus):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set outputsenabled=\'' + newstatus + '\' where name=\'' + channelname + '\'')
def set_control_input(self, inputid):
from pilib import dirs
from iiutilities.dblib import sqlitequery
sqlitequery(dirs.dbs.control,
'update channels set controlinput = \'' + inputid + '\' where name = \'' + channel.name + '\'')
def application(environ, start_response):
import cgi
import json
import os, sys, inspect
# Set top folder to allow import of modules
top_folder = os.path.split(os.path.realpath(os.path.abspath(os.path.split(inspect.getfile( inspect.currentframe() ))[0])))[0]
if top_folder not in sys.path:
sys.path.insert(0,top_folder)
from cupid import pilib, controllib
from iiutilities import dblib, utility, datalib
# post_env = environ.copy()
# post_env['QUERY_STRING'] = ''
# post = cgi.FieldStorage(
# fp=environ['wsgi.input'],
# environ=post_env,
# keep_blank_values=True
# )
#
# formname=post.getvalue('name')
# output = {}
# output['message'] = 'Output Message: '
# for k in post.keys():
# d[k] = post.getvalue(k)
try:
request_body_size = int(environ.get('CONTENT_LENGTH', 0))
except ValueError:
request_body_size = 0
request_body = environ['wsgi.input'].read(request_body_size)
post = json.loads(request_body.decode('utf-8'))
output = {}
output['message'] = ''
status = '200 OK'
wsgiauth = True
authverified = False
if wsgiauth:
# Verfiy that session login information is legit: hashed password, with salt and username, match
# hash stored in database.
import hashlib
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.users)
if 'username' in post and post['username']:
output['message'] += 'Session user is ' + post['username'] + '. '
else:
output['message'] += 'No session user found. '
post['username'] = ''
if post['username']:
try:
condition = "name='" + post['username'] + "'"
user_data = safe_database.read_table_row('users', condition=condition)[0]
except:
output['message'] += 'Error in user sqlite query for session user "' + post['username'] + '". '
output['message'] += 'Condition: ' + condition + '. Path: ' + pilib.dirs.dbs.safe
user_data = {'accesskeywords': 'demo', 'admin': False}
else:
# Get session hpass to verify credentials
hashedpassword = post['hpass']
hname = hashlib.new('sha1')
hname.update(post['username'])
hashedname = hname.hexdigest()
hentry = hashlib.new('md5')
hentry.update(hashedname + pilib.salt + hashedpassword)
hashedentry = hentry.hexdigest()
if hashedentry == user_data['password']:
# successful auth
output['message'] += 'Password verified. '
authverified = True
# TODO: implement usermeta
else:
# Demo status
authverified = True
user_data = {'authlevel':0}
else:
output['message'] += 'WSGI authorization not enabled. '
if authverified or not wsgiauth:
output['authorized'] = True
try:
action = post['action']
except KeyError:
output['message'] = 'no action in request'
action = ''
else:
output['message'] += '{} action keyword found'.format(action)
if output['authorized'] and action:
output['action_allowed'] = pilib.check_action_auths(action, user_data['authlevel'])
else:
output['action_allowed'] = False
if output['authorized'] and output['action_allowed']:
output['message'] += 'Found action. '
if action == 'testdbvn':
from iiutilities.dblib import dbvntovalue
try:
output['data'] = dbvntovalue(post['dbvn'])
except:
output['message'] += 'Error in dbvn evaluation. '
output['data'] = 'error'
else:
output['message'] += 'Seems to have worked out. '
elif action == 'testlogical':
from iiutilities.datalib import evaldbvnformula
try:
output['data'] = evaldbvnformula(post['logical'])
except:
output['message'] += 'Error in logical evaluation. '
output['data'] = 'error'
else:
output['message'] += 'Seems to have worked out. '
elif action == 'testmodule':
output['message'] += 'Testing module: '
if 'modulename' in post:
import cupid.cupidunittests
output['message'] += post['modulename']
output['data'] = cupid.cupidunittests.testmodule(post['modulename'])
else:
output['message'] += 'Modulename not found. '
elif action == 'testfunction':
output['message'] += 'Testing function: '
if 'testname' in post:
import cupid.cupidunittests
output['message'] += post['testname']
# output['data'] = cupid.tests.testfunction(d['testname'])
output['data'] = cupid.cupidunittests.testfunction(post['testname'])
# output['data'] = str(cupid.tests.testfunction('systemstatus'))
else:
output['message'] += 'Testname not found. '
elif action == 'modifychannelalarm':
controllib.handle_modify_channel_alarm(post, output)
from cupid.actions import processactions
# process only this action.
processactions(name=post['actionname'])
elif action == 'modifychannel':
controllib.handle_modify_channel(post, output)
elif action == 'getalarmscount':
control_db = dblib.sqliteDatabase(pilib.dirs.dbs.control)
actions = control_db.read_table('actions')
output['data'] = {'totalalarms':len(actions),'channelalarms':0, 'activealarms':0, 'activechannelalarms':0}
for action in actions:
if action['conditiontype'] == 'channel':
output['data']['channelalarms'] += 1
if action['active']:
output['data']['activechannelalarms'] += 1
if action['active']:
output['data']['activealarms'] += 1
elif action == 'copy_log_to_archive':
pilib.app_copy_log_to_archive(post, output)
elif action == 'getlogscount':
logtablenames = dblib.sqliteDatabase(pilib.dirs.dbs.log).get_table_names()
output['data'] = {'logscount':len(logtablenames)}
elif action == 'test_action':
output['message'] += 'Testing action. '
controldb = dblib.sqliteDatabase(pilib.dirs.dbs.control)
actiondict = controldb.read_table('actions',condition='"name"=\'' + post['actionname'] + "'")[0]
from cupid.actions import action
test_action = action(actiondict)
test_action.test()
elif action == 'update_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
safe_database.set_single_value('wireless', 'password', post['password'], "SSID='" + post['ssid'] + "'")
elif action == 'add_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
insert = {'SSID':post['ssid'], 'auto':1, 'priority':1}
if 'password' in post:
insert['password'] = post['password']
safe_database.insert('wireless',insert)
elif action == 'delete_network':
safe_database = dblib.sqliteDatabase(pilib.dirs.dbs.safe)
safe_database.delete('wireless', "SSID='" + post['ssid'] + "'")
# elif action == 'dump':
# # this has to go.
# if 'database' in d:
# dbpath = pilib.dbnametopath(d['database'])
# if dbpath:
# if 'tablelist' in d and 'outputfile' in d:
# dbpath = pilib.dbnametopath(d['database'])
# dblib.sqlitedatadump(dbpath, d['tablelist'], d['outputfile'])
# output['message'] = 'data dumped'
# elif 'tablename' in d and 'outputfile' in d:
# dblib.sqlitedatadump(dbpath, [d['tablename']], d['outputfile'])
# output['message'] = 'data dumped. '
# else:
# output['message'] += 'keys not present for dump. '
# else:
# output['message'] += 'keys not present for dump. '
# else:
# output['message'] += 'keys not present for dump. '
elif action in ['userdelete', 'useradd', 'usermodify']:
"""
This needs to be consolidate with the other useradd, modify algorithm written already.
Probably do this when we update the user permissions interface.
"""
# Ensure that we are authorized for this action
if action == 'userdelete':
try:
dblib.sqlitequery(pilib.dirs.dbs.users, "delete from users where name='" + post['usertodelete'] + "'")
except:
output['message'] += 'Error in delete query. '
else:
output['message'] += 'Successful delete query. '
elif action == 'usermodify':
if 'usertomodify' in post:
querylist=[]
if 'newpass' in post:
from pilib import salt
# Get session hpass to verify credentials
hashedpassword = post['newpass']
hname = hashlib.new('sha1')
hname.update(post['usertomodify'])
hashedname = hname.hexdigest()
hentry = hashlib.new('md5')
hentry.update(hashedname + salt + hashedpassword)
hashedentry = hentry.hexdigest()
querylist.append('update users set password='******'" + post['usertomodify'] + "'")
if 'newemail' in post:
querylist.append("update users set email='" + post['newemail'] + "' where name='" + post['usertomodify'] + "'")
if 'newauthlevel' in post:
querylist.append("update users set authlevel='" + post['newauthlevel'] + "' where name='" + post['usertomodify'] + "'")
try:
dblib.sqlitemultquery(pilib.dirs.dbs.users, querylist)
except:
output['message'] += 'Error in modify/add query: ' + ",".join(querylist)
else:
output['message'] += 'Successful modify/add query. ' + ",".join(querylist)
else:
output['message'] += 'Need usertomodify in query. '
elif action == 'useradd':
try:
username = post['newusername']
except:
username = '******'
try:
newemail = post['newemail']
except:
newemail = '*****@*****.**'
try:
newauthlevel = post['newauthlevel']
except:
newauthlevel = 0
query = "insert into users values(NULL,'" + username + "','','" + newemail + "',''," + str(newauthlevel) + ")"
try:
dblib.sqlitequery(pilib.dirs.dbs.users, query)
except:
output['message'] += "Error in useradd sqlite query: " + query + ' . '
else:
output['message'] += "Successful query: " + query + ' . '
elif action == 'getfiletext':
try:
filepath = post['filepath']
if 'numlines' in post:
numlines = int(post['numlines'])
else:
numlines = 9999
output['message'] += 'Using numlines: ' + str(numlines) + ' for read action. '
if 'startposition' in post:
startposition = post['startposition']
else:
startposition = 'end'
output['message'] += 'Reading from position ' + startposition + '. '
except KeyError:
output['message'] += 'Sufficient keys for action getfile text do not exist. '
except:
output['message'] += 'Uncaught error in getfiletext. '
else:
try:
file = open(filepath)
lines = file.readlines()
except:
output['message'] += 'Error reading file in getfiletext action. '
else:
output['data'] = []
if startposition == 'end':
try:
output['data'] = datalib.tail(file, numlines)[0]
except:
output['message'] += 'Error in tail read. '
else:
linecount = 0
for line in lines:
linecount += 1
if linecount > numlines:
break
else:
output['data'].append(line)
elif action == 'getmbtcpdata':
try:
clientIP = post['clientIP']
register = post['register']
length = post['length']
except KeyError:
output['message'] += 'Sufficient keys do not exist for the command. Requires clientIP, register, and length. '
else:
from iiutilities.netfun import readMBcodedaddresses
# try:
output['response'] = readMBcodedaddresses(clientIP, int(register), int(length))
elif action == 'queuemessage':
output['message'] += 'Queue message. '
if 'message' in post:
try:
dblib.sqliteinsertsingle(pilib.dirs.dbs.motes, 'queuedmessages', [datalib.gettimestring(), post['message']])
except:
import traceback
exc_type, exc_value, exc_traceback = sys.exc_info()
output['message'] += 'Error in queue insert query: {}. '.format(traceback.format_exc())
else:
output['message'] += 'Message insert successful'
else:
output['message'] += 'No message present. '
elif action == 'setsystemflag' and 'systemflag' in post:
database = pilib.dirs.dbs.system
dblib.setsinglevalue(database, 'systemflags', 'value', 1, "name=\'" + post['systemflag'] + "'")
elif action == 'rundaemon':
from cupiddaemon import rundaemon
rundaemon()
# TODO: Eliminate this scary thing.
elif action == 'setvalue':
utility.log(pilib.dirs.logs.control, "Setting value in wsgi", 1, 1)
# we use the auxiliary 'setsinglecontrolvalue' to add additional actions to update
if all(k in post for k in ('database', 'table', 'valuename', 'value')):
dbpath = pilib.dbnametopath(post['database'])
if dbpath:
output['message'] += 'Carrying out setvalue for value ' + post['valuename'] + ' on ' + post['table'] + ' in ' + dbpath
if 'condition' in post:
pilib.setsinglecontrolvalue(dbpath, post['table'], post['valuename'], post['value'], post['condition'])
elif 'index' in post:
condition = 'rowid= ' + post['index']
pilib.setsinglecontrolvalue(dbpath, post['table'], post['valuename'], post['value'], condition)
else:
pilib.setsinglecontrolvalue(dbpath, post['table'], post['valuename'], post['value'])
else:
output['message'] += 'Problem translating dbpath from friendly name: ' + post['database']
else:
output['message'] += 'Insufficient data for setvalue '
elif action == 'updateioinfo':
if all(k in post for k in ['database', 'ioid', 'value']):
query = dblib.makesqliteinsert('ioinfo', [post['ioid'], post['value']], ['id', 'name'])
try:
dblib.sqlitequery(pilib.dirs.dbs.control, query)
except:
output['message'] += 'Error in updateioinfo query execution: ' + query +'. into database: ' + pilib.dirs.dbs.control
output['message'] += 'ioid: ' + post['ioid'] + ' . '
else:
output['message'] += 'Executed updateioinfo query. '
else:
output['message'] += 'Insufficient data for updateioinfo query ! '
# TODO: properly incorporate and test channel class functions here, and then sub it.
elif action == 'modify_channel':
controllib.app_modify_channel(post, output)
elif action == 'deletechannelbyname' and 'database' in post and 'channelname' in post:
dbpath = pilib.dbnametopath(post['database'])
dblib.sqlitequery(dbpath, 'delete channelname from channels where name=\"' + post['channelname'] + '\"')
elif action == 'updatecameraimage':
output['message'] += 'Take camera image keyword. '
import cupid.camera
if 'width' in post:
width = post['width']
else:
width = 800
try:
values = cupid.camera.takesnap(width=width)
except:
output['message'] += 'Error taking image. '
else:
output['message'] += 'Appears successful. Path : ' + values['imagepath'] + '. Timestamp : ' + values['timestamp'] + '. '
output['data'] = values
elif action == 'getcurrentcamtimestamp':
output['message'] += 'getcurrentcamtimestamp keyword found. '
try:
with open('/var/www/webcam/images/current.jpg.timestamp') as f:
data = f.read()
except:
output['message'] += 'Error reading file as requested. '
else:
output['data'] = data
else:
output['message'] += 'Action keyword present(' + action + '), but not handled. '
else:
output['message'] += 'Authentication unsuccessful or action not authorized.'
status = '401 Not Authorized'
foutput = json.dumps(output, indent=1)
response_headers = [('Content-type', 'application/json')]
start_response(status, response_headers)
return [foutput]
def application(environ, start_response):
import cgi
import json
import os,sys,inspect
# Set top folder to allow import of modules
top_folder = os.path.split(os.path.realpath(os.path.abspath(os.path.split(inspect.getfile( inspect.currentframe() ))[0])))[0]
if top_folder not in sys.path:
sys.path.insert(0,top_folder)
from iiutilities.datalib import gettimestring
from iiutilities.dblib import sqlitequery
# post_env = environ.copy()
# post_env['QUERY_STRING'] = ''
# post = cgi.FieldStorage(
# fp=environ['wsgi.input'],
# environ=post_env,
# keep_blank_values=True
# )
#
# formname=post.getvalue('name')
#
# post={}
# for k in post.keys():
# post[k] = post.getvalue(k)
try:
request_body_size = int(environ.get('CONTENT_LENGTH', 0))
except ValueError:
request_body_size = 0
request_body = environ['wsgi.input'].read(request_body_size)
post = json.loads(request_body.decode('utf-8'))
output = {}
output['message'] = ''
status = '200 OK'
if 'sessionid' in post.keys() and 'event' in post.keys() and 'realIP' in post.keys() and 'apparentIP' in post.keys():
# sessionid contains the session id
sessionid = post.getvalue('sessionid')
if post.getvalue('event') == 'access':
accesstime = gettimestring()
username = post.getvalue('username')
apparentIP = post.getvalue('apparentIP')
realIP = post.getvalue('realIP')
sqlitequery('/var/www/data/authlog.db',"insert into sessionlog values ( \'" + username + "\',\'" + sessionid + "\',\'" + accesstime + "\'," + "\'access\' ,\'" + apparentIP + "\',\'" + realIP + "\' )")
output = "Output processed for " + realIP + " & " + apparentIP
else:
output = 'error: no session field sent'
response_headers = [('Content-type', 'text/plain'), ('Content-Length',str(len(output)))]
start_response(status,response_headers)
return [output]
def set_mode(self, mode):
from pilib import dirs
from iiutilities.dblib import sqlitequery
sqlitequery(dirs.dbs.control, 'update channels set mode=\'' + str(mode) + '\' where name=\'' + channel.name + '\'')
def setaction(controldbpath, channelname, action):
from iiutilities.dblib import sqlitequery
sqlitequery(controldbpath,
'update channels set action = \'' + str(action) + '\' where name = \'' + channelname + '\'')
def application(environ, start_response):
import cgi
import json
import os, sys, inspect
# Set top folder to allow import of modules
top_folder = os.path.split(
os.path.realpath(
os.path.abspath(
os.path.split(inspect.getfile(inspect.currentframe()))[0])))[0]
if top_folder not in sys.path:
sys.path.insert(0, top_folder)
from iiutilities.datalib import gettimestring
from iiutilities.dblib import sqlitequery
# post_env = environ.copy()
# post_env['QUERY_STRING'] = ''
# post = cgi.FieldStorage(
# fp=environ['wsgi.input'],
# environ=post_env,
# keep_blank_values=True
# )
#
# formname=post.getvalue('name')
#
# post={}
# for k in post.keys():
# post[k] = post.getvalue(k)
try:
request_body_size = int(environ.get('CONTENT_LENGTH', 0))
except ValueError:
request_body_size = 0
request_body = environ['wsgi.input'].read(request_body_size)
post = json.loads(request_body.decode('utf-8'))
output = {}
output['message'] = ''
status = '200 OK'
if 'sessionid' in post.keys() and 'event' in post.keys(
) and 'realIP' in post.keys() and 'apparentIP' in post.keys():
# sessionid contains the session id
sessionid = post.getvalue('sessionid')
if post.getvalue('event') == 'access':
accesstime = gettimestring()
username = post.getvalue('username')
apparentIP = post.getvalue('apparentIP')
realIP = post.getvalue('realIP')
sqlitequery(
'/var/www/data/authlog.db',
"insert into sessionlog values ( \'" + username + "\',\'" +
sessionid + "\',\'" + accesstime + "\'," + "\'access\' ,\'" +
apparentIP + "\',\'" + realIP + "\' )")
output = "Output processed for " + realIP + " & " + apparentIP
else:
output = 'error: no session field sent'
response_headers = [('Content-type', 'text/plain'),
('Content-Length', str(len(output)))]
start_response(status, response_headers)
return [output]