初始化提交
This commit is contained in:
王立帮
166
boards/default_src/micropython/origin/build/lib/debugnet.py
Normal file
166
boards/default_src/micropython/origin/build/lib/debugnet.py
Normal file
@@ -0,0 +1,166 @@
|
||||
"""
|
||||
Debugnet(HTTP,MQTT)
|
||||
|
||||
MicroPython library for network request debugging
|
||||
=======================================================
|
||||
|
||||
#Preliminary composition 20230225
|
||||
|
||||
@dahanzimin From the Mixly Team
|
||||
"""
|
||||
import time
|
||||
from umqtt import MQTTClient
|
||||
from ubinascii import hexlify
|
||||
from machine import unique_id
|
||||
from urequests import Response
|
||||
from usocket import socket,getaddrinfo,SOCK_STREAM
|
||||
from mixiot import WILL_TOPIC,ADDITIONAL_TOPIC
|
||||
|
||||
class socket_d(socket):
|
||||
def __init__(self,*args,debug=False,**kw):
|
||||
super().__init__(*args,**kw)
|
||||
self._debug=debug
|
||||
self.client_len=0
|
||||
self.server_len=0
|
||||
|
||||
def write(self,*args):
|
||||
super().write(*args)
|
||||
self.client_len=min(self.client_len+len(args[0]),65535)
|
||||
if self._debug:
|
||||
print('client:',args[0])
|
||||
|
||||
def readline(self,*args):
|
||||
buf=super().readline(*args)
|
||||
self.server_len=min(self.server_len+len(buf),65535) if buf else self.server_len
|
||||
if self._debug:
|
||||
print('server:',buf)
|
||||
return buf
|
||||
|
||||
def read(self,*args):
|
||||
buf=super().read(*args)
|
||||
self.server_len=min(self.server_len+len(buf),65535) if buf else self.server_len
|
||||
if self._debug:
|
||||
print('server:',buf)
|
||||
return buf
|
||||
|
||||
#HTTP
|
||||
def request(method, url, data=None, json=None, headers={}, stream=None, parse_headers=True, debug=False):
|
||||
redir_cnt = 1
|
||||
while True:
|
||||
try:
|
||||
proto, dummy, host, path = url.split("/", 3)
|
||||
except ValueError:
|
||||
proto, dummy, host = url.split("/", 2)
|
||||
path = ""
|
||||
if proto == "http:":
|
||||
port = 80
|
||||
elif proto == "https:":
|
||||
import ussl
|
||||
port = 443
|
||||
else:
|
||||
raise ValueError("Unsupported protocol: " + proto)
|
||||
if ":" in host:
|
||||
host, port = host.split(":", 1)
|
||||
port = int(port)
|
||||
ai = getaddrinfo(host, port, 0, SOCK_STREAM)
|
||||
ai = ai[0]
|
||||
resp_d = None
|
||||
if parse_headers is not False:
|
||||
resp_d = {}
|
||||
s = socket_d(ai[0], ai[1], ai[2], debug=debug)
|
||||
try:
|
||||
s.connect(ai[-1])
|
||||
if proto == "https:":
|
||||
s = ussl.wrap_socket(s, server_hostname=host)
|
||||
s.write(b"%s /%s HTTP/1.0\r\n" % (method, path))
|
||||
if not "Host" in headers:
|
||||
s.write(b"Host: %s\r\n" % host)
|
||||
for k in headers:
|
||||
s.write(k)
|
||||
s.write(b": ")
|
||||
s.write(headers[k])
|
||||
s.write(b"\r\n")
|
||||
if json is not None:
|
||||
assert data is None
|
||||
import ujson
|
||||
data = ujson.dumps(json)
|
||||
s.write(b"Content-Type: application/json\r\n")
|
||||
if data:
|
||||
s.write(b"Content-Length: %d\r\n" % len(data))
|
||||
s.write(b"Connection: close\r\n\r\n")
|
||||
if data:
|
||||
s.write(data)
|
||||
l = s.readline()
|
||||
l = l.split(None, 2)
|
||||
status = int(l[1])
|
||||
reason = ""
|
||||
if len(l) > 2:
|
||||
reason = l[2].rstrip()
|
||||
while True:
|
||||
l = s.readline()
|
||||
if not l or l == b"\r\n":
|
||||
break
|
||||
if l.startswith(b"Transfer-Encoding:"):
|
||||
if b"chunked" in l:
|
||||
raise ValueError("Unsupported " + l)
|
||||
elif l.startswith(b"Location:") and 300 <= status <= 399:
|
||||
if not redir_cnt:
|
||||
raise ValueError("Too many redirects")
|
||||
redir_cnt -= 1
|
||||
url = l[9:].decode().strip()
|
||||
status = 300
|
||||
break
|
||||
if parse_headers is False:
|
||||
pass
|
||||
elif parse_headers is True:
|
||||
l = l.decode()
|
||||
k, v = l.split(":", 1)
|
||||
resp_d[k] = v.strip()
|
||||
else:
|
||||
parse_headers(l, resp_d)
|
||||
except OSError:
|
||||
s.close()
|
||||
raise
|
||||
if status != 300:
|
||||
break
|
||||
resp = Response(s)
|
||||
resp.status_code = status
|
||||
resp.reason = reason
|
||||
resp.client_len=s.client_len
|
||||
resp.server_len=s.server_len
|
||||
if resp_d is not None:
|
||||
resp.headers = resp_d
|
||||
return resp
|
||||
|
||||
class MQTT_Client(MQTTClient):
|
||||
def __init__(self,*args,debug=False,**kw):
|
||||
super().__init__(*args,**kw)
|
||||
self.sock = socket_d(debug=debug)
|
||||
|
||||
@property
|
||||
def client_len(self): #The length of client data obtained
|
||||
_len=self.sock.client_len
|
||||
self.sock.client_len=0
|
||||
return _len
|
||||
|
||||
@property
|
||||
def server_len(self): #The length of server data obtained
|
||||
_len=self.sock.server_len
|
||||
self.sock.server_len=0
|
||||
return _len
|
||||
|
||||
def time_msg(self,utc=28800): #Get server time information
|
||||
msg=self.wait_msg()
|
||||
if isinstance(msg, dict):
|
||||
if msg['topic'] =='$SYS/hello':
|
||||
val=time.gmtime(int(msg['msg'])//1000-946684800+utc)[0:7]
|
||||
return str(val).replace(' ','')[1:-1]
|
||||
|
||||
#MQTT
|
||||
def init_MQTT_client(address, username, password, MQTT_USR_PRJ, debug=False):
|
||||
client = MQTT_Client(hexlify(unique_id()), address, 1883, username, password, debug=debug)
|
||||
client.set_last_will(topic=MQTT_USR_PRJ+WILL_TOPIC, msg=client.client_id, qos=2)
|
||||
if client.connect()==0:
|
||||
client.publish(MQTT_USR_PRJ+ADDITIONAL_TOPIC, client.client_id, qos=0)
|
||||
time.sleep_ms(200)
|
||||
return client
|
||||
Reference in New Issue
Block a user