cosmetics_shop/apps/customer/models.py

# coding=utf-8

from django.db import models, transaction
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.IntegerField(verbose_name=u'累计消费积分', default=0, editable=False)

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

    @staticmethod
    def get_or_create_customer(user, referrer):
        '''

        :param user: 用户
        :param referrer: 推荐人 被推荐人是新用户的时候 要创建上级分销记录
        :return:
        '''
        customer = Customer.objects.filter(user=user).first()
        if customer:
            return customer
        with transaction.atomic():
            customer = Customer.objects.create(user=user, name=user.username, tel=user.username)
            if referrer:
                customer.create_distributor(referrer)
        return customer

    def create_distributor(self, re_customer):
        referrer = Customer.objects.filter(id=re_customer).first()
        if not referrer:
            return ''
        distributor = SuperiorDistributor.objects.create(customer=self, one_level=referrer)
        re_distributor = SuperiorDistributor.objects.filter(customer=referrer).first()
        if re_distributor:
            distributor.two_level = re_distributor.one_level
            distributor.three_level = re_distributor.two_level
            distributor.four_level = re_distributor.three_level
            distributor.five_level = re_distributor.four_level
            distributor.save()

    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

'''
-------------------------------------------------分销商备注---------------------------------------
A——>B
只有B是新用户的时候B才能成为A的一级分销商
A——>B
AB都不是分销商 B也是A的一级分销商

A——>B——>C——>D——>E——>F——>G——>H                     按分销商级别记录
0   1   2   3   4   5
BCDEF分别是A的五级分销商
F消费
F是A的五级分销商  A按五级返利比例获取返利
F是B的四级分销商  B按四级返利比例获取返利......
计算返利时 我要去找F分别是谁的几级分销商 不好计算
A——>B——>C——>D——>E——>F——>G——>H                      按各级分销商获取返利的返利级别记录(计算方便所以选择这种方式)
5   4   3   2   1   0
F消费
F是A的五级分销商  A按五级返利比例获取返利
F是B的四级分销商  B按四级返利比例获取返利......
计算返利时 只需要找F的各级返利是谁 好计算返利
A——>B——>C——>D——>E——>F——>G——>H
5   4   3   2   1   0
如果F消费 要根据系统设置比例分别给他的上级分销商(F/E/D/C/B/A)返利
如果E不是分销商那么E就不获得返利
如果E不是分销商那么D是按二级返利比例获取返利
如果E后来成为分销商后 F再消费E还能获取返利

要查看X的一级分销商都有谁  只需要查一级返利关联X的客户

查分销商Y下级分销商
SuperiorDistributor.objects.filter(Q(one_level=Y) | Q(two_level=Y) | Q(three_level=Y) | Q(four_level=Y) | Q(five_level=Y))的customer就是Y的下级分销
'''
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)
    two_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, referrer):
        '''
        微信登录小程序
        判断是否有推荐人，如果有且推荐人是分销商，就要创建上级分销表，同时查询推荐人是否存在上级分销表 如果存在就要将推荐人的上级分销相应的添加到本人的上级分销表里
        :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.get_or_create_customer(user, referrer)
        wx_customer.customer = customer
        wx_customer.save()
        return customer


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.CharField(verbose_name=u'省', max_length=100, null=True)
    city = models.CharField(verbose_name=u'市', max_length=100, null=True)
    area = models.CharField(verbose_name=u'县/区', max_length=100, null=True)
    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, self.city, self.area, 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()