设置中心增加数据库升级按钮

controllers/admin.py

@@ -93,6 +93,19 @@ def admin_get_ver():
     online_ver,msg = getOnlineVer()
     return jsonify({'local_ver':getLocalVer(),'online_ver':online_ver,'msg':msg})
 
+@admin.route('/update_db')
+def admin_update_db():
+    if not verfy_token():
+        # return render_template('login.html')
+        return R.error('请登录后再试')
+    cmd = 'flask db migrate && flask db upgrade'
+    if not os.path.exists('migrations'):
+        cmd = 'flask db init && '+cmd
+    print(f'开始执行cmd:{cmd}')
+    result = os.system(cmd)
+    print(f'cmd执行结果:{result}')
+    return R.success('数据库升级完毕')
+
 @admin.route('/update_ver')
 def admin_update_ver():
     if not verfy_token():

models/ruleclass.py

@@ -5,6 +5,7 @@
 # Date  : 2022/9/6
 
 from base.database import db
+from datetime import datetime
 
 class RuleClass(db.Model):
     __tablename__ = 'rule_class'
@@ -13,6 +14,10 @@ class RuleClass(db.Model):
     class_name = db.Column(db.String(255))
     class_url = db.Column(db.String(255))
     cookie = db.Column(db.String(255))
+    state = db.Column(db.Integer, default=0)
+    order = db.Column(db.Integer, default=0)
+    create_date = db.Column(db.DateTime, index=True, default=datetime.now)
+    write_date = db.Column(db.DateTime, index=True, default=datetime.now,onupdate=datetime.now)
 
     def __repr__(self):
         return "<RuleClass(name='%s', class_name='%s', class_url='%s',cookie='%s')>" % (

templates/settings.html

@@ -46,6 +46,19 @@
       $(".save_conf1").click(function () {
         console.log(this);
       });
+      $('#update_db').click(function (){
+        $.get("/admin/update_db",function(data,status){
+                   console.log(data);
+                   if(data.code === 200){
+                       alert(data.msg);
+                      // location.reload();
+                   }else{
+                       alert(data.msg);
+                       console.log('升级失败了...');
+                       return false
+                   }
+           });
+      });
     });
 </script>
 <h2 class="title">欢迎使用DR-PY配置界面<div><span class="ver_title">当前版本: {{ ver }}</span><span
@@ -54,6 +67,8 @@
 <h4>
     <button type="button" class="yongyin"><a href="/admin">返回管理页</a></button>
 <!--    <button type="button" class="yongyin"><a href="/index">返回首页</a></button>-->
+  <button type="button" class="yongyin3"><a href="javascript:void(0);"
+            class="funcbtn" id="update_db">升级数据库</a></button>
   <button type="button" class="yongyin2"><a href="javascript:copy('cp ./tmp/dr_py-master/app.py ./app.py');"
             class="funcbtn" id="update_by_self">升级指令</a></button>
     <button type="button" class="yongyin2"><a href="javascript:copy('cp ./tmp/dr_py-master/base/custom.conf base');"