cosmetics_shop/apps/customer/models.py

# coding=utf-8

from django.db import models
from django.utils import timezone
from django.conf import settings
from django.contrib.auth import get_user_model

from utils.wx.wechat import WeChat
from utils.exceptions import CustomError
from utils.wx.WXBizDataCrypt import WXBizDataCrypt

from apps.WechatApplet.models import WechatApplet

User = get_user_model()


class Customer(models.Model):
    user = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name=u'用户', editable=False, related_name='customer_user', on_delete=models.PROTECT)
    is_distributor = models.BooleanField(verbose_name=u"分销商", default=False)

    name = models.CharField(max_length=100, verbose_name=u"姓名")
    tel = models.CharField(max_length=50, verbose_name=u'电话')
    gender = models.PositiveSmallIntegerField(choices=settings.GENDER_CHOICES, verbose_name=u'性别', null=True)
    face = models.CharField(max_length=200, verbose_name=u'头像', null=True)

    balance = models.BigIntegerField(verbose_name=u'余额', default=0, editable=False)
    points = models.IntegerField(verbose_name=u'积分', default=0, editable=False)

    success_count = models.IntegerField(verbose_name=u'成交次数', default=0, editable=False)
    total_amount = models.BigIntegerField(verbose_name=u'累计消费金额', default=0, editable=False)
    total_point = models.BigIntegerField(verbose_name=u'累计消费积分', default=0, editable=False)

    class Meta:
        db_table = "customer"
        verbose_name = u'客户管理'
        ordering = ['-id']
        index_together = (
            'name',
            'tel',
        )
        default_permissions = ()
        permissions = []

    def setInfo(self, appid, openid, encryptedData, iv):
        wx_customer = CustomerWechat.objects.filter(openid=openid, wechat_app__authorizer_appid=appid).first()
        if not wx_customer:
            raise CustomError(u'未找到相应的微信客户！')
        if wx_customer.customer and wx_customer.customer.id != self.id:
            raise CustomError(u'该微信已同步其他客户！')
        if not wx_customer.customer:
            wx_customer.customer = self
            wx_customer.save()
        pc = WXBizDataCrypt(wx_customer.wechat_app.authorizer_appid, wx_customer.session_key)
        result = pc.decrypt(encryptedData, iv)
        self.name = result['nickName']
        self.gender = result['gender']
        self.face = result['avatarUrl']
        self.save()
        return self.face, self.name


class SuperiorDistributor(models.Model):
    customer = models.ForeignKey(Customer, verbose_name=u'客户', editable=False, related_name='superior_distributor_customer', on_delete=models.PROTECT)
    one_level = models.ForeignKey(Customer, verbose_name=u'一级分销', related_name='superior_distributor_one_level', on_delete=models.PROTECT, null=True, blank=True)
    tow_level = models.ForeignKey(Customer, verbose_name=u'二级分销', related_name='superior_distributor_tow_level', on_delete=models.PROTECT, null=True, blank=True)
    three_level = models.ForeignKey(Customer, verbose_name=u'三级分销', related_name='superior_distributor_three_level', on_delete=models.PROTECT, null=True, blank=True)
    four_level = models.ForeignKey(Customer, verbose_name=u'四级分销', related_name='superior_distributor_four_level', on_delete=models.PROTECT, null=True, blank=True)
    five_level = models.ForeignKey(Customer, verbose_name=u'五级分销', related_name='superior_distributor_five_level', on_delete=models.PROTECT, null=True, blank=True)

    class Meta:
        db_table = "superior_distributor"
        verbose_name = u'上级分销'
        ordering = ['-id']
        default_permissions = ()
        permissions = []


