import signal
import sys
import time
import relay_control


class Bt():
    global relay_control

    def __init__(self, addrA=0x20, addrB=0x23):
        import smbus
        self.bus = smbus.SMBus(1)

        self.A = relay_control.RelayBoard(self.bus, addrA)
        self.B = relay_control.RelayBoard(self.bus, addrB)
        
    def OpenCapacitor(self):
        print('Odzkratujeme ...')
        self.B.on(0)

        
    def ShortCapacitor(self):
        print('Zazkratujeme ...')
        self.B.off(0)
        
    def ConnectHVsuppy(self):
        print('Pripojime zdroj ...')
        self.B.on(1)

    def DisconnectHVsuppy(self):
        print('Odpojime zdroj ...')
        self.B.off(1)
        

    def SetCWoutput(self):
        print('CW orientace ...')
        bus.write_byte_data(0x20, 0x06, ~(0x1 << 0))
        time.sleep(0.1)
        bus.write_byte_data(0x20, 0x06, ~(0x0 << 0))

    def SetACWoutput(self):
        print('ACW orientace ...')
        bus.write_byte_data(0x20, 0x06, ~(0x1 << 1))
        time.sleep(0.1)
        bus.write_byte_data(0x20, 0x06, ~(0x0 << 1))

def SetNormalCharging():
    global Bt;Bt = Bt()

    print('Normal Charging ...')
    Bt.A.on(0)
    time.sleep(0.5)
    Bt.A.off(0)
        
def SetReverseCharging():
    global Bt;Bt = Bt()

    print('Reverse Charging ...')
    Bt.A.on(1)
    time.sleep(0.5)
    Bt.A.off(1)

def ChargeCapacitor():
    global Bt
    Bt = Bt()
    
    Bt.OpenCapacitor();time.sleep(1)
    Bt.ConnectHVsuppy();time.sleep(3)
    Bt.DisconnectHVsuppy();

def ShortCapacitor():
    global Bt
    Bt=Bt()
    Bt.ShortCapacitor()


def Sandbox():
    print "hello"