开启数据权限

介绍

数据权限相关说明：

数据权限针对具体的视图集生效，可以针对某个视图集进行开启或关闭。默认为关闭，开启步骤如下。

修改模型

在需要开启数据权限的模型，添加数据归属部门id和数据归属用户id

示例：mysite\myapp_demo\book\models.py

python

class DemoBook(BaseModel):
    # ...

    # （添加下面代码）
    # 数据归属的部门与用户（开启数据权限时需要添加）
    dept_id = models.BigIntegerField(
        null=True, blank=True, db_comment="数据归属部门ID", help_text="数据归属部门ID"
    )
    user_id = models.BigIntegerField(
        null=True, blank=True, db_comment="数据归属用户ID", help_text="数据归属用户ID"
    )

修改后，生成数据库迁移文件，并迁移

python

# myapp_demo应用生成数据库表，将myapp_demo修改为您的应用
python manage.py makemigrations myapp_demo
python manage.py migrate myapp_demo

修改序列化器

上一步模型修改后，修改对应的序列化器，添加数据归属部门与用户字段

示例：mysite\myapp_demo\book\serializers.py

python

class DemoBookSerializer(serializers.ModelSerializer):
    """图书序列化器"""

    class Meta:
        model = DemoBook
        fields = [
            "id",
            "title",
            "author",
            "publisher",
            "publish_date",
            "category",
            "total_copies",
            "price",
            "create_time",
            "dept_id",  # 数据归属部门
            "user_id",  # 数据归属用户
        ]
        read_only_fields = ["id", "create_time"]

修改视图

修改需要开启数据权限的视图

示例：mysite\myapp_demo\book\views.py

python

@extend_schema(tags=["管理后台-demo-图书"])
class DemoBookViewSet(CustomModelViewSetNoSimple):
    # ...
    
    
    # （添加下面代码）
    # 开启数据权限
    data_permissions = {"dept_column": "dept_id", "user_column": "user_id"}
    # 重写perform_create方法，补充数据归属部门与用户
    def perform_create(self, serializer):
        return serializer.save(
            creator=self.request.user.id,
            updater=self.request.user.id,
            dept_id=(
                self.request.user.dept_id.id
                if hasattr(self.request.user, "dept_id")
                else None
            ),
            user_id=self.request.user.id,
        )