diff --git a/web/admin/site/site.py b/web/admin/site/site.py
index 7fc98afe7..84720f5e1 100644
--- a/web/admin/site/site.py
+++ b/web/admin/site/site.py
@@ -251,6 +251,16 @@ def get_rewrite_tpl():
tplname = request.form.get('tplname', '')
return MwSites.instance().getRewriteTpl(tplname)
+# 设置站点Rewrite
+@blueprint.route('/set_rewrite', endpoint='set_rewrite',methods=['POST'])
+@panel_login_required
+def set_rewrite():
+ data = request.form.get('data', '')
+ path = request.form.get('path', '')
+ encoding = request.form.get('encoding', '')
+ return MwSites.instance().setRewrite(path,data,encoding)
+
+
# 设置Rewrite模版名
@blueprint.route('/set_rewrite_tpl', endpoint='set_rewrite_tpl',methods=['POST'])
@panel_login_required
diff --git a/web/utils/site.py b/web/utils/site.py
index bd1e9bc1b..56b0c333e 100644
--- a/web/utils/site.py
+++ b/web/utils/site.py
@@ -616,6 +616,20 @@ class sites(object):
return mw.returnData(False, '模版不存在!')
return mw.returnData(True, 'OK', path)
+ def setRewrite(self,path,data,encoding):
+ if not os.path.exists(path):
+ mw.writeFile(path, '')
+
+ mw.backFile(path)
+ mw.writeFile(path, data)
+ isError = mw.checkWebConfig()
+ if(type(isError) == str):
+ mw.restoreFile(path)
+ msg = 'ERROR:
' + isError.replace("\n", '
') + ''
+ return mw.returnJson(False, msg)
+ mw.restartWeb()
+ return mw.returnData(True, '设置成功!')
+
def setRewriteTpl(self,name,data):
path = self.getNgxRewriteDir() +'/'+ name + ".conf"
if os.path.exists(path):