from django.db import models
"""
基類,其他類繼承即可獲得對應(yīng)的字段
"""
class BaseModel(models.Model):
updated_tm = models.DateTimeField(auto_now=True)
created_tm = models.DateTimeField(auto_now_add=True)
class Meta:
abstract = True
"""
項(xiàng)目列表
"""
class ProjectList( BaseModel):
project_id = models.AutoField(help_text="項(xiàng)目ID", primary_key=True)
project_name = models.SlugField(max_length=128, help_text="項(xiàng)目名稱", unique=True)
class Meta:
db_table = 'project_list'
verbose_name = '項(xiàng)目列表信息'
verbose_name_plural = verbose_name
"""
項(xiàng)目詳情
"""
class ProjectDetail(BaseModel):
project_introduction = models.CharField(max_length=128, help_text="用戶簡介")
project_info = models.ForeignKey(to=ProjectList, on_delete=models.DO_NOTHING, db_constraint=False, related_name='user')
class Meta:
db_table = project_detail'
verbose_name = '項(xiàng)目詳情信息'
verbose_name_plural = verbose_name
一带膜、同時(shí)新增數(shù)據(jù)到兩個(gè)數(shù)據(jù)庫表
from rest_framework import filters, mixins, generics
from rest_framework.viewsets import GenericViewSet
class ProjectListView(mixins.CreateModelMixin, GenericViewSet):
queryset = Account.objects.filter(is_delete=0).all().order_by("-created_tm")
def create(self, request, *args, **kwargs):
# 先定義兩個(gè)dict
project_list_dict = {}
project_detail_dict = {}
# 使用 ._meta.get_fields() 獲取數(shù)據(jù)庫表的全部字段,再使用field.name獲取全部的字段名稱鸳谜,放入各自的dict膝藕,不存在的自動(dòng)忽略掉
for item in request.data.items():
if item in [field.name for field in ProjectList._meta.get_fields()]:
project_list_dict [item[0]] = item[1]
elif item in [field.name for field in ProjectDetail._meta.get_fields()]:
project_detail_dict [item[0]] = item[1]
# 從上面的models可以看到,項(xiàng)目詳情與項(xiàng)目列表兩張表是有關(guān)聯(lián)的咐扭,所以芭挽,新增數(shù)據(jù)的時(shí)候懒棉,先現(xiàn)在列表,新增成功览绿,生成ID后,再新增詳情穗慕,并通過ID與列表數(shù)據(jù)關(guān)聯(lián)起來
# 檢查項(xiàng)目是否存在饿敲,項(xiàng)目名稱唯一約束,Serializer內(nèi)設(shè)置必填
try:
# 存在就直接失敗
ProjectList.objects.get(project_name=request.data.get('project_ame'))
return xxx
except Account.DoesNotExist:
# 如果不存在逛绵,先調(diào)用ProjectList表新增數(shù)據(jù)
try:
project_create = ProjectList.objects.update_or_create(defaults=project_list_dict, project_name=project_list_dict["project_ame"])
# 如果ProjectList表新增數(shù)據(jù)成功怀各,則通過project_info,在ProjectDetail新增對應(yīng)的數(shù)據(jù)
if project_create:
project_id = ProjectList.objects.filter(project_name=project_list_dict["project_ame"]).values('project_id').first()
ProjectDetail.objects.update_or_create(defaults=project_detail_dict, project_info=project_id["project_id"])
return xxx
except Exception as e:
return e
二术浪、同時(shí)返回兩個(gè)表的數(shù)據(jù)(插拔式)
- 插拔式的好處:即插即用瓢对,需要時(shí)才調(diào)用,不需要?jiǎng)t不調(diào)用
- 從ProjectDetail這個(gè)models我們可以看到胰苏,是沒有定義project_id和project_name的硕蛹,但我們是與ProjectList關(guān)聯(lián)的,所以硕并,我們可以通過project_info法焰,返回ProjectList已定義的project_id和project_name
# 我們只需要把ProjectDetail改成以下這樣子就可以了
class ProjectDetail(BaseModel):
project_introduction = models.CharField(max_length=128, help_text="用戶簡介")
project_info = models.ForeignKey(to=ProjectList, on_delete=models.DO_NOTHING, db_constraint=False, related_name='user')
class Meta:
db_table = project_detail'
verbose_name = '項(xiàng)目詳情信息'
verbose_name_plural = verbose_name
@property
def project_id(self):
return self.project_info.project_id
@property
def project_name(self):
return self.project_info.project_name
# 然后在序列化器內(nèi)直接調(diào)用project_id和project_name即可
class ProjectDetailSerializer(serializers.ModelSerializer):
class Meta:
model = ProjectDetail
fields = ('project_id', 'project_name', 'project_introduction',)