update

web/admin/setting/__init__.py

@@ -9,6 +9,8 @@
 # ---------------------------------------------------------------------------------
 
 import re
+import json
+import os
 
 from flask import Blueprint, render_template
 from flask import request
@@ -17,6 +19,7 @@ from admin import model
 from admin.user_login_check import panel_login_required
 
 import core.mw as mw
+import utils.config as utils_config
 
 # 默认页面
 blueprint = Blueprint('setting', __name__, url_prefix='/setting', template_folder='../../templates')
@@ -25,8 +28,6 @@ blueprint = Blueprint('setting', __name__, url_prefix='/setting', template_folde
 def index():
     return render_template('default/setting.html')
 
-
-
 @blueprint.route('/get_panel_list', endpoint='get_panel_list', methods=['POST'])
 @panel_login_required
 def get_panel_list():
@@ -106,5 +107,61 @@ def set_admin_path():
 
 
 
+# 设置BasicAuth认证
+@blueprint.route('/set_basic_auth', endpoint='set_basic_auth', methods=['POST'])
+@panel_login_required
+def set_basic_auth():
+    basic_user = request.form.get('basic_user', '').strip()
+    basic_pwd = request.form.get('basic_pwd', '').strip()
+    basic_open = request.form.get('is_open', '').strip()
+    
+    __file = mw.getCommonFile()
+    path = __file['basic_auth']
+
+    is_open = True
+    if basic_open == 'false':
+        is_open = False
+
+    if basic_open == 'false':
+        model.setOption('basic_auth', json.dumps({'open':False}))
+        mw.writeLog('面板设置', '设置BasicAuth状态为: %s' % is_open)
+        return mw.returnData(True, '删除BasicAuth成功!')
+
+    if basic_user == '' or basic_pwd == '':
+        return mw.returnData(False, '用户和密码不能为空!')
+
+    salt = mw.getRandomString(6)
+    data = {}
+    data['salt'] = salt
+    data['basic_user'] = mw.md5(basic_user + salt)
+    data['basic_pwd'] = mw.md5(basic_pwd + salt)
+    data['open'] = is_open
+
+    model.setOption('basic_auth', json.dumps(data))
+    mw.writeLog('面板设置', '设置BasicAuth状态为: %s' % is_open)
+    return mw.returnData(True, '设置成功!')
+
+
+# 默认站点目录
+@blueprint.route('/set_status_code', endpoint='set_status_code', methods=['POST'])
+@panel_login_required
+def set_status_code():
+    status_code = request.form.get('status_code', '').strip()
+    if re.match(r"^\d+$", status_code):
+        status_code = int(status_code)
+        if status_code != 0:
+            if status_code < 100 or status_code > 999:
+                return mw.returnData(False, '状态码范围错误!!')
+    else:
+        return mw.returnData(False, '状态码范围错误!')
+
+    info = utils_config.getUnauthStatus(code=str(status_code))
+    model.setOption('unauthorized_status', str(status_code))
+    mw.writeLog('面板设置', '将未授权响应状态码设置为:{0}:{1}'.format(status_code,info['text']))
+    return mw.returnData(True, '设置成功!')
+
+
+
+