triplet-bmu/cli.py at master · khaffner/triplet-bmu · 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
147
148
149
# -*- coding: utf-8 -*-
"""
"""
import ev_diag
import os
from sys import exit
import serial
import configuration as cc

ev_diag.serport = cc.config['DEFAULT']['Port']


def clear():
    os.system('cls' if os.name == 'nt' else 'clear')


def print_header():
    clear()
    print(cc.output_as_str('header'))


def exit_prog(answer):
    if answer == ":q" or answer == "quit":
        exit()
        quit()


def print_menu():
    print_header()
    print(cc.output_as_str('diag'), '\n')
    print(cc.output_as_str('print header'))
    answer = input('')
    if answer == '1':
        ev_diag.ion.print_data(ev_diag.ion.bmu_data)
    elif answer == '2':
        ev_diag.ion.print_data(ev_diag.ion.cell_info)
    elif answer == '3':
        ev_diag.ion.print_data(ev_diag.ion.cell_info)


def port_tool():
    while 1:
        print_header()
        print(cc.output_as_str('port header'), ev_diag.serport, '\n')
        print(cc.output_as_str('port instr'))
        answer = input('')
        if answer == '2':
            break
        elif answer == '1':
            print(cc.output_as_str('port input'))
            nport = input('')
            try:
                nport = 'COM'+str(int(nport))
            except ValueError:
                pass
            if nport.startswith('COM') is True and nport[:3]:
                try:
                    int(nport[3:])
                    ev_diag.serport = nport
                except ValueError:
                    pass
        cc.config.set('DEFAULT', 'port', ev_diag.serport)
        cc.writeconf()
        exit_prog(answer)
    clear()


def baud_tool():
    while 1:
        print_header()
        print(cc.output_as_str('baud header'), '\n')
        print(cc.output_as_str('baud instr'))
        answer = input('')
        if answer == '1':
            try:
                result = ev_diag.is_baudrate_ok()
            except serial.SerialException:
                result = False
            if result is True:
                print('\n', cc.output_as_str('con ok'))
            else:
                print('\n', cc.output_as_str('con not ok'))
            input(cc.output_as_str('enter2cont'))
        elif answer == '2':
            print(cc.output_as_str('ask yn'))
            answer = input('')
            if answer == 'y':
                try:
                    ev_diag.switch_baudrate()
                    print(cc.output_as_str('tried change br'))
                except serial.SerialException:
                    print(cc.output_as_str('con error'))
                input(cc.output_as_str('enter2cont'))
        elif answer == '3':
            break
        exit_prog(answer)


def diagnosis_menu():
    while 1:
        print_header()
        print(cc.output_as_str('diag'), '\n')
        print(cc.output_as_str('diag header'))
        answer = input('')
        if answer == '1':
            print(cc.output_as_str('read data'))
            try:
                ev_diag.connect()
                ev_diag.serialcon.do_setup(ev_diag.triplet_setup)
                ev_diag.ion.read_bmu_data(ev_diag.serialcon)
                ev_diag.ion.calculate_values()
                ev_diag.serialcon.reset()
                ev_diag.disconnect()
            except serial.SerialException:
                print(cc.output_as_str('con error'))
                input(cc.output_as_str('enter2cont'))
        elif answer == '2':
            print_menu()
            input(cc.output_as_str('enter2cont'))
        elif answer == '3':
            if ev_diag.ion.bmu_data != {}:
                ev_diag.ion.save_bmu_data()
            else:
                print(cc.output_as_str('no data'))
                input(cc.output_as_str('enter2cont'))
        elif answer == '4':
            break
        exit_prog(answer)
    clear()


opened = False

while 1:
    print_header()
    if opened is False:
        opened = True
        print(cc.output_as_str('quit instruction'), '\n')
    else:
        print('\n')
    print(cc.output_as_str('main header'))
    answer_main = input()
    if answer_main == '3':
        baud_tool()
    elif answer_main == '2':
        port_tool()
    elif answer_main == '1':
        diagnosis_menu()
    exit_prog(answer_main)