Routes.py 文件源码

def uploadPlugins():
    f = request.files['file']
    file_name = ''
    try:
        if f:
            fname = secure_filename(f.filename)
            suffix = fname.split('.')[-1]
            if suffix in ['py','json']:
                path = FILE_PATH + fname
                if os.path.exists(FILE_PATH + fname):
                    fname = fname.split('.')[0] + '_' + str(datetime.now().second) + "." + suffix
                    path = FILE_PATH + fname
                f.save(path)
                if os.path.exists(path):
                    file_name = fname.split('.')[0]
                    # redis publish
                    with open(path) as pf:
                        r = getStrictRedis()
                        r.publish('updateplugins', json.dumps({"filename":file_name+"."+suffix, "content":pf.read()}))
                    # update esplugins
                    vulScan.updatePlugins(file_name, suffix)

        return jsonify({"message":"ok"})
    except Exception as e:
        pass