usb_meter/usb_meter.py at master · bousqi/usb_meter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
#!/usr/bin/env python3

import sys
import time
import click
import serial
import serial.tools.list_ports
from serial import tools
from struct import unpack

if sys.platform == 'win32':
    from colorama import init
    init()

def parse_packet(packet):
    if len(packet) < 130:
        return None

    data = unpack('>'+'H'*(len(packet)//2),packet)
    p_vcc  = (float)(data[1]/100)
    p_cur  = data[2]
    p_temp = data[5]
    p_cap  = data[9]
    p_ene  = (float)(data[11]/1000)
    p_dp   = (float)(data[48]/100)
    p_dm   = (float)(data[49]/100)
    p_res  = (float)(data[62]/10)

    packet_val = {}
    packet_val["Vcc"] = p_vcc
    packet_val["Cur"] = p_cur
    packet_val["T°"]  = p_temp
    packet_val["Cap"] = p_cap
    packet_val["Ene"] = p_ene
    packet_val["D+"]  = p_dp
    packet_val["D-"]  = p_dm
    packet_val["Res"] = p_res

    # data_text = ''.join(['0x%02X ' % i for i in data])
    # print (data_text)

    return packet_val


def is_COM_valid(port):
    port_list = serial.tools.list_ports.comports()
    name_list = [p.device for p in port_list]
    return (port in name_list)


def print_list():
    port_list = serial.tools.list_ports.comports()
    print (" Dev  - Name [ Desc ]")
    for p in port_list:
        print ("%s - %s [ %s ]"%(p.device, p.name, p.description))

@click.command()
@click.option('--port', '-p',
              help="COM Port to connect to USB Tester")
@click.option('--baudrate', '-b',
              default=9600,
              help="Baudrate speed")
@click.option('--refresh', '-r',
              default=1000,
              help="Refresh speed in ms")
@click.option('--details', '-d',
              default=False, is_flag=True,
              help="Display all meter")
@click.option('--list', '-l',
              default=False, is_flag=True,
              help="List all COM ports available")
def run(port, baudrate, refresh, details, list):
    if list:
        print_list()
        return

    if port == None:
        print ('Try "usb_monitor.py --help" for help.')
        print ('Error: Missing option "--port" / "-p".\n')
        print ('Use one of the port below :\n')
        print_list()
        return

    if not is_COM_valid(port):
        print ('%s port does not exists'%port)
        print ('Use one of the port below :\n')
        print_list()
        return

    print ("Connecting to %s at %dbps"%(port, baudrate))
    z1serial = serial.Serial(port=port, baudrate=baudrate)
    z1serial.timeout = 2  # set read timeout


    minVcc = 10
    maxVcc = 0
    minCur = 1000
    maxCur = 0

    # print (z1serial.is_open)  # True for opened
    if z1serial.is_open:
        while True:
            try:
                z1serial.write(b'\xF0')
                data = z1serial.read(130)
            except:
                print("\n\nSerial Lost ?")
                return
            # data_text = ''.join(['0x%02X ' % i for i in data])
            # print (data_text)

            state = parse_packet(data)
            print("\033[2J", end='')
            if state == None:
                print('\rNo data...', end='')
                continue

            maxVcc = state["Vcc"] if state["Vcc"] > maxVcc else maxVcc
            minVcc = state["Vcc"] if state["Vcc"] < minVcc else minVcc
            maxCur = state["Cur"] if state["Cur"] > maxCur else maxCur
            minCur = state["Cur"] if state["Cur"] < minCur else minCur

            print("%s : %.2fv (v%.2f ^%.2f)"%("Vcc", state["Vcc"], minVcc, maxVcc), end='')
            print(" - ", end='')
            print("%s : %4dmA (v%4d ^%4d)"%("Cur", state["Cur"], minCur, maxCur), end='')
            if details:
                print(" / ", end='')
                print("%2d°"%(state["T°"]))
                print("[ ", end='')
                print("D+- %.2fv/%.2fv"%(state["D+"],state["D-"]), end='')
                print(" | ", end='')
                print("Res %3.1fohms"%(state["Res"]), end='')
                print(" | ", end='')
                print("Ene %.2fWh"%(state["Ene"]), end='')
                print(" | ", end='')
                print("Cap %dmAh"%(state["Cap"]), end='')
                print(" ] ", end='')
            print("", end='\r')

            # print("Vcc : %.2fv\t Cur : %4dmA"%(data_s[1]/100, data_s[2]), end='\r')

            time.sleep(refresh/1000)
    else:
        print ('z1serial not open')

run()