feature: 新增数据模型

2023-09-27 17:49:13 +08:00
parent 4db050c93d
commit 9bf89e5919
9 changed files with 210 additions and 2 deletions
@@ -0,0 +1,54 @@
+from pear_admin.extensions import db
+
+from ._base import BaseORM
+
+
+class RightsORM(BaseORM):
+    __tablename__ = "ums_rights"
+
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(20), nullable=False, comment="权限名称")
+    code = db.Column(db.String(30), comment="权限标识")
+    type = db.Column(db.String(30), comment="权限类型")
+    url = db.Column(db.String(30), comment="路径地址")
+
+    icon = db.Column(db.String(128), comment="图标")
+    status = db.Column(db.Boolean, default=True, comment="是否开启")
+    sort = db.Column(db.Integer, default=1)
+    open_type = db.Column(db.String(128), comment="打开方式")
+    pid = db.Column(
+        db.Integer,
+        db.ForeignKey("ums_rights.id"),
+        default=0,
+        comment="父类编号",
+    )
+
+    # parent = db.relationship("RightsORM", remote_side=[pid])  # 自关联
+
+    def json(self):
+        return {
+            "id": self.id,
+            "name": self.name,
+            "code": self.code,
+            "type": self.type,
+            "url": self.url,
+            "icon": self.icon,
+            "status": self.status,
+            "sort": self.sort,
+            "open_type": self.open_type,
+            "pid": self.pid,
+        }
+
+    def menu_json(self):
+        type_map_dict = {"menu": 0, "path": 1}
+
+        return {
+            "id": self.id,
+            "pid": self.pid,
+            "rights_id": self.id,
+            "title": self.name,
+            "type": type_map_dict[self.type],
+            "href": self.url,
+            "icon": self.icon,
+            "sort": self.sort,
+        }