mirror of
https://gitee.com/Lyon1998/pikapython.git
synced 2025-01-15 17:02:53 +08:00
86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
|
import hmac
|
||
|
|
import base64
|
||
|
|
import random
|
||
|
|
import mqtt
|
||
|
|
from PikaStdDevice import Time
|
||
|
|
|
||
|
|
|
||
|
|
class IOT:
|
||
|
|
def __init__(self):
|
||
|
|
self._signMethodTable = ["hmac-md5", "hmac-sha1", "hmac-sha256"]
|
||
|
|
|
||
|
|
def randStr(self, len):
|
||
|
|
a = ""
|
||
|
|
for i in range(len):
|
||
|
|
a = a + str(random.randint(0, 9))
|
||
|
|
return a
|
||
|
|
|
||
|
|
def getTimeStamp(self, t):
|
||
|
|
return int(Time.time()*1000) + t # todo 时间戳64位整形实现
|
||
|
|
|
||
|
|
def aliyun(self, clientId: str, productKey: str, deviceName: str, deviceSecret: str,
|
||
|
|
signMethod="hmac-md5", regionID="cn-shanghai", ssl=False):
|
||
|
|
if clientId == None or productKey == None or deviceName == None or deviceSecret == None:
|
||
|
|
print("[Error]input param is None")
|
||
|
|
return False
|
||
|
|
if signMethod not in self._signMethodTable:
|
||
|
|
print("[Error]not support signMethod")
|
||
|
|
return False
|
||
|
|
if ssl:
|
||
|
|
securemode = "2"
|
||
|
|
self.port = int(443)
|
||
|
|
else:
|
||
|
|
securemode = "3"
|
||
|
|
self.port = int(1883)
|
||
|
|
|
||
|
|
self.address = productKey + ".iot-as-mqtt." + regionID + ".aliyuncs.com"
|
||
|
|
self.username = deviceName + "&" + productKey
|
||
|
|
self.clientid = clientId + "|securemode=" + securemode + \
|
||
|
|
",signmethod="+signMethod.replace("-", "")+"|"
|
||
|
|
hmac_payload = "clientId" + clientId + "deviceName" + \
|
||
|
|
deviceName + "productKey" + productKey
|
||
|
|
self.password = hmac.new(deviceSecret.encode(),
|
||
|
|
msg=hmac_payload.encode(),
|
||
|
|
digestmod=signMethod).hexdigest()
|
||
|
|
return True
|
||
|
|
|
||
|
|
def tencentIotHub(self, productId, deviceName, deviceSecret, signMethod="hmac-sha1", expiryTime=3600, ssl=False):
|
||
|
|
if productId == None or deviceName == None or deviceSecret == None:
|
||
|
|
print("[Error]input param is None")
|
||
|
|
return False
|
||
|
|
if signMethod not in self._signMethodTable:
|
||
|
|
print("[Error]not support signMethod")
|
||
|
|
return False
|
||
|
|
connid = self.randStr(5)
|
||
|
|
expiry = self.getTimeStamp(expiryTime)
|
||
|
|
self.address = productId + ".iotcloud.tencentdevices.com"
|
||
|
|
self.port = int(1883)
|
||
|
|
self.clientid = productId + deviceName
|
||
|
|
self.username = self.clientid + ";12010126;" + connid + ";" + str(expiry)
|
||
|
|
token = hmac.new(base64.b64decode(deviceSecret.encode()),
|
||
|
|
msg=self.username.encode(),
|
||
|
|
digestmod=signMethod).hexdigest()
|
||
|
|
self.password = token + ";" + signMethod.replace("-", "")
|
||
|
|
return True
|
||
|
|
|
||
|
|
def onenet(self): ...
|
||
|
|
|
||
|
|
def connect(self, keepalive=600):
|
||
|
|
self._client = mqtt.MQTT(self.address, port=self.port, clinetID=self.clientid,
|
||
|
|
username=self.username, password=self.password, keepalive=keepalive)
|
||
|
|
return self._client.connect()
|
||
|
|
|
||
|
|
def disconnect(self):
|
||
|
|
return self._client.disconnect()
|
||
|
|
|
||
|
|
def subsribe(self, topic, cb, qos=0):
|
||
|
|
return self._client.subscribe(topic, qos, cb)
|
||
|
|
|
||
|
|
def publish(self, topic, payload, qos=0):
|
||
|
|
return self._client.publish(qos, topic, payload)
|
||
|
|
|
||
|
|
|
||
|
|
def new():
|
||
|
|
iot = IOT()
|
||
|
|
return iot
|