ePowerSwitch/ePowerSwitch.py at master · bajo/ePowerSwitch · 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
#!/usr/bin/env python
# Filename: ePowerSwitch.py

"""
EPowerSwitch module provides a way to set and retrieve the status of a socket on a remote-control socket strip from Leuning
See http://www.leunig.de/_en/_pro/remote-power-switch/eps.htm for the 4 socket model (ePowerSwitch-4).
Tested on ePowerSwitch-4.
"""

import base64
import string
import re
import time
import urllib
import urllib2
import sys
try:
    import prettytable
except:
    print('No pretty output possible')
    pretty = None

class EpsSocket:
    """
    class for ePowerSwitch power outlets
    each EpsSocket has a number, name, status and an unknown parameter
    """
    def __init__(self, number, name, status, param, cycles):
        self.number = number
        self.name = name
        self.status = status
        self.param = param
        self.cycles = cycles

    number = 0
    name = ''
    status = 0
    param = 0
    cycles = 0

class EpsSwitch:
    """
    class for ePowerSwitch power outlets
    each EPowerSwitch has 1, 2, 4 or 8 EpsSockets, name, ip, user and pw
    """

    def __init__(self, ip, port, user, password):
        self.port = str(port)
        self.url = 'http://'+ip+':'+self.port+'/'
        self.user = user
        self.password = password
        base64string = base64.encodestring(
                        '%s:%s' % (self.user, self.password))[:-1]
        self.authheader =  "Basic %s" % base64string

    name = ''
    port = '80'
    url = ''
    sockets = []
    user = ''
    password = ''

    def getData(self):
        """
        getData function
        retrieve data from ePowerSwitch power outletd
        data is the name, socketname, status and count of power cycles
        """

        del self.sockets[:]
        content = None
        #content = self.getData2(self.url+'cmd.html')
        get_req =  urllib2.Request(self.url+'cmd.html')
        get_req.add_header("Authorization", self.authheader)
        handle = urllib2.urlopen(get_req)
        try:
            content = handle.read()
        except:
            print 'cmd.html ERROR'
        if content == '401 Authorization Required':
            print 'Webserver is too slow. \nPlease repeat the request'
            return False
        else:
            cycles = dict()
            content2 = None
            try:
                get_req =  urllib2.Request(self.url+'config/misc_f.html')
                get_req.add_header("Authorization", self.authheader)
                handle = urllib2.urlopen(get_req)
                content2 = handle.read()
            except IOError, e:
                print 'config/misc_f.html ERROR'
            if content2:
                lines2 = string.split(content2, '\n')
                sockets2 = [s for s in lines2 if "Socket" in s]
                for item in sockets2:
                    a = re.search(r"Socket(.+)\)",item)
                    b = a.group(1).split(',')[0].split('"')[0].strip()
                    c = a.group(1).split(',')[1].strip()
                    cycles[b] = c

            lines = string.split(content, '\n')
            tmp_name = [s for s in lines if "H1" in s]
            sockets = [s for s in lines if "socket(" in s]
            self.name = re.search(r"<H1>(.+)<\/H1>",tmp_name[0]).group(1).strip()

            for socket in sockets:
                item = re.search(r"\((.+)\)",socket).group(1).strip()
                number = int(item.split(',')[0])
                name = item.split(',')[1].split()[0][1:]
                status = int(item.split(',')[2])
                param = int(item.split(',')[3])
                if len(cycles) < number:
                    cycles[str(number)] = 'None'
                self.sockets.append(EpsSocket(number, name, status, param, cycles[str(number)]))
            return True

    def showStatus(self):
        """ print status of all sockets for the ePowerSwitch """
        if (self.getData()):
            print self.name
            x = prettytable.PrettyTable(['socket number', 'connected device', 'status', 'power cycles'])
            for socket in self.sockets:
                if socket.status:
                    status = 'On'
                else:
                    status = 'Off'
                x.add_row([socket.number, socket.name, status, socket.cycles])
            print x

    def setStatus(self, number, status):
        """ set status of socket """
        if status.lower() in {'on','1'}:
            data = urllib.urlencode({'P'+str(number) : '1'})
        elif status.lower() in {'off','0'}:
            data = urllib.urlencode({'P'+str(number) : '0'})
        else:
            return False
        set_req =  urllib2.Request(self.url+'cmd.html', data)
        set_req.add_header("Authorization", self.authheader)
        try:
            response = urllib2.urlopen(set_req)
        except:
            response = urllib2.urlopen(set_req)
        return True