class CustomerWechat(models.Model):
    wechat_app = models.ForeignKey(WechatApplet, verbose_name=u'小程序', on_delete=models.PROTECT, editable=False)
    customer = models.ForeignKey(Customer, verbose_name=u'客户', on_delete=models.PROTECT, editable=False, null=True)
    openid = models.CharField(max_length=512, verbose_name=u"openid")
    session_key = models.CharField(max_length=512, verbose_name=u'session_key',null=True)

    class Meta:
        db_table = "customer_wechat"
        verbose_name = u'微信客户'
        unique_together = [
            ('openid')
        ]
        default_permissions = ()

    @staticmethod
    def login(code, appid):
        '''
        微信自动登录小程序
        :param code:
        :param appid:
        :return:
        '''
        wx = WechatApplet.getByAppid(appid)
        res = WeChat.code2Session(appid, wx.secret, code)
        instance = CustomerWechat.objects.filter(openid=res['openid'], wechat_app=wx).first()
        if not instance:
            instance = CustomerWechat.objects.create(wechat_app=wx, openid=res['openid'], session_key=res['session_key'])
        else:
            instance.session_key = res['session_key']
            instance.save()
        return instance

    @staticmethod
    def bindWechat(appid, openid, phoneEncryptedData, phoneIv):
        '''
        微信登录小程序
        判断是否有推荐人，如果有且推荐人是分销商，就要创建上级分销表，同时查询推荐人是否存在上级分销表 如果存在就要将推荐人的上级分销相应的添加到本人的上级分销表里
        :param appid:
        :param openid:
        :param phoneEncryptedData:
        :param phoneIv:
        :return:
        '''
        wx_customer = CustomerWechat.objects.filter(openid=openid, wechat_app__authorizer_appid=appid).first()
        if not wx_customer:
            raise CustomError(u'未找到相应的微信客户！')

        pc = WXBizDataCrypt(appid, wx_customer.session_key)
        phon_data = pc.decrypt(phoneEncryptedData, phoneIv)
        tel = phon_data['purePhoneNumber']
        if wx_customer.customer:
            if wx_customer.customer.user.username != tel:
                raise CustomError(u'微信绑定的手机号与系统记录的不符!')
            else:
                return wx_customer.customer
        user = User.objects.filter(username=tel).first()
        if not user:
            user = User.objects.create_customer(tel, password=tel, **{'is_active': True})
        customer = Customer.objects.filter(user=user).first()
        if not customer:
            customer = Customer.objects.create(user=user, name=tel, tel=tel)
        wx_customer.customer = customer
        wx_customer.save()
        return customer


class Area(models.Model):
    PROVINCE = 0
    CITY = 1
    COUNTY = 2
    LEVEL_CHOICES = (
        (PROVINCE, u'省级'),
        (CITY, u'市级'),
        (COUNTY, u'县区级'),
    )

    name = models.CharField(verbose_name=u'名称', max_length=50, blank=True, null=True)
    level = models.IntegerField(verbose_name=u'层级', choices=LEVEL_CHOICES)
    province_name = models.CharField(u'所属省', max_length=50, blank=True, null=True)
    city_name = models.CharField(u'所属地市', max_length=50, blank=True, null=True)
    province = models.ForeignKey('Area', verbose_name=u'所属省', related_name='province_children', on_delete=models.PROTECT, blank=True, null=True)
    city = models.ForeignKey('Area', verbose_name=u'所属地市', related_name='city_children', on_delete=models.PROTECT, blank=True, null=True)

    def __unicode__(self):
        return '%s' % (self.name)

    class Meta:
        verbose_name = u"地区"
        db_table = "area"


class CustomerAddress(models.Model):
    customer = models.ForeignKey(Customer, verbose_name=u'客户', on_delete=models.PROTECT, editable=False)
    name = models.CharField(verbose_name=u'收货人姓名', max_length=20)
    tel = models.CharField(verbose_name=u'收货人电话', max_length=11)
    province = models.ForeignKey(Area, verbose_name=u'省', related_name='address_province', on_delete=models.PROTECT)
    city = models.ForeignKey(Area, verbose_name=u'地市', related_name='address_city', null=True, on_delete=models.PROTECT)
    county = models.ForeignKey(Area, verbose_name=u'区县', related_name='address_county', null=True, on_delete=models.PROTECT)
    address = models.CharField(verbose_name=u'详细地址', max_length=500)
    default = models.BooleanField(verbose_name=u'默认', default=False)
    delete = models.BooleanField(verbose_name=u'删除', default=False, editable=False)
    create_time = models.DateTimeField(verbose_name=u'创建时间', default=timezone.now, editable=False)

    class Meta:
        db_table = 'customer_address'
        verbose_name = u'收货地址'
        ordering = ['-default', '-id']
        default_permissions = ()

    def get_address(self):
        return '{}{}{}{}'.format(self.province.name, self.city.name, self.county.name, self.address)

    def destory(self, queryset):
        if self.default:
            self.default = False
            default = queryset.first()
            if default:
                default.default = True
                default.save()
        self.delete = True
        self.save()