update

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

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: <br><a style="color:red;">' + isError.replace("\n", '<br>') + '</a>'
+            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):