Django之同时新增数据到两个数据库表与同时返回两个表的数据(插拔式)

• models：比如有以下三个模型
  

1. from django.db import models
3. """
4. 基类，其他类继承即可获得对应的字段
5. """
6. class BaseModel(models.Model):
7.     updated_tm = models.DateTimeField(auto_now=True)
8.     created_tm = models.DateTimeField(auto_now_add=True)
10.     class Meta:
11.         abstract = True
13. """
14. 项目列表
15. """
16. class ProjectList( BaseModel):
17.     project_id = models.AutoField(help_text="项目ID", primary_key=True)
18.     project_name = models.SlugField(max_length=128, help_text="项目名称", unique=True)
20.     class Meta:
21.         db_table = 'project_list'
22.         verbose_name = '项目列表信息'
23.         verbose_name_plural = verbose_name
25. """
26. 项目详情
27. """
28. class ProjectDetail(BaseModel):
29.     project_introduction = models.CharField(max_length=128, help_text="用户简介")
30.     project_info = models.ForeignKey(to=ProjectList, on_delete=models.DO_NOTHING, db_constraint=False, related_name='user')
32.     class Meta:
33.         db_table = project_detail'
34.         verbose_name = '项目详情信息'
35.         verbose_name_plural = verbose_name

复制代码

一、同时新增数据到两个数据库表

• views
  

1. from rest_framework import filters, mixins, generics
2. from rest_framework.viewsets import GenericViewSet
5. class ProjectListView(mixins.CreateModelMixin, GenericViewSet):
6.     queryset = Account.objects.filter(is_delete=0).all().order_by("-created_tm")
8. def create(self, request, *args, **kwargs):
9.     # 先定义两个dict
10.     project_list_dict = {}
11.     project_detail_dict = {}
13.     # 使用 ._meta.get_fields() 获取数据库表的全部字段，再使用field.name获取全部的字段名称，放入各自的dict，不存在的自动忽略掉
14.     for item in request.data.items():
15.         if item in [field.name for field in ProjectList._meta.get_fields()]:
16.             project_list_dict [item[0]] = item[1]
17.         elif item in [field.name for field in ProjectDetail._meta.get_fields()]:
18.             project_detail_dict [item[0]] = item[1]
20.     # 从上面的models可以看到，项目详情与项目列表两张表是有关联的，所以，新增数据的时候，先现在列表，新增成功，生成ID后，再新增详情，并通过ID与列表数据关联起来
21.     # 检查项目是否存在，项目名称唯一约束，Serializer内设置必填
22.         try:
23.             # 存在就直接失败
24.             ProjectList.objects.get(project_name=request.data.get('project_ame'))
25.             return xxx
26.         except Account.DoesNotExist:
27.             # 如果不存在，先调用ProjectList表新增数据
28.             try:
29.                 project_create = ProjectList.objects.update_or_create(defaults=project_list_dict, project_name=project_list_dict["project_ame"])
30.                 # 如果ProjectList表新增数据成功，则通过project_info，在ProjectDetail新增对应的数据
31.                 if project_create:
32.                     project_id = ProjectList.objects.filter(project_name=project_list_dict["project_ame"]).values('project_id').first()
33.                     ProjectDetail.objects.update_or_create(defaults=project_detail_dict, project_info=project_id["project_id"])
34.                     return xxx
35.             except Exception as e:
36.                 return e

复制代码

二、同时返回两个表的数据(插拔式)

• 插拔式的好处：即插即用，需要时才调用，不需要则不调用
  
• 从ProjectDetail这个models我们可以看到，是没有定义project_id和project_name的，但我们是与ProjectList关联的，所以，我们可以通过project_info，返回ProjectList已定义的project_id和project_name
  

1. # 我们只需要把ProjectDetail改成以下这样子就可以了
2. class ProjectDetail(BaseModel):
3.     project_introduction = models.CharField(max_length=128, help_text="用户简介")
4.     project_info = models.ForeignKey(to=ProjectList, on_delete=models.DO_NOTHING, db_constraint=False, related_name='user')
6.     class Meta:
7.         db_table = project_detail'
8.         verbose_name = '项目详情信息'
9.         verbose_name_plural = verbose_name
11.     @property
12.     def  project_id(self):
13.         return self.project_info.project_id
15.     @property
16.     def  project_name(self):
17.         return self.project_info.project_name
19. # 然后在序列化器内直接调用project_id和project_name即可
20. class ProjectDetailSerializer(serializers.ModelSerializer):
21.     class Meta:
22.         model = ProjectDetail
24.         fields = ('project_id', 'project_name', 'project_introduction',)

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

继续阅读请点击广告