sergey
/
cyberpower
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

registerups.py: remove unneeded code 

and add connection exception handling
This commit is contained in:
Sergey Morozov 2019-08-26 13:17:55 +03:00
parent 9317eb30b9
commit c2750889e1
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
registerups/registerups.py
View File

@ -18,10 +18,7 @@ class cps_ups:
for i in ET.fromstring(self.connection.getresponse().read()).iter("data"): for i in ET.fromstring(self.connection.getresponse().read()).iter("data"):
keepalive_data=i.attrib keepalive_data=i.attrib
self.connection.close() self.connection.close()
result = {'boot': keepalive_data['boot'], return(keepalive_data)
'upt': keepalive_data['upt'],
'id': keepalive_data['id']}
return(result)
def __genxml__(self, keepalive_data): def __genxml__(self, keepalive_data):
data_tag = '<data ver="1.1"'\ data_tag = '<data ver="1.1"'\
'hash="hmac-md5-128" '\ 'hash="hmac-md5-128" '\
@ -63,8 +60,12 @@ class cps_ups:
self.contact = contact self.contact = contact
self.location = location self.location = location
self.bedt = bedt self.bedt = bedt
post_data = self.__genxml__(self.__get_keepalive_data__()) try:
self.connection.request("POST", "/setup.xml", body = post_data) post_data = self.__genxml__(self.__get_keepalive_data__())
self.connection.request("POST", "/setup.xml", body = post_data)
except (OSError, ConnectionError) as err:
print('Unable to register: ' + str(err))
sys.exit(3)
response = self.connection.getresponse().read() response = self.connection.getresponse().read()
self.connection.close() self.connection.close()
return(response) return(response)