Colony

#!/usr/bin/python

import sys, serial

import sqlite3

class Station(serial.Serial):

        def __init__(self, serial_path='/dev/arduino', baudrate=19200):

                try:

                        serial.Serial.__init__(self, serial_path, baudrate)

                except serial.SerialException as e:

                        print 'Warning: Unable to open serial communication to station (%s)'%e

                        sys.exit(0)

        def db_register(self, sensor_name, db_cursor):

                t = (sensor_name,)

                db_cursor.execute('create table %s (value INTEGER)' % t)

        def db_add(self, sensor_name, db_cursor, db_conn, value):

                t = (sensor_name, value)

                db_cursor.execute('insert into %s (value) values (%s)' % t)

                db_conn.commit()

# SQLite3 database setup

db_path = '/home/colony/calibration_station/cal_sta_database.db'

ST_conn = sqlite3.connect(db_path)

ST_cursor = ST_conn.cursor()

ST = Station()

ST.db_register('sensor0', ST_cursor)

ST.db_register('sensor1', ST_cursor)

#        RO_serial = serial.Serial(port='/dev/ttyUSB1', baudrate=115200)   # Robot

while True:

        if ST.inWaiting() > 0:

                try:

                        msg = ST.readline()[:-1]

                except OSError as e:

                        print 'Error: %s' %e

                t = msg.split(':')

                print t

                if (len(t) == 3 and t[1] == '0'):

                        ST.db_add('sensor0', ST_cursor, ST_conn, t[2])

                elif (len(t) == 3 and t[1] == '1'):

                        ST.db_add('sensor1', ST_cursor, ST_conn, t[2])

ST_cursor.close()

ST.close()