2022/1/23

entrance/extend/orm.py

@@ -1,3 +1,43 @@
-from flask_sqlalchemy import SQLAlchemy
+import datetime
+from typing import Tuple
 
-db: SQLAlchemy = SQLAlchemy()
+from flask import request
+from flask_sqlalchemy import SQLAlchemy, BaseQuery
+
+
+class Query(BaseQuery):
+
+    def all_json(self) -> list:
+        print(self)
+        return [dict(i) for i in self.all()]
+
+    def soft_delete(self):
+        return self.update({"delete_at": datetime.datetime.now()})
+
+    def logic_all(self):
+        return self.filter_by(delete_at=None).all()
+
+    def layui_paginate(self):
+        """
+        layui表格分页
+        page
+        limit
+        """
+        return self.paginate(page=request.args.get('page', type=int),
+                             per_page=request.args.get('limit', type=int),
+                             error_out=False)
+
+    def layui_paginate_json(self) -> Tuple[list, int]:
+        """
+        layui表格分页
+        page
+        limit
+        """
+        p = self.paginate(page=request.args.get('page', type=int),
+                          per_page=request.args.get('limit', type=int),
+                          error_out=False)
+
+        return [dict(i) for i in p.items], p.total
+
+
+db = SQLAlchemy(query_class=Query)