module_universal_io/tools/tester_dio.py

from tester import *
from modbus import Modbus
from digital_io import IO_Digital
from serial import Serial
import colorama
from colorama import Fore


class DIO_Tester(Tester):
    def __init__(self, dut_debug_port, mb_port, tester_addr=1, dut_address=2):
        super().__init__(dut_debug_port)

        serial_port = Serial(mb_port, 115200, timeout=0.05, parity='N', xonxoff=False)
        self.modbus_tester = Modbus(serial_port, tester_addr)
        self.modbus_dut = Modbus(serial_port, dut_address)
        
        self.tester = IO_Digital(self.modbus_tester)
        self.dut = IO_Digital(self.modbus_dut)


    def startup(self):
        """Предварительная настройка tester, dut"""
        
        # Отключаем безопасные режимы
        self.tester.sys.set_save_mode(0)
        self.dut.sys.set_save_mode(0)        

        # Подача питания на DUT
        self.tester.set_output(1, 1)
        time.sleep(0.3)


    def test_di(self):
        """Тест цифровых вхдов"""
        print('[Тестер] Установка выхода 2 --> 0')
        self.tester.set_output(2, 0)
        time.sleep(0.5)

        ret = self.dut.get_inputs_bit()
        even_inputs = ret[::2] # четные
        odd_inputs = ret[1::2] # нечетные

        if odd_inputs != '1111':
            print(Fore.RED + f'Нечетные входы не замкнуты [{odd_inputs}]. ERROR.')
        else:
            print(Fore.GREEN + 'Нечетные входы замкнуты. OK.')

        if even_inputs != '0000':
            print(Fore.RED + f'Четные входы не разомкнуты [{even_inputs}]. ERROR.')
        else:
            print(Fore.GREEN + 'Четные входы разомкнуты. OK.')

        print('[Тестер] Установка выхода 2 --> 1')
        self.tester.set_output(2, 1)
        time.sleep(0.5)

        ret = self.dut.get_inputs_bit()
        even_inputs = ret[::2] # четные
        odd_inputs = ret[1::2] # нечетные

        if odd_inputs != '0000':
            print(Fore.RED + f'Нечетные входы не разомкнуты [{odd_inputs}]. ERROR.')
        else:
            print(Fore.GREEN + 'Нечетные входы разомкнуты. OK.')

        if even_inputs != '1111':
            print(Fore.RED + f'Четные входы не замкнуты [{even_inputs}]. ERROR.')
        else:
            print(Fore.GREEN + 'Четные входы замкнуты. OK.')


    def test_do(self, do_number=1):
        """Проверка выхода"""

        self.dut.set_outputs_state(0b0000_0001)
        self.dut.set_outputs(0b1111_1111)
        time.sleep(1)
        self.dut.set_outputs(0b1111_1110)
        time_start = time.time()
        time.sleep(1)
        while True:
            ret = self.dut.get_loads_bit()
            time.sleep(0.1)
            if ret != '00000000':
                print(ret)
                print(time.time() - time_start)
                break


        return
        self.dut.set_outputs(0b1111_1100)
        time.sleep(1)
        ret = self.dut.get_loads_bit()
        print(ret)



    def test_rtc(self):
        """Проверка часов, батарейки и переключателя modbus адреса"""
        print("[Tester] Установка времени на DUT")
        self.dut.sys.set_rtc()
        # Снимаем питание с DUT
        self.tester.set_output(1, 0)
        print(Fore.YELLOW + "Переключите адрес на DUT в положение 'все переключатели вверх'.")
        print(Fore.YELLOW + "Затем нажмите 'Enter'")
        input()
        # Подаем питание на DUT
        self.tester.set_output(1, 1)
        time.sleep(3)

        # Меняем modbus-адрес DUT 
        self.modbus_dut.address = 15
        dut_time = self.dut.sys.get_rtc()
        print("Время DUT", dut_time)

        print(time.time() + self.dut.sys.utc_offset, dut_time/1000.0)
        print(abs(dut_time/1000.0 - (time.time() + self.dut.sys.utc_offset)))

        if abs(dut_time/1000.0 - (time.time() + self.dut.sys.utc_offset)) > 1:
            print(Fore.RED + f'Время не соответствует реальному. ERROR.')
        else:
            print(Fore.GREEN + 'Тест RTC и переключателя адреса. OK.')
 

    def test_led(self):

        # Зеленые LED индикациии входов
        print(Fore.YELLOW + "Посмотрите на зеленые светодиоды индикации входов")
        self.tester.set_output(2, 1)
        time.sleep(1)
        self.tester.set_output(2, 0)
        time.sleep(1)
        self.tester.set_output(2, 1)
        time.sleep(1)
        self.tester.set_output(2, 0)
        print("Нажмите 'Enter'")
        input()

        # Зеленые LED индикации 
        print(Fore.YELLOW + "Посмотрите на зеленые светодиоды индикации выходов")
        time.sleep(2)
        # включаем все выходы
        self.dut.set_outputs_state(0b1111_1111)
        for i in range(1, 9):
            self.dut.set_output(i, 1)
            time.sleep(1)
        for i in range(8, 0, -1):
            self.dut.set_output(i, 0)
            time.sleep(1)
        print("Нажмите 'Enter'")
        input()

        # Красные LED индикации нагрузки/аварии выходов
        print(Fore.YELLOW + "Посмотрите на красные светодиоды индикации аварий")
        time.sleep(2)

        self.dut.set_outputs_state(0b0000_0000)
        time.sleep(2)

        for i in range(1, 9):
            self.dut.set_output_state(i, 1)
            time.sleep(1)
        for i in range(8, 0, -1):
            self.dut.set_output_state(i, 0)
            time.sleep(1)

        print("Нажмите 'Enter'")
        input()


    def foo(self):
        print(self.modbus_dut.address)
        print(self.modbus_tester.address)

        self.modbus_dut.address = 15
        print(self.modbus_dut.address)

    def menu(self):

        self.tester.set_output(1, 1)
        self.startup()
                
        questions = [
	    inquirer.List('action',
                message="Тестирование MDIO_88",
                choices=['Прошить IAP и FW', 
                        'Получить системные настройки',
                        'Полный тест',
                        'Тест цифровых входов',
                        'Тест выхода 1',
                        'Тест RTC',
                        'Тест LED'],
            ),
	    ]
        answers = inquirer.prompt(questions)

        if answers['action'] == 'Прошить IAP и FW':
            self.write_iap_fw('bin_dio/fw.bin', 'bin_iap/iap.bin')
        elif answers['action'] == 'Получить системные настройки':
            print('Получение системных настроек')
            self.dut.sys.get_system_vars()
        elif answers['action'] == 'Полный тест':
            print('Запуск цикла тестирования')
        elif answers['action'] == 'Тест цифровых входов':
            self.test_di()
        elif answers['action'] == 'Тест выхода 1':
            self.test_do()
        elif answers['action'] == 'Тест RTC':
            self.test_rtc()
        elif answers['action'] == 'Тест LED':
            self.test_led()






def main():
    tester = DIO_Tester('COM3', 'COM11', 1, 2)
    tester.menu()
    # tester.foo()
    # tester.dut_powerup()


if __name__ == '__main__':
    main()