Django 模型管理器

一、models.py

1. 编写 AddressManager 类来继承 models.Manager

2. 添加 get_default_address(self, user) 方法查询数据库后将对应的数据返回

3. 在 Address 模型类中创建一个 AddressManager 管理器类的对象用于外部调用

from django.db import models


class AddressManager(models.Manager):
    """地址模型管理器"""
    def get_default_address(self, user):
        """获取用户默认收货地址"""
        try:
            address = self.get(user=user, is_default=True)
        except self.model.DoesNotExist:
            address = None

        return address


class Address(BaseModel):
    '''地址模型类'''
    user = models.ForeignKey('User', verbose_name='所属账户')
    receiver = models.CharField(max_length=20, verbose_name='收件人')
    addr = models.CharField(max_length=256, verbose_name='收件地址')
    zip_code = models.CharField(max_length=6, null=True, verbose_name='邮政编码')
    phone = models.CharField(max_length=11, verbose_name='联系电话')
    is_default = models.BooleanField(default=False, verbose_name='是否默认')

    objects = AddressManager()

    class Meta:
        db_table = 'df_address'
        verbose_name = '地址'
        verbose_name_plural = verbose_name

二、views.py

from django.shortcuts import render
from apps.user.models import Address


class UserAddressView(View):
    """用户中心-地址页"""
    def get(self, request):
        """显示地址页"""
        address = Address.objects.get_default_address(request.user)

        return render(request, 'user_center_site.html', {'page': 'address', 'address': address})