liyuyi@c-top.com.cn 4 năm trước cách đây
mục cha
commit
6dbd1e5cd1
3 tập tin đã thay đổi với 34 bổ sung30 xóa
  1. 10 5
      asr_client.py
  2. 19 12
      database.py
  3. 5 13
      main.py

Những thai đổi đã bị hủy bỏ vì nó quá lớn
+ 10 - 5
asr_client.py


+ 19 - 12
database.py

@@ -1,21 +1,23 @@
 # 导入:
-from hashlib import md5
-from sqlalchemy.orm import sessionmaker
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy import Column, Integer, String,DateTime,Text,create_engine,and_
 from datetime import datetime
 from urllib import parse
 
+from sqlalchemy import Column, Integer, String, DateTime, Text, create_engine, and_
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
 # 创建对象的基类:
 Base = declarative_base()
 passowrd = parse.quote_plus('hcst@2020')
+
+
 # 定义User对象:
 class Task(Base):
     # 表的名字:
     __tablename__ = 'tb_asr_result'
 
     # 表的结构:
-    md5 =  Column(String(100))
+    md5 = Column(String(100))
     task_id = Column(Integer, primary_key=True)
     task_status = Column(Integer)
     task_result = Column(Text)
@@ -24,13 +26,15 @@ class Task(Base):
     create_time = Column(DateTime, nullable=True, default=datetime.now)
     update_time = Column(DateTime)
 
+
 # 初始化数据库连接:
-engine = create_engine('mysql+mysqlconnector://hcst:'+passowrd+'@139.186.165.84:3306/db_ai_word')
+engine = create_engine('mysql+mysqlconnector://hcst:' + passowrd + '@139.186.165.84:3306/db_ai_word')
 # 创建DBSession类型:
 DBSession = sessionmaker(bind=engine)
 
-def query(md5,task_status,task_id):
-    condition = (1==1)
+
+def query(md5, task_status, task_id):
+    condition = (1 == 1)
     if md5 != None:
         condition = and_(condition, Task.md5 == md5)
     if task_status != None:
@@ -46,7 +50,8 @@ def query(md5,task_status,task_id):
     session.close()
     return task
 
-def insert(task:Task):
+
+def insert(task: Task):
     # 创建session对象:
     session = DBSession()
     # 添加到session:
@@ -58,10 +63,11 @@ def insert(task:Task):
     session.close()
     return task
 
+
 def update(task):
     # 创建session对象:
     session = DBSession()
-    taskdb = session.query(Task).filter(Task.task_id==task.task_id).one()
+    taskdb = session.query(Task).filter(Task.task_id == task.task_id).one()
     taskdb.task_status = task.task_status
     taskdb.word_text = task.word_text
     taskdb.word_split = task.word_split
@@ -71,10 +77,11 @@ def update(task):
     # 关闭session:
     session.close()
 
+
 if __name__ == '__main__':
     print('test query')
     # task = Task(task_id = '3',task_status=1,md5='dx')
     # task = insert(task)
-    task = query(None,None,1)[0]
+    task = query(None, None, 1)[0]
     print(task.md5)
-    #update(task)
+    # update(task)

+ 5 - 13
main.py

@@ -1,20 +1,12 @@
-from typing import Optional
-from fastapi import FastAPI
 from concurrent.futures import ThreadPoolExecutor
-from sqlalchemy.sql.elements import Null
-from sqlalchemy.sql.expression import null
-from datetime import date
-import requests
-import uuid
-import os
+from typing import Optional
+
 import uvicorn
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
 from asr_client import send_asr_request, send_task_request
 from database import insert, update, query, Task
-from datetime import datetime
-from fastapi.middleware.cors import CORSMiddleware
-import hashlib
-import traceback
-import json
 
 threadPool = ThreadPoolExecutor(max_workers=4)
 app = FastAPI()