MonitorAll.py

#!/usr/bin/python
import os
import sys

import HumidityAndTemp, TempOneWire, time


#########################
# Functions
#########################

def writeTemp(values):
    file = open("All.csv", 'a')

    s = str(values)
    file.write(s)
    file.write("\n")

    file.close()


def addSpacing():
    file = open("All.csv", 'a')
    file.write("\n")
    print ("\n")
    file.close()


global sleep, file_name, options
sleep = 600
file_name = "results.csv"
options = "a"


def incorrect_usage(argv, required_number):
    if (len(argv) - 1) != required_number:
        print "Incorrect number of arguments. Provided " + str(
                len(argv) - 1) + " but requires " + str(required_arg)

    print "Example usage:"
    print "./MonitorAll.py -s=600 -f=results.csv -o"
    print "     -s Sleep time in seconds between readings"
    print "     -f File to write results to"
    print "     -o or -a Overwrite or Append to the results file if it already exists (Must provide just one)"
    exit(0)


def check_args(argv):
    global sleep, file_name, options
    met_sleep = False
    met_file = False
    met_options = False
    argument_list = []
    for arg in argv:
        if arg.startswith("-"):
            arg = arg[1:]
            temp_split = arg.split("=")
            argument_list.append(temp_split)
            if len(temp_split) == 2 and temp_split[0] == "s":
                sleep = temp_split[1]
                met_sleep = True
                try:
                    float(sleep)
                except ValueError:
                    met_sleep = False
            elif len(temp_split) == 2 and temp_split[0] == "f":
                file_name = temp_split[1]
                met_file = True
            elif len(temp_split) == 1 and temp_split[0] == "o" or temp_split[0] == "a":
                options = temp_split[0]
                met_options = True

    if not met_sleep or not met_file or not met_options:
        print "Missing args: "
        if not met_sleep:
            print "-s"
        if not met_file:
            print "-f"
        if not met_options:
            print "-o or -a"
        return False

    return True


#########################
# Main
#########################
print """
***************************************
    ____                    __      ____  _
   / __ \____ _      ______/ /_  __/ __ \(_)
  / /_/ / __ \ | /| / / __  / / / / /_/ / /
 / _, _/ /_/ / |/ |/ / /_/ / /_/ / ____/ /
/_/ |_|\____/|__/|__/\__,_/\__, /_/   /_/
                          /____/
***************************************
"""
print "Welcome to RowdyPi Temperature Sensors"
required_arg = 3
if (len(sys.argv) - 1) != required_arg or not check_args(sys.argv):
    incorrect_usage(sys.argv, required_arg)

string_options = ""
if options == "a":
    string_options = "appending"
elif options == "o":
    string_options = "overwriting"
    options = "w"

print "Beginning temperature readings"
print "Writing (" + string_options + ") to file [" + file_name + "] with readings every [" + str(sleep) + "] seconds"

while (True):

    bothVals = HumidityAndTemp.getHumAndTemp()
    justTempVals = TempOneWire.getReadings()
    allValues = bothVals + justTempVals

    tempHumidityIndexes = {0: "Date", 1: "ID", 2: "Pin Number",
                           3: "Temperature", 4: "Humidity"}
    tempWireIndexes = {0: "Date", 1: "ID", 2: "Temperature_wire"}
    temp_humidity = "Temp,Hum"
    temp = "Temp"
    both_count = 0
    wire_count = 0

    id_date = 0
    id_key = 1
    id_pin = 2
    id_both_temp = 3
    id_both_hum = 4
    id_wire_temp = 2
    for key, value in tempHumidityIndexes.iteritems():
        if value == "ID":
            id_key = key
        elif value == "Date":
            id_date = key
        elif value == "Pin Number":
            id_ping = key
        elif value == "Temperature":
            id_both_temp = key
        elif value == "Humidity":
            id_both_hum = key

    for key, value in tempWireIndexes.iteritems():
        if value == "Temperature_wire":
            id_wire_temp = key
    line = ""
    if not os.path.isfile(file_name) or (options == "w"):
        line += "Date"
        for res in allValues:
            line += "," + res[id_key]
            if len(res) == 5:
                line += ","
                both_count += 1
            elif len(res) == 3:
                wire_count += 1

        line += "\n"
        for i in xrange(both_count):
            line += "," + temp_humidity

        for i in xrange(wire_count):
            line += "," + temp

        line += "\n"

    line += allValues[0][0]
    for res in allValues:
        data = ""
        if len(res) == 5:
            data += "," + res[3] + "," + res[4]
        elif len(res) == 3:
            data += "," + res[2]

        line += data

    line += "\n"

    print line

    file_results = open(file_name, options)

    s = str(line)
    file_results.write(s)

    file_results.close()
    options = "a"
    time.sleep(float(sleep))