物联网控制台数据同步--客户端Python脚本

import json
from http.server import BaseHTTPRequestHandler, HTTPServer
import logging
import hashlib
from sys import argv

#用户设置token
token = "test"

class Handler(BaseHTTPRequestHandler):
    def _set_response(self):
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):#处理get请求
        logging.info("GET request,\nPath: %s\nHeaders:\n%s\n", str(self.path), str(self.headers))#打印get请求
        if str(self.path) == "/api/explorer":#判断URL的路径
            a = self.checkSignature(token)
            self.send_response(200)
            self.send_header('Content-type', 'text/plain')
            self.end_headers()
            self.wfile.write(bytes(a,encoding = "utf8"))#返回验证参数

    def do_POST(self):
        content_length = int(self.headers['Content-Length'])  # <--- Gets the size of data
        post_data = self.rfile.read(content_length)  # <--- Gets the data itself
        logging.info("POST request,\nPath: %s\nHeaders:\n%s\n\nBody:\n%s\n",
                     str(self.path), str(self.headers), post_data.decode('utf-8'))#打印post请求
        if str(self.path) == "/api/explorer":#判断URL路径
            self.explorerHandle(post_data)#处理数据
            return
        data = json.dumps({'code': 404, 'status': 'not found'})
        self.send_response(404)
        self.send_header('Content-type', 'application/json')
        self.end_headers()
        self.wfile.write(data.encode('utf-8'))

    def explorerHandle(self, body):#处理数据函数
        # 处理explorer请求
        file = open("device_report.txt", 'w')
        res =json.dumps(json.loads(body),cls=MyEncoder)#转换格式  bytes -> dict
        file.write(res)  # 写入文件

    def checkSignature(self,token):#模拟控制台加密算法生成数字签名与请求中的签名相对比
        signature = self.headers.get_all("signature")[0]
        Echostr = self.headers.get_all("Echostr")
        s = []
        s.append(token)
        s.append(self.headers.get_all("timestamp")[0])
        s.append(self.headers.get_all("nonce")[0])
        s.sort()
        tmpArr = s[0] + s[1] + s[2]
        
        sha1 = hashlib.sha1()
        sha1.update(tmpArr.encode('utf-8'))
        res = sha1.hexdigest()
        
        if res == signature:
            return Echostr[0]
        self.send_response(401)

class MyEncoder(json.JSONEncoder):#将post请求的参数由HTTPMessage转换成json
    def default(self, obj):
        if isinstance(obj, bytes):
            return str(obj, encoding='utf-8');
        return json.JSONEncoder.default(self, obj)


def run(server_class=HTTPServer, handler_class=Handler, port=3000):
    logging.basicConfig(level=logging.INFO)
    server_address = ('0.0.0.0', port)
    httpd = server_class(server_address, handler_class)
    logging.info('Starting httpd...\n')
    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        pass
    httpd.server_close()
    logging.info('Stopping httpd...\n')


if __name__ == '__main__':
   
    if len(argv) == 2:
        run(port=int(argv[1]))
    else:
        run()

GET / HTTP/1.1
Host: **.**.**.**:4443
User-Agent: Go-http-client/1.1
Content-Type: application/json
Echostr: UPWIAFASvDUFcTEE
Nonce: testrance
Signature: abb6c316a8134596d825c5a1295bfa6f7657664d
Timestamp: 1623149590
Accept-Encoding: gzip

HTTP/1.1 200 OK
Date: Tue, 08 Jun 2021 10:53:10 GMT
Content-Length: 16
Content-Type: text/plain; charset=utf-8

UPWIAFASvDUFcTEE

{
"payload": {
    "clientToken": "DEP****YAS4-38",
    "method": "report",
    "params": {
        "brightness": 58,
        "color": 2,
        "name": "dev001",
        "power_switch": 1
    }
},
"seq": 1000000000,
"timestamp": 1579055948,
"topic": "$thing/up/property/DEP****YAS4/dev001",
"devicename": "dev001",
"productid": "DEP****YAS4"
}