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
|
import requests
import sys
from datetime import datetime
from bs4 import BeautifulSoup
def build_request(params: dict):
items = []
for key, value in params.items():
items.append(f'<{key}>{value}</{key}>')
return '<request>'+''.join(items)+'</request>'
class WebAPI:
def __init__(self, ip: str):
self.ip = ip
self.headers = {}
pass
def auth(self):
response = self.request("webserver/SesTokInfo")
cookie = response.find('SesInfo').get_text()
token = response.find('TokInfo').get_text()
self.headers['Cookie'] = cookie
self.headers['__RequestVerificationToken'] = token
self.headers['Content-Type'] = 'text/xml'
def device_information(self):
return self.request('device/information')
def device_signal(self):
return self.request('device/signal')
def get_sms(self, count=10, page=1):
request = build_request({
'PageIndex': page,
'ReadCount': count,
'BoxType': 1,
'SortType': 0,
'Ascending': 0,
'UnreadPreferred': 1
})
response = self.request('sms/sms-list', data=request)
sms_list = []
for message in response.find_all('Message'):
sms = SMS(
index=int(message.find('Index').get_text()),
phone=message.find('Phone').get_text(),
content=message.find('Content').get_text(),
date=message.find('Date').get_text()
)
sms_list.append(sms)
return sms_list
def send_sms(self):
pass
def dataswitch(self, on=True):
return self.request('dialup/mobile-dataswitch', data=build_request({
'dataswitch': 1 if on else 0
}))
def reboot(self):
return self.request('device/control', data=build_request({
'Control': 1
}))
def request(self, endpoint: str, data=None):
url = f'http://{self.ip}/api/{endpoint}'
r = requests.get(url, headers=self.headers) if data is None else requests.post(url, data=data, headers=self.headers)
r.encoding = 'utf-8'
soup = BeautifulSoup(r.text, 'lxml-xml')
error = soup.find('error')
if error:
code = 0
message = ''
code_node = error.find('code')
message_node = error.find('message')
if code_node:
code = int(code_node.get_text())
if message_node:
message = message_node.get_text()
raise APIError(code, message=message)
return soup.find('response')
class SMS:
def __init__(self, index=None, phone=None, content=None, date=None):
self.index = index
self.phone = phone
self.content = content
self.date = date
def timestamp(self):
# input example: 2020-08-11 14:55:51
return int(datetime.strptime(self.date, '%Y-%m-%d %H:%M:%S').strftime("%s"))
class APIError(Exception):
def __init__(self, error_code, message='', *args, **kwargs):
self.error_code = error_code
self.traceback = sys.exc_info()
try:
msg = '[{0}] {1}'.format(error_code, message.format(*args, **kwargs))
except (IndexError, KeyError):
msg = '[{0}] {1}'.format(error_code, message)
super().__init__(msg)
|
