datalogger_CO2.py

#!/usr/bin/python

version='v0.3.9'

import serial
import string
import time
import argparse

#path and device name
import xsensor_device
#path and file name
import xsensor_path

##set column names
#colnames(t)=c("time","index","temperature","CO2","lamp")

#?	Alphasense NDIR
#N	1281.1
#T	26.6
#V	 400


#CLI arguments
serialDev='/dev/ttyUSB0'
fake='test.raw'
module='NDIRwCO2'+'_'+'AM01NNNN'
hostname='raspc2aN'
current_time=time.localtime()
info_file_name=xsensor_path.get_info_file_name(current_time,hostname)
file_name=xsensor_path.get_data_file_name(module,current_time)
raw_file_name=xsensor_path.get_raw_file_name(module,current_time)
parser = argparse.ArgumentParser(
  description='log data from NDIR/CO2 sensor.'
  +' At least, 3 files will be written:'
  +' 1. single information file (.info),'
  +' 2. one raw sensor ouput (.raw) file per day,'
  +' 3. one data (.txt) file per day.'
 ,epilog='example: %(prog)s --device '+serialDev
  +'  #will write e.g. module #'+module+' data as 1."'
  +info_file_name+'", 2."'+file_name+'" and 3."'+raw_file_name+'" files.'
  )
parser.add_argument('-d','--device',default=serialDev, help='USB device name or '+fake+' (e.g. '+serialDev+')')
parser.add_argument('-v','--version',action='version',version='%(prog)s '+version)
args = parser.parse_args()

##--device CLI argument
serialDev=args.device
if(serialDev==fake): #test#
  print 'test' #test#
  ser=open(fake, 'r') #test#
else:
  ser=serial.Serial(serialDev, 19200, timeout=67)
print 'pack module lines from ', serialDev

#sensor parameter arrays
nb=4 #3 sensors and others
record=range(nb)
value=range(nb)
send=range(nb-1)
send[0]='N'
send[1]='T'
send[2]='V'

#get current time
current_time=time.localtime()
str_time=time.strftime('%Y/%m/%d %H:%M:%S\n',current_time)
print str_time

#set info file name from date
import platform
hostname=platform.node()
info_file_name=xsensor_path.get_info_file_name(current_time,hostname)

#set module name
module=(module.split('_'))[0] #from generic for Help
module=module+'_'+xsensor_device.get_serial_device_name(serialDev)

#write to information file
fi=open(info_file_name,"a")
fi.write("\n\n"+str_time)
fi.write("datalogger."+version+".py\n")
fi.write("running on "+hostname+"\n")
fi.write(serialDev+"\n")
fi.write(module)
fi.close() #information file

#set both raw and data file names from both module and date
file_name=xsensor_path.get_data_file_name(module,current_time)
raw_file_name=xsensor_path.get_raw_file_name(module,current_time)

iteration=0

if(serialDev==fake): #test#
  mode=fake
else:
  mode='serial'
print 'start writing/reading '+mode+' ...'
while(True): #loop on both sensors and time
  for i in range(0,nb-1):
    #ask for data
## if(serialDev!=fake): #!test#
    ser.write(send[i]+"\r")
    #wait and get data
    line=ser.readline()
    #exit at end of file
    if not line: break
    #show
    ##print "|",line,"|"
    #write to raw file
    fr=open(raw_file_name,"a")
    fr.write(line)
    fr.close()
    #add sensor parameters in arrays
    value[i]=line.replace("\r\n","")
    record[i]=float(value[i])
  #exit at end of file
  if not line: break
  #generate module line from arrays
  current_time=time.localtime()
  str_time=time.strftime('%Y/%m/%d %H:%M:%S',current_time)
  line=module+";"+str_time+";"+str(iteration)
  for i in range(0,nb-1):
    line+=";"+value[i]
#    line+=";"+str(record[i])
  line+="\n"
  print line
  #setup file name from date
  file_name=xsensor_path.get_data_file_name(module,current_time)
  raw_file_name=xsensor_path.get_raw_file_name(module,current_time)
  #write to file
  fo=open(file_name,"a")
  fo.write(line)
  fo.close()
  #next record index
  iteration+=1
  time.sleep(12)

if(serialDev==fake): #test#
  ser.close()