Python - 在 Flask 中将查询结果从服务器返回给客户端

Python - Returning query result from server to client in Flask

这是我目前所做的。

app = Flask(__name__)
api = Api(app)

# Variable to store the result file count in the Tool directory
fileCount = 0

# Variable to store the query result generated by the Tool
queryResult = 0    
    
# Method to read .txt files generated by the Tool
def readFile():
    global fileCount
    global queryResult
    # Path where .txt files are created by the Tool
    path = "<path>"
    tempFileCount = len(fnmatch.filter(os.listdir(path), '*.txt'))
    if (fileCount != tempFileCount):
        fileCount = tempFileCount
        list_of_files = glob.iglob(path + '*.txt')
        latest_file = max(list_of_files, key=os.path.getctime)
        print("\nLast modified file: " + latest_file)
        with open(latest_file, "r") as myfile:
            queryResult = myfile.readlines()
            print(queryResult) # I would like to return this queryResult to the client
            
scheduler = BackgroundScheduler()
scheduler.add_job(func=readFile, trigger="interval", seconds=10)
scheduler.start()   

# Shut down the scheduler when exiting the app
atexit.register(lambda: scheduler.shutdown())   
   
# Method to write url parameters in JSON to a file
def write_file(response):
    time_stamp = str(time.strftime("%Y-%m-%d_%H-%M-%S"))  
    with open('data' + time_stamp + '.json', 'w') as outfile:
        json.dump(response, outfile)
    print("JSON File created!")
   

class GetParams(Resource):
    def get(self):
        response = json.loads(list(dict(request.args).keys())[0])  
        write_file(response)  
           
api.add_resource(GetParams, '/data')  # Route for GetJSON()

if __name__ == '__main__':
    app.run(port='5890', threaded=True)
data = {
    'query': 'SELECT * FROM table_name'
}

url = 'http://127.0.0.1:5890/data'

session = requests.Session()
retry = Retry(connect=3, backoff_factor=0.5)  
adapter = HTTPAdapter(max_retries=retry)
session.mount('http://', adapter)
session.mount('https://', adapter)

resp = session.get(url, params=json.dumps(data))
print(resp)

任何人都可以帮助我如何将此查询结果发送回客户端?

编辑:我希望服务器每次在 Tool 目录中遇到新文件时将 queryResult 发送回客户端,即,每次找到新文件时,它都会提取结果(它正在这样做当前)并将其发送回客户端。

你想做的事情叫做Web Worker Architecture

要将实时摘要 queryResult 从后台作业传递到客户端应用程序,您可以结合使用消息队列(推荐使用 Kafka,RabbitMQ 也可以)和 Web-Socket .当客户端向 /data 端点发送请求时,您应该 return 返回一些唯一的令牌(如果您的用户是匿名的,则返回 UUID,如果经过身份验证,则返回用户 ID)。您应该将相同的标记添加到结果文件的名称中。当您的后台工作人员完成文件处理后,它使用令牌(来自文件名)创建 Kafka 或 RabbitMQ 主题,如 topic_for_user_id_1337topic_for_uuid_jqwfoj-123qwr,并将 queryResult 作为消息发布。

同时,你的客户端应该建立一个 web-socket 连接(Flask 对 web-sockets 来说很糟糕,但无论如何也没有几个好的库可以做到这一点,比如 socketio)并通过它传递令牌到您的后端,因此它将创建一个消息队列订阅者,订阅一个带有令牌名称的主题,因此当后台作业完成时,网络后端将接收一条消息并通过网络套接字将其传递给用户。

P.S。如果听起来过于复杂,您可以避免使用 MQ 和 WS,将 queryResult 放入数据库并创建端点以检查它是否存在于数据库中。如果没有,您 return 类似于 not ready yet 并且客户端会在几秒钟后重试,如果它已准备就绪 - 您 return 来自数据库的 queryResult