Эх сурвалжийг харах

添加获取IotCore的临时证书和私钥1

tanghongbin 4 жил өмнө
parent
commit
c371d9e9f2

+ 20 - 13
Controller/IotCoreController.py

@@ -27,21 +27,28 @@ class IotCoreView(View):
         response = ResponseObject()
 
         if operation == 'createProvisioningClaim':
-            return self.create_provisioning_claim(response)
+            return self.create_provisioning_claim(request_dict, response)
         else:
             return response.json(404)
 
-    def create_provisioning_claim(self, response):
-        client = boto3.client('iot', region_name=AWS_IOT_SES_ACCESS_REGION, aws_access_key_id=AWS_IOT_SES_ACCESS_ID,
-                              aws_secret_access_key=AWS_IOT_SES_ACCESS_SECRET)
-        result = client.create_provisioning_claim(templateName='Ansjer_Iot_Queue')
+    def create_provisioning_claim(self, request_dict, response):
 
-        res = {
-            'certificateId': result['certificateId'],
-            'certificatePem': result['certificatePem'],
-            'publicKey': result['keyPair']['PublicKey'],
-            'privateKey': result['keyPair']['PrivateKey'],
-            'endpoint': 'a2rqy12o004ad8-ats.iot.us-east-1.amazonaws.com'
-        }
+        country = request_dict.get('country', None)
+        state = request_dict.get('state', None)
 
-        return response.json(0, {'res': res})
+        if country and state:
+
+            client = boto3.client('iot', region_name=AWS_IOT_SES_ACCESS_REGION, aws_access_key_id=AWS_IOT_SES_ACCESS_ID,
+                                  aws_secret_access_key=AWS_IOT_SES_ACCESS_SECRET)
+            result = client.create_provisioning_claim(templateName='Ansjer_Iot_Queue')
+
+            res = {
+                'certificateId': result['certificateId'],
+                'certificatePem': result['certificatePem'],
+                'publicKey': result['keyPair']['PublicKey'],
+                'privateKey': result['keyPair']['PrivateKey'],
+                'endpoint': 'a2rqy12o004ad8-ats.iot.us-east-1.amazonaws.com'
+            }
+            return response.json(0, {'res': res})
+        else:
+            return response.json(444)