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)

entrance/migrations/versions/d51040631f21_.py

@@ -0,0 +1,32 @@
+"""empty message
+
+Revision ID: d51040631f21
+Revises: 74133ba99af7
+Create Date: 2022-01-22 23:30:04.372043
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'd51040631f21'
+down_revision = '74133ba99af7'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('admin_test',
+    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False, comment='ID'),
+    sa.Column('name', sa.String(length=20), nullable=True, comment='名'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('admin_test')
+    # ### end Alembic commands ###