Forráskód Böngészése

上传文件至 'classify-ai-backend'

qiufb 1 éve
szülő
commit
1d3fde9511
1 módosított fájl, 10 hozzáadás és 6 törlés
  1. 10 6
      classify-ai-backend/main.py

+ 10 - 6
classify-ai-backend/main.py

@@ -14,10 +14,11 @@ CONNECT_TYPE = 'connect'
 ERROR_TYPE = 'error'
 CLOSE_TYPE = 'close'
 XLS_TYPE = 'xls'
-DB_TYPE = 'db'
-TXT_TYPE = 'txt'
 XLSX_TYPE = 'xlsx'
-XML_TYPE = 'xml'
+TXT_TYPE = 'txt'
+WAV_TYPE = 'wav'
+JSON_TYPE = 'json'
+NPY_TYPE = 'npy'
 
 
 class MessageOperation:
@@ -35,16 +36,19 @@ class MessageOperation:
             if self.msg_type == XLS_TYPE:
                 self.response_status, self.response_content = car_classification(
                     self.msg_content)
-            elif self.msg_type == DB_TYPE:
+            elif self.msg_type == XLSX_TYPE:
                 self.response_status, self.response_content = car_classification(
                     self.msg_content)
             elif self.msg_type == TXT_TYPE:
                 self.response_status, self.response_content = car_classification(
                     self.msg_content)
-            elif self.msg_type == XLSX_TYPE:
+            elif self.msg_type == WAV_TYPE:
+                self.response_status, self.response_content = car_classification(
+                    self.msg_content)
+            elif self.msg_type == JSON_TYPE:
                 self.response_status, self.response_content = car_classification(
                     self.msg_content)
-            elif self.msg_type == XML_TYPE:
+            elif self.msg_type == NPY_TYPE:
                 self.response_status, self.response_content = car_classification(
                     self.msg_content)
             elif self.msg_type == CLOSE_TYPE: