lhyt_shop/apps/customer/models.py

# coding=utf-8

import requests
from django.db import transaction
from django.db import models
from django.utils import timezone
from django.conf import settings

from apps.images.models import Images
from apps.account.models import User

from utils.wx.wechat import WeChat
from utils.wx.WXBizDataCrypt import WXBizDataCrypt
from utils.exceptions import CustomError
from apps.foundation.models import Config


class Customer(models.Model):
    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.ForeignKey(Images, verbose_name=u'头像', on_delete=models.PROTECT, null=True)

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

    def setInfo(self, nickName, gender, avatarUrl):
        self.name = nickName
        if gender == 1:
            self.gender = settings.MALE
        elif gender == 2:
            self.gender = settings.FEMALE
        else:
            self.gender = settings.UNKNOW
        if self.face == None and avatarUrl != None:
            self.face = Images.downLoadFace(self, avatarUrl)
        self.save()


class CustomerWechat(models.Model):
    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):
        secret = Config.getConfigValue(Config.KEY_WECHAT_SESSION_KEY)
        res = WeChat.code2Session(appid, secret, code)
        instance = CustomerWechat.objects.filter(openid=res['openid']).first()
        if not instance:
            instance = CustomerWechat.objects.create(
                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):
        customer_wechat = CustomerWechat.objects.filter(openid=openid).first()
        if not customer_wechat:
            raise CustomError(u'未找到相应的微信客户！')

        pc = WXBizDataCrypt(appid, customer_wechat.session_key)
        phon_data = pc.decrypt(phoneEncryptedData, phoneIv)

        tel = phon_data['purePhoneNumber']

        if customer_wechat.customer:
            if customer_wechat.customer.user.username != tel:
                raise CustomError(u'微信绑定的手机号与系统记录的不符！')
            else:
                return customer_wechat.customer

        customer = Customer.objects.filter(tel=tel).first()
        if not customer:
            Customer.objects.create(tel=tel, name=tel)
        customer_wechat.customer = customer
        customer_wechat.save()
        return customer