diff --git a/app.py b/app.py
index 664083949fbb03959bcc5c07658ac1e7c70f0c35..eb57ba074ab25636caec66ecec011692cc968dea 100644
--- a/app.py
+++ b/app.py
@@ -15,6 +15,7 @@ warnings.filterwarnings('ignore')
import os
from flask import Flask, jsonify, abort,request,redirect,make_response,render_template,send_from_directory,url_for
+from werkzeug.utils import secure_filename
from js.rules import getRuleLists
from utils import error,parser
from utils.web import *
@@ -117,6 +118,26 @@ def login_api():
else:
return jsonify(error.failed('登录失败,用户名或密码错误'))
+@app.route('/upload', methods=['GET', 'POST'])
+def upload_file():
+ cookies = request.cookies
+ # print(cookies)
+ token = cookies.get('token', '')
+ # print(f'mytoken:{token}')
+ if not verfy_token(token):
+ return render_template('login.html')
+ if request.method == 'POST':
+ f = request.files['file']
+ # print(request.files)
+ filename = secure_filename(f.filename)
+ savePath = f'js/{filename}'
+ # print(savePath)
+ f.save(savePath)
+ return jsonify(error.success('文件上传成功'))
+ else:
+ # return render_template('upload.html')
+ return jsonify(error.failed('文件上传失败'))
+
@app.route('/vod')
def vod():
t0 = time()
diff --git a/templates/admin.html b/templates/admin.html
index 5b1be956482e1e16ec88a51b1e1a1ed8df63db64..dfd2aec0141e21b3c1b834e6273e55553898a91a 100644
--- a/templates/admin.html
+++ b/templates/admin.html
@@ -12,5 +12,9 @@
欢迎使用py_dr管理员界面
你可以在此页面在线上传规则文件到js目录或者删除js目录的文件
+