py-webrelais/relays.py at master · schinken/py-webrelais · 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
import pylibftdi
#import parallel

class SainSmart:

    def __init__(self, serial):
        self.device = pylibftdi.BitBangDevice(serial)
        self.device.direction = 0xFF

    def update(self, byte):
        self.device.port = byte

class ParPort:

    def __init__(self):
        self.device = parallel.Parallel()

    def update(self, byte):
        self.conn.setData(byte)

class Relay:

    def __init__(self, delegate):
        self.delegate = delegate
        self.state = [False]*8
        self._update()

    def _update(self):

        result = 0x00
        for i, val in enumerate(self.state[:8]):
            if val:
                result |= (1 << i)
            else:
                result &= ~(1 << i)

        self.delegate.update(result)

    def set_relay(self, relay, value):
        self.state[relay] = value
        self._update()

    def set_relays(self, value):
        self.state = [value]*8
        self._update()

    def get_relay(self, relay):
        return self.state[relay]

    def get_relays(self):
        return self.state

    def toggle_relay(self, relay):
        self.state[relay] = not self.state[relay]
        self._update()

class RelaysProxy:

    relais = []
    pin_to_relais = {}

    def add_relay(self, offset, num_relais, relais):
        cur_offset = len(self.pin_to_relais)

        for relay in xrange(offset, offset+num_relais):
            self.pin_to_relais[relay] = (relais, relay-cur_offset)

        self.relais.append(relais)

    def set_relay(self, relay, value):
        relais, id = self.pin_to_relais[relay]
        relais.set_relay(id, value)

    def get_relay(self, relay):
        relais, id = self.pin_to_relais[relay]
        return relais.get_relay(id)

    def get_relays(self):
        result = []
        for relais in self.relais:
            result = result + relais.get_relays()

        return result

    def toggle_relay(self, relay):
        relais, id = self.pin_to_relais[relay]
        relais.toggle_relay(id)

    def num_relais(self):
        return len(self.pin_to_relais)