class User(models.Model):
 name = models.CharField(max_length=255)
 room = models.ForeignKey(Room, db_constraint=False)
class Room(models.Model):
 status = models.IntegerField(default=1)

# -*- coding:UTF-8 -*-
from django.db import models
#導(dǎo)入django自帶的User模型進行擴展
from django.contrib.auth.models import AbstractUser
 
class Role(models.Model):
  """
  用戶角色
  """
  role_name = models.CharField(max_length=100,verbose_name="角色名",help_text="角色名")
 
  class Meta:
    verbose_name = "角色"
    verbose_name_plural = verbose_name
    #用于指定不同的app使用不同的數(shù)據(jù)庫
    # app_label = "users"
    #使用自定義指定的表明jt_role
    db_table = "jt_role"
 
  def __str__(self):
    return self.role_name
 
class Department(models.Model):
  """
  部門
  """
  depat_name = models.CharField(max_length=64, verbose_name="部門名稱", help_text="部門名稱")
 
  class Meta:
    verbose_name = "部門"
    verbose_name_plural = verbose_name
    #用于指定不同的app使用不同的數(shù)據(jù)庫
    # app_label = "users"
    #使用自定義指定的表明jt_role
    db_table = "jt_department"
 
  def __str__(self):
    return self.depat_name
 
class UserProfile(models.Model):
  """
  在Django的User模型上進行拓展,id字段使用id
  """
  name = models.CharField(max_length=64, verbose_name="姓名")
  depat_id = models.IntegerField(verbose_name="部門id")
 
  class Meta:
    verbose_name = "用戶"
    verbose_name_plural = verbose_name
    #用于指定不同的app使用不同的數(shù)據(jù)庫
    # app_label = "users"
    db_table = "jt_users"
 
class UserRole(models.Model):
  """
  用戶角色關(guān)系，為提高性能，不使用manytomany來實現(xiàn)
  """
  user_id = models.IntegerField(verbose_name="用戶id")
  role_id = models.IntegerField(verbose_name="角色id")
  is_delete = models.BooleanField(verbose_name="是否邏輯刪除", default=False)
 
  class Meta:
    verbose_name = "用戶角色關(guān)系"
    verbose_name_plural = verbose_name
    # 使用自定義指定的表明jt_user_role
    db_table = "jt_user_role"

from rest_framework import serializers
from .models import UserRole,UserProfile,Role,Department
 
class UserDetailSerializer(serializers.Serializer):
  """
  用戶詳情序列表類
  """
  name = serializers.CharField()
  depat_name = serializers.SerializerMethodField()
  roles = serializers.SerializerMethodField()
  class Meta:
    model = UserProfile
    fields = ("name", "depat_name","roles")
 
  #重點中的重點
  def get_roles(self, obj):
    """
    自定義獲取多對多數(shù)據(jù)
    :param obj: 當(dāng)前user的實例
    :return: 當(dāng)前用戶的全部角色(數(shù)組)
    :思路：先通過當(dāng)前的用戶，查詢用戶角色關(guān)系表，獲得全部的角色id，再通過角色id獲得角色名
    """
    user = obj
    role_ids = UserRole.objects.filter(user_id__exact=user.id).values_list('role_id').all()
    roles = Role.objects.filter(id__in=role_ids).all()
    ret = [ ]
    for item in roles:
      ret.append(item.role_name)
    return ret
 
  def get_depat_name(self,obj):
    """
    獲取部門名稱
    :param obj: 當(dāng)前user的實例
    :return: 當(dāng)前用戶所在部門名稱
    """
    user = obj
    depat_name = Department.objects.filter(id = user.depat_id)[0].depat_name
    return depat_name