drhd-control/test.py at master · didim99/drhd-control · 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
import json
import random
from ipaddress import IPv4Address

from driver import HDMIMatrix
from driver.discovery import NetworkExplorer
from driver.protocol import UDPPacket, TCP_PORT


class MatrixTester(object):
    config: dict = None
    explorer: NetworkExplorer = None
    device: HDMIMatrix = None

    def __init__(self, config: dict):
        self.config = config

    def find(self):
        self.explorer = NetworkExplorer(self.on_device_found)
        self.explorer.logging(self.config.get("log_udp", "warning"))
        self.explorer.start(self.config.get('ip', '0.0.0.0'))

    def on_device_found(self, data: UDPPacket):
        self.explorer.pause(True)
        self.start_test(data.devIP)

    def start_test(self, ip: IPv4Address):
        self.device = HDMIMatrix((ip, TCP_PORT))
        self.device.logging(self.config.get("log_tcp", "warning"))
        self.device.connect()
        if self.explorer is not None:
            self.explorer.stop()
        self.test_matrix()
        self.device.disconnect()

    def test_matrix(self):
        for i in range(self.device.num_out):
            self.device.get_source_for(i + 1)
            self.device.get_output_status(i + 1)
        for i in range(self.device.num_in):
            self.device.get_input_status(i + 1)
        for i in range(self.device.num_out):
            src = random.randint(1, 4)
            self.device.map_port(src, i + 1)
        self.device.get_port_mapping()


def run():
    with open('./config.json', 'r') as file:
        config = json.load(file)
    controller = MatrixTester(config)
    if "dev_ip" not in config:
        controller.find()
    else:
        ip = IPv4Address(config['dev_ip'])
        controller.start_test(ip)


if __name__ == '__main__':
    run()