Colony

#!/usr/bin/python

import sys, serial, MySQLdb

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()

class Robot(serial.Serial):

        def __init__(self, serial_path='/dev/ttyUSB0', baudrate=115200):

                try:

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

                except serial.SerialException as e:

                        print 'Warning: Unable to open serial communication to robot (%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

ST_conn = MySQLdb.connect(host = 'localhost',

                          user = 'colony',

                          passwd = 'mrbickles',

                          db = 'calibration_station')

ST_cursor = ST_conn.cursor()

ST_cursor.execute('SELECT * FROM raw_data;')

for row in ST_cursor:

        print row

ST = Station()

#ST.db_register('sensor0', ST_cursor)

#ST.db_register('sensor1', ST_cursor)

RO = Robot('/dev/ttyUSB1')

while True:

        if RO.inWaiting() > 0:

                try:

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

                except OSError as e:

                        print 'Error: %s' %e

                t = msg.split(':')

                print t

        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()