EDISON DAM060 Logging v3

import signal
import sys
import time
import serial
import math
import serial.tools.list_ports
import io
import getopt

Excl = input('Excel export y/n: ')
if(Excl == 'y' or Excl == 'Y'):
    import xlwt
    book = xlwt.Workbook(encoding="utf-8")
    sheet1 = book.add_sheet("Sheet 1")
    sheet1.write(1, 1, "Measure")
    sheet1.write(1, 2, "Value")
    sheet1.write(1, 3, "Unit")
    Spc = 2

interval = '1.0'

ports = list(serial.tools.list_ports.comports())
for p in ports:
    print(p)

def showhelp():
    print( '        va18b.py -i <time in secs> -d <device>')
    print( '   Eg:  va18b.py -i 1.5 -d /dev/ttyUSB1')


try:
    opts, args = getopt.getopt(sys.argv[1:],"hd:i:",['help'])
except getopt.GetoptError:
    print( 'Error: -i, -d option require a value.')
    showhelp()
    sys.exit(1)

for opt, arg in opts:
    if opt in ('-h','--help'):
      showhelp()
      sys.exit(1)
    if opt == '-i':
      interval = arg
    if opt == '-d':
      device = arg

try:
    InpP = 'COM'+input("\nZadej cislo portu na otevreni: ")
    port=serial.Serial(port=InpP,
               baudrate=2400,
               bytesize=serial.EIGHTBITS,
               stopbits=serial.STOPBITS_ONE,
               parity=serial.PARITY_NONE,
               timeout=None)

    if not port.isOpen():
        port.open()

except IOError as err:
    print( '\nError:' + str(err) + '\n')
    sys.exit(1)


def signal_handler(signal, frame):
    port.flushInput()
    port.close()
    sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)

# Every packet is 14 bytes long.
def get_bytes():
    i=0
    substr=''
    while i<=13:
        byte=port.read(1)
        # converting every byte to binary format keeping the low nibble.
        substr += '{0:08b}'.format(ord(byte))[4:]
        i += 1
    return substr

def TF(strn):
    try:
        return float(strn)
    except:
        return float(9999999999)

def stream_decode(substr):
    ac       = int(substr[0:1])
    dc       = int(substr[1:2])
    auto     = int(substr[2:3])
    pclink   = substr[3:4]
    minus    = int(substr[4:5])

    digit1   = substr[5:12]
    dot1     = int(substr[12:13])
    digit2   = substr[13:20]
    dot2     = int(substr[20:21])
    digit3   = substr[21:28]
    dot3     = int(substr[28:29])
    digit4   = substr[29:36]

    micro    = int(substr[36:37])
    nano     = int(substr[37:38])
    kilo     = int(substr[38:39])
    diotst   = int(substr[39:40])
    mili     = int(substr[40:41])
    percent  = int(substr[41:42])
    mega     = int(substr[42:43])
    contst   = int(substr[43:44])
    cap      = int(substr[44:45])
    ohm      = int(substr[45:46])
    rel      = int(substr[46:47])
    hold     = int(substr[47:48])
    amp      = int(substr[48:49])
    volts    = int(substr[49:50])
    hertz    = int(substr[50:51])
    lowbat   = int(substr[51:52])
    minm     = int(substr[52:53])
    fahrenh  = substr[53:54]
    celcius  = int(substr[54:55])
    maxm     = int(substr[55:56])

    digit = {"1111101":"0",
             "0000101":"1",
             "1011011":"2",
             "0011111":"3",
             "0100111":"4",
             "0111110":"5",
             "1111110":"6",
             "0010101":"7",
             "1111111":"8",
             "0111111":"9",
             "0000000":" ",
             "1101000":"L"}

    value = ("-" if minus else "") +\
            digit.get(digit1,"") + ("." if dot1 else "") +\
            digit.get(digit2,"") + ("." if dot2 else "") +\
            digit.get(digit3,"") + ("." if dot3 else "") +\
            digit.get(digit4,"")
    value=str(TF(value))

    flags = "".join([ "   AC   |"   if ac     else "",
                      "   DC   |"   if dc     else "",
                      "  Auto  |"   if auto   else "",
                      " Diodet |"   if diotst else "",
                      " Contin |"   if contst else "",
                      "  Capm  |"   if cap    else "",
                      "  Rel   |"   if rel    else "",
                      "  Hold  |"   if hold   else "",
                      "  Minv  |"   if minm   else "",
                      "  Maxv  |"   if maxm   else "",
                      "  LowB  |"   if lowbat else ""])

    units = ("n   "    if nano    else "") +\
            ("u   "    if micro   else "") +\
            ("k   "    if kilo    else "") +\
            ("m   "    if mili    else "") +\
            ("M   "    if mega    else "") +\
            ("%   "    if percent else "") +\
            ("Ohm "    if ohm     else "") +\
            ("Amp "    if amp     else "") +\
            ("Volt"    if volts   else "") +\
            ("Hz  "    if hertz   else "") +\
            ("C   "    if celcius else "")

    Outp = [value + " |" + flags + " " + units, value]

    if "n   " in units:
        Outp.append("n (10^−9)")
        Outp.append("*math.pow(10, -9)")
    elif "u   " in units:
        Outp.append("u (10^−6)")
        Outp.append("*math.pow(10, -6)")
    elif "m   " in units:
        Outp.append("m (10^−3)")
        Outp.append("*math.pow(10, -3)")
    elif "k   " in units:
        Outp.append("k (10^3)")
        Outp.append("*math.pow(10, 3)")
    elif "M   " in units:
        Outp.append("M (10^6)")
        Outp.append("*math.pow(10, 6)")
    else:
        Outp.append(" ")
        Outp.append("")

    if(Excl == 'y' or Excl == 'Y'):
        if "Ohm " in units:
            Outp2="Ohm "
        elif "Amp " in units:
            Outp2="Amp "
        elif "Volt" in units:
            Outp2="Volt"
        elif "Hz  " in units:
            Outp2="Hz  "
        elif "C   " in units:
            Outp2="C   "

    if(Excl == 'y' or Excl == 'Y'):
        return [Outp,Outp2]
    else:
        return [Outp,""]

sys.stderr.write('Press enter to log\n')
sys.stderr.write('Press Ctrl+C to exit.\n\n')

def GTW(data):
    return eval(str(TF(data[0][1]))+str(data[0][3]))

while 1:
    if(Excl == 'y' or Excl == 'Y'):
        InpXv = input('X Value: ')
    else:
        input()

    port.flushInput()
    time.sleep(0.2)
    substr = get_bytes()
    data = stream_decode(substr)
    sys.stderr.write('Read: %.2f' %TF(data[0][1]))
    try:
        sys.stderr.write('%s' %data[0][2])
    except:
        pass
    if(Excl == 'y' or Excl == 'Y'):
        sys.stderr.write('\n')
    if(Excl == 'y' or Excl == 'Y'):
        sheet1.write(Spc, 1, InpXv)
        sheet1.write(Spc, 2, GTW(data))
        sheet1.write(Spc, 3, data[1])
        book.save("log.xls")
        Spc = Spc+1