解决flask路由调用函数的尴尬问题

最新推荐文章于 2024-07-24 14:51:00 发布

Noah747

最新推荐文章于 2024-07-24 14:51:00 发布

阅读量1.5k

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/crystal_shrimps/article/details/107190467

版权

本文介绍了在Flask应用中遇到的创建评估任务与扫描功能同步问题，详细探讨了三种解决办法：1）使用异步调用，通过装饰器实现多线程；2）尝试利用全局变量，但发现路由执行后不会继续执行其他函数；3）尝试try-finally结构，但由于finally的执行顺序问题，仍无法解决问题。最终，作者采用了异步调用来成功地实现了先返回结果再执行扫描的目标。

摘要由CSDN通过智能技术生成

背景

遇到一个问题，之前写了创建评估任务–调用扫描–结果提交漏洞这么一个功能，运行的时候发现创建完评估任务提交一个请求直接执行扫描，不扫完就没有return结果，所以造成了没扫描完不能成功创建评估任务的尴尬局面。

脚本的正常逻辑应该是提交请求–判断合法性–返回结果–执行扫描–提交漏洞

解决办法

先说最后解决的方法

1、最终解决办法–异步调用

参考：python调用异步函数执行
写个装饰器定义多线程函数，在扫描脚本前加装饰器。

scan函数能跟接收请求的svnaddr函数异步执行，这样也能达到先return再scan的效果

def myasync(f):
    def wrapper(*args, **kwargs):
        thr = Thread(target=f, args=args, kwargs=kwargs)
        thr.start()
        
    return wrapper
    

@app.route('/svnaddr', methods=['POST'])
def svnaddr():
    if flag == False:
        # 判断参数合法性               
        code = 1
        message = str(taskid)+' create success !'
        handleScan(taskid, repo, model, scantime, status)
    ret