carwin_activity/apps/activity/models.py

#coding=utf-8

import datetime

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

from apps.exceptions import CustomError
from apps.account.models import Branch
from apps.customer.models import Customer, CustomerWechat
from util.wechatpay import WechatPay, WeChatResponse, SplitAccountFuc


class Activity(models.Model):
    branch = models.ForeignKey(Branch, verbose_name=u"门店", on_delete=models.PROTECT)
    title = models.CharField(max_length=200, verbose_name=u"标题")
    describe = models.TextField(verbose_name=u'活动介绍')
    end_date = models.DateField(verbose_name=u"截止日期")
    amount = models.FloatField(verbose_name=u"费用", default=0)
    rebate = models.FloatField(verbose_name=u"返利", default=0)
    check_status = models.PositiveSmallIntegerField(choices=settings.CHECK_STATUS_CHOICES, verbose_name=u"审核", editable=False, default=settings.DEFAULT)
    branch_name = models.CharField(max_length=200, verbose_name=u"门店名称", null=True, blank=True)
    branch_tel = models.CharField(max_length=50, verbose_name=u"门店电话", null=True, blank=True)
    branch_address = models.CharField(max_length=200, verbose_name=u"门店地址", null=True, blank=True)
    required_signs = models.CharField(verbose_name=u'必填项', max_length=300, null=True, editable=False)
    create_time = models.DateTimeField(verbose_name=u"添加时间", auto_now_add=True, editable=False)
    enabled = models.BooleanField(verbose_name=u"在用", default=True)
    delete = models.BooleanField(verbose_name=u"删除", default=False)

    @staticmethod
    def getById(id):
        try:
            id = int(id)
        except:
            raise CustomError(u'无效的活动Id')
        instance = Activity.objects.filter(pk=id).first()
        if not instance:
            raise CustomError(u'未找到相应的活动')
        return instance

    class Meta:
        db_table = "activity"
        ordering = ['-id']
        verbose_name = u"活动管理"

    def checkStatus(self):
        if not self.enabled:
            raise CustomError(u'该活动已禁用！')
        if self.delete:
            raise CustomError(u'该活动已删除！')
        if self.check_status != settings.PASS:
            raise CustomError(u'该活动尚未生效！')
        now = datetime.datetime.date(timezone.now())
        if self.end_date and now > self.end_date:
            raise CustomError(u'该活动已过期！')

    @staticmethod
    def getById(id):
        try:
            id = int(id)
        except:
            raise CustomError(u'无效的活动ID')
        instance = Activity.objects.filter(pk=id, delete=False, enabled=True).first()
        if not instance:
            raise CustomError(u'未找到相应的活动')
        return instance


class SplitAccount(models.Model):
    WAIT = 1
    PROCESSING = 2
    FINISHED = 3
    STATUS_CHOICES = (
        (WAIT, u'待分账'),
        (PROCESSING, u'处理中'),
        (FINISHED, u'分账完成'),
    )

    branch = models.ForeignKey(Branch, verbose_name=u"门店", on_delete=models.PROTECT, editable=False)
    no = models.CharField(max_length=64, verbose_name=u"单号")
    create_time = models.DateTimeField(verbose_name=u"创建时间", default=timezone.now)
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"分账状态", default=WAIT)
    amount = models.BigIntegerField(verbose_name=u"分账金额", null=True)  # 单位分
    order_id = models.CharField(max_length=100, verbose_name=u"微信分账单号", null=True, editable=False)  # 微信分账单号，微信支付系统返回的唯一标识

    class Meta:
        db_table = "split_account"
        verbose_name = u"分账"
        ordering = ('-id',)
        index_together = (
            'create_time',
            'status',
        )
        unique_together = (
            'no',
        )
        default_permissions = ()


    def unfreezeAccount(self):
        # 解冻剩余资金  如果由于客户或其他原因 无法完成分账  就手动解除资金冻结
        pass

    def handSplitAccont(self, spd, spr, pay):
        '''手动申请分账(在自动申请分账失败后  手动申请分账)'''
        if spd.status == SplitAccountDetail.SUCCESS:
            raise CustomError(u'该分账已成功，禁止重复申请！')
        appid = ''
        mchid = ''
        private_key = ''
        cert_serial_no = ''
        apiv3_key = ''
        spc = SplitAccountFuc(appid, mchid, private_key, cert_serial_no, apiv3_key, cert_dir=None, proxy=None)
        # 状态是"待分账" 的查询分账
        if spd.status == SplitAccountDetail.PENDING:
            self.splictAccountQuery(spc, pay.no, self.no, spd)

            if spd.status == SplitAccountDetail.SUCCESS:
                raise CustomError(u'该分账已成功！')

            if spd.status == SplitAccountDetail.PENDING:
                raise CustomError(u'该分账正在处理！')
        # 添加分账接收人
        self.splictAccountAddReceiver(spc, spr)
        # 分账申请
        receivers = [{'account': spd.account, 'amount': spd.amount, 'description': "支付推荐佣金"}]
        success, data = spc.splitaccount_order(pay.no, self.no, receivers)
        if not success:
            raise CustomError(u'分账申请失败！')
        # 分账请求成功  结果不一定成功
        self.order_id = data['order_id']
        if data['state'] == 'PROCESSING':
            self.status = SplitAccount.PROCESSING
        if data['state'] == 'FINISHED':
            self.status = SplitAccount.FINISHED
        self.save()
        for item in data['receivers']:
            # 现在只对一个推荐人分账
            if item['account'] == spd.account:
                if item['result'] == 'PENDING':
                    spd.status = SplitAccountDetail.PENDING
                elif item['result'] == 'SUCCESS':
                    spd.status = SplitAccountDetail.SUCCESS
                elif item['result'] == 'CLOSED':
                    spd.status = SplitAccountDetail.CLOSED
                    spd.fail_reason = item['fail_reason']
                spd.create_time = item['create_time']
                spd.finish_time = item['finish_time']
                spd.detail_no = item['detail_id']
                spd.save()

    def splictAccountAddReceiver(self, split_account_func, spr):
        '''添加分账接收人'''
        if spr.status != SplitAccountReceiver.ADD:
            add = split_account_func.splitaccount_addreceiver(spr.account)
            if not add:
                raise CustomError(u'添加分账接收方失败！')
            spr.status = SplitAccountReceiver.ADD
            spr.save()

    def splictAccountQuery(self, split_account_func, transaction_id, out_order_no, detail):
        '''查询分账结果'''
        success, data = split_account_func.splitaccount_orderquery(transaction_id, out_order_no)
        if not success:
            raise CustomError(u'[%s]查询分账结果失败！' % out_order_no)
        if data['state'] == 'PROCESSING':
            self.status = SplitAccount.PROCESSING
        if data['state'] == 'FINISHED':
            self.status = SplitAccount.FINISHED
        self.save()
        for item in data['receivers']:
            if detail and detail.account == item['account']:
                if item['result'] == 'SUCCESS':
                    detail.status = SplitAccountDetail.SUCCESS
                elif item['result'] == 'CLOSED':
                    detail.status = SplitAccountDetail.CLOSED
                    detail.fail_reason = item['fail_reason']
                detail.create_time = item['create_time']
                detail.finish_time = item['finish_time']
                detail.detail_no = item['detail_id']
                detail.save()

    def splitAccount(self, spd, spr, pay):
        if spd.status != SplitAccountDetail.DEFAULT:
            return
        appid = ''
        mchid = ''
        private_key = ''
        cert_serial_no = ''
        apiv3_key = ''
        spc = SplitAccountFuc(appid, mchid, private_key, cert_serial_no, apiv3_key, cert_dir=None, proxy=None)

        if spr.status != SplitAccountReceiver.ADD:
            add = spc.splitaccount_addreceiver(spr.account)
            if not add:
                # 添加分账接收方失败 不再继续
                return
            spr.status = SplitAccountReceiver.ADD
            spr.save()

        receivers = [{'account': spd.account, 'amount': spd.amount, 'description': "支付推荐佣金"}]
        success, data = spc.splitaccount_order(pay.no, self.no, receivers)
        if not success:
            # 分账请求失败 不再继续
            return
         # 分账请求成功  结果不一定成功
        self.order_id = data['order_id']
        if data['state'] == 'PROCESSING':
            self.status = SplitAccount.PROCESSING
        if data['state'] == 'FINISHED':
            self.status = SplitAccount.FINISHED
        self.save()
        for item in data['receivers']:
            # 现在只对一个推荐人分账
            if item['account'] == spd.account:
                if item['result'] == 'PENDING':
                    spd.status = SplitAccountDetail.PENDING
                elif item['result'] == 'SUCCESS':
                    spd.status = SplitAccountDetail.SUCCESS
                elif item['result'] == 'CLOSED':
                    spd.status = SplitAccountDetail.CLOSED
                    spd.fail_reason = item['fail_reason']
                spd.create_time = item['create_time']
                spd.finish_time = item['finish_time']
                spd.detail_no = item['detail_id']
                spd.save()

    @staticmethod
    def _addnew(branch, order_id, rebate, customer):
        if rebate <= 0:
            return ''
        customer_wechat = CustomerWechat.objects.filter(customer=customer).first()
        if not customer_wechat:
            return ''

        no = "FZ" + timezone.now().strftime('%y%m%d%H%M%S') + str(order_id)
        instance = SplitAccount.objects.create(
            branch=branch,
            no=no,
            status=SplitAccount.WAIT,
            amount=rebate
        )

        detail = SplitAccountDetail.objects.create(
            main=instance,
            customer=customer,
            account=customer_wechat.openid,
            amount=rebate
        )

        receiver = SplitAccountReceiver.objects.filter(account=detail.account).first()
        if not receiver:
            SplitAccountReceiver.objects.create(
                account=detail.account
            )
        return instance, detail, receiver


class SplitAccountDetail(models.Model):
    DEFAULT = 0
    PENDING = 1
    SUCCESS = 2
    CLOSED = 3
    STATUS_CHOICES = (
        (DEFAULT, u'未处理'),
        (PENDING, u'待分账'),
        (SUCCESS, u'分账成功'),
        (CLOSED, u'已关闭'),
    )

    main = models.ForeignKey(SplitAccount, verbose_name=u'分账单', on_delete=models.PROTECT, editable=False)
    customer = models.ForeignKey(Customer, verbose_name=u'客户', on_delete=models.PROTECT, editable=False)
    account = models.CharField(max_length=100, verbose_name=u'接收账户') # 接受方的openid
    amount = models.BigIntegerField(verbose_name=u'分账金额')  # 单位是分
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"分账结果", default=DEFAULT)
    fail_reason = models.CharField(max_length=1000, verbose_name=u"分账失败原因", null=True, blank=True)
    create_time = models.DateTimeField(verbose_name=u"分账创建时间", null=True, blank=True)
    finish_time = models.DateTimeField(verbose_name=u"分账完成时间", null=True, blank=True)
    detail_no = models.CharField(verbose_name=u'分账明细单号', max_length=100, null=True, editable=False)

    class Meta:
        db_table = "split_account_detail"
        verbose_name = u"分账详细"
        ordering = ('-id',)
        index_together = ('status',)
        default_permissions = ()


class SplitAccountReceiver(models.Model):
    UN_ADD = 1
    ADD = 2
    DELETE = 3
    STATUS_CHOICES = (
        (UN_ADD, u'待添加'),
        (ADD, u'已添加'),
        (DELETE, u'已删除'),
    )

    account = models.CharField(max_length=100, verbose_name=u'接收账户') # 接受方的openid
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"状态", default=UN_ADD)
    create_time = models.DateTimeField(verbose_name=u"创建时间", default=timezone.now)

    class Meta:
        db_table = "split_account_receiver"
        verbose_name = u"分账接收人"
        ordering = ('-id',)
        index_together = ('account', 'status', )
        default_permissions = ()


class Pay(models.Model):

    WAIT = 1
    CONFIRM = 2
    CLOSED = 3
    STATUS_CHOICES = (
        (WAIT, u'待付款'),
        (CONFIRM, u'已付款'),
        (CLOSED, u'已关闭'),
    )

    branch = models.ForeignKey(Branch, verbose_name=u"门店", on_delete=models.PROTECT, editable=False)
    no = models.CharField(max_length=64, verbose_name=u"单号")
    create_time = models.DateTimeField(verbose_name=u"创建时间", default=timezone.now)
    customer = models.ForeignKey(Customer, verbose_name=u'客户', on_delete=models.PROTECT)
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"支付状态")
    profit_sharing = models.CharField(max_length=10, verbose_name=u"是否分账", default="N")
    precreate_amount = models.BigIntegerField(verbose_name=u"预支付金额") # 单位分
    amount = models.BigIntegerField(verbose_name=u"实际支付金额", null=True) # 单位分

    class Meta:
        db_table = "pay"
        verbose_name = u"支付"
        ordering = ('-id',)
        index_together = (
            'create_time',
            'status',
        )
        unique_together = (
            'no',
        )
        default_permissions = ()

    def payClosed(self):
        if self.status != Pay.WAIT:
            return
        self.status = Pay.CLOSED
        self.save()

    def payConfirm(self, amount):
        if self.status != Pay.WAIT:
            return

        self.status = Pay.CONFIRM
        self.amount = amount
        self.save()
        order = Order.objects.filter(pay=self).first()
        if order:
            order.orderPayConfirm(self.amount)

    @staticmethod
    def getByNo(no):
        instance = Pay.objects.filter(no=no).first()
        if not instance:
            raise CustomError(u'未找到相应的支付单！')
        return instance

    @staticmethod
    def wechatPay(app, branch, customer, amount, openid, profit_sharing):
        instance = Pay._addnew(branch, customer, amount, profit_sharing)
        return instance, instance._wechatUnifiedOrder(app, openid)

    @staticmethod
    def _addnew(branch, customer, amount, profit_sharing):
        if amount <= 0:
            raise CustomError(u'无效的付款金额！')

        no = timezone.now().strftime('%y%m%d%H%M%S') + str(customer.id)
        instance = Pay.objects.create(
            branch=branch,
            no=no,
            customer=customer,
            status=Pay.WAIT,
            precreate_amount=amount,
            profit_sharing=profit_sharing
        )
        return instance

    def _wechatUnifiedOrder(self, app, openid):
        wechatpay = WechatPay(app.authorizer_appid, app.agent_num, app.agent_key)
        wechatpay.unifiedOrder(self.no, self.precreate_amount, openid, self.profit_sharing)
        data = wechatpay.getAppString()
        return data

    def split_account(self):
        order = Order.objects.filter(pay=self).first()
        if order and self.status == Pay.CONFIRM and self.profit_sharing == "Y":
            sp, spd, spr = SplitAccount._addnew(self.branch, self.id, self.rebate, self.recommend_member)
            order.splitaccount = sp
            order.save()
            sp.splitAccount(spd, spr, self)


class Order(models.Model):
    DEFAULT = 0
    FINISH = 1
    FAIL = 2
    STATUS_CHOICES = (
        (DEFAULT, u'待付款'),
        (FINISH, u'已完成'),
        (FAIL, u'失败'),
    )
    branch = models.ForeignKey(Branch, verbose_name=u"门店", on_delete=models.PROTECT, editable=False)
    activity = models.ForeignKey(Activity, verbose_name=u"活动", on_delete=models.PROTECT)
    pay = models.ForeignKey(Pay, verbose_name='支付信息', on_delete=models.PROTECT, null=True)
    splitaccount = models.ForeignKey(SplitAccount, verbose_name='分账信息', on_delete=models.PROTECT, null=True)
    member = models.ForeignKey(Customer, verbose_name=u"会员", on_delete=models.PROTECT, editable=False)
    amount = models.BigIntegerField(verbose_name=u"费用", default=0)  # 单位分
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"状态", default=DEFAULT, editable=False)
    model = models.CharField(max_length=200, verbose_name=u"车型", null=True, blank=True)
    tel = models.CharField(max_length=100, verbose_name=u"电话", null=True, blank=True)
    number = models.CharField(max_length=50, verbose_name=u"车牌号", null=True, blank=True)
    name = models.CharField(max_length=50, verbose_name=u"姓名", null=True, blank=True)
    recommend_member = models.ForeignKey(Customer, verbose_name=u"推荐人", related_name='recommend_member_ref_customer', on_delete=models.PROTECT, null=True, blank=True, editable=False)
    rebate = models.BigIntegerField(verbose_name=u"返利", default=0)  # 单位分
    create_time = models.DateTimeField(verbose_name=u"添加时间", auto_now_add=True, editable=False)
    delete = models.BooleanField(verbose_name=u"删除", default=False)

    class Meta:
        db_table = "order"
        ordering = ['-id']
        verbose_name = u"订单管理"

    def payOrder(self, openid, app):
        # 如果订单上的金额不等于活动金额 (比如下单没有支付，后来活动金额修改，现在活动金额不等于订单上活动金额) 还按下单时保存到额金额支付
        if self.status != Order.DEFAULT:
            raise CustomError(u'订单非代付款状态，禁止付款')
        if self.pay:
            pay_no = self.pay.no
            # 先查询订单状态
            checkRexponse = WeChatResponse(app.authorizer_appid, app.agent_num, app.agent_key)
            total_fee = checkRexponse.orderquery(pay_no)
            if int(total_fee) == int(self.amount):
                wechatpay = WechatPay(app.authorizer_appid, app.agent_num, app.agent_key)
                data = wechatpay.unifiedOrder(openid, pay_no, self.amount, self.pay.profit_sharing)
                return data
            self.pay.payClosed()

        profit_sharing = "N"
        if self.rebate > 0 and self.recommend_member:
            profit_sharing = "Y"
        pay, data = Pay.wechatPay(app, self.branch, self.member, self.amount, openid, profit_sharing)
        self.pay = pay
        self.save()
        return data

    def orderPayConfirm(self, amount):
        # 在这进行分账  先去查分账接收人的状态  如果没有就创建一条记录 然后添加分账接收人 然后分账
        # 如果有且状态是已添加  就直接分账
        # 如果有且状态是待添加或已删除 就先添加分账接收人 然后分账
        self.status = Order.FINISH
        self.amount = amount
        self.save()
        coupons = ActivityCoupon.objects.filter(activity=self.activity, coupon__enabled=True)
        now = datetime.datetime.date(timezone.now())
        for item in coupons:
            if item.off_type == Coupon.FIXED_DATE:
                end_date = item.end_date
            else:
                end_date = now + datetime.timedelta(days=item.end_days)
            MemberCoupon.objects.create(
                activity=self.activity,
                member=self.member,
                coupon=item,
                receive_date=now,
                end_date=end_date
            )


class Coupon(models.Model):
    FIXED_DATE = 0
    RECEIVE_TIMING = 1
    OFF_TYPE_CHOICES = (
        (FIXED_DATE, u'固定日期'),
        (RECEIVE_TIMING, u'领取计时'),
    )
    branch = models.ForeignKey(Branch, verbose_name=u"门店", on_delete=models.PROTECT)
    name = models.CharField(max_length=200, verbose_name=u"名称")
    off_type = models.PositiveSmallIntegerField(choices=OFF_TYPE_CHOICES, verbose_name=u"核销方式", default=FIXED_DATE)
    end_date = models.DateField(verbose_name=u"截止日期", null=True, blank=True)
    end_days = models.IntegerField(verbose_name=u"截止天数", null=True, blank=True)
    create_time = models.DateTimeField(verbose_name=u"添加时间", auto_now_add=True, editable=False)
    enabled = models.BooleanField(verbose_name=u"在用", default=True)

    class Meta:
        db_table = "coupon"
        ordering = ['-id']
        verbose_name = u"优惠券"

    @staticmethod
    def getById(id):
        try:
            id = int(id)
        except:
            raise CustomError(u'无效的优惠券')
        instance = Coupon.objects.filter(pk=id).first()
        if not instance:
            raise CustomError(u'未找到相应的优惠券')
        return instance


class MemberCoupon(models.Model):
    activity = models.ForeignKey(Activity, verbose_name=u"活动", on_delete=models.PROTECT)
    member = models.ForeignKey(Customer, verbose_name=u"会员", on_delete=models.PROTECT)
    coupon = models.ForeignKey(Coupon, verbose_name=u"优惠券", on_delete=models.PROTECT)
    receive_date = models.DateField(verbose_name=u"领取日期", null=True, blank=True)
    end_date = models.DateField(verbose_name=u"有效期至", null=True, blank=True)
    write_off = models.BooleanField(verbose_name=u"核销", default=False)
    write_off_time = models.DateTimeField(verbose_name=u"核销时间", editable=False, null=True, blank=True)

    class Meta:
        db_table = "member_coupon"
        ordering = ['-id']
        verbose_name = u"会员优惠券"

    def writeOff(self):
        now = datetime.datetime.date(timezone.now())
        if self.write_off:
            raise CustomError(u'该优惠券已核销，禁止重复使用！')
        if self.end_date and now > self.end_date:
            raise CustomError(u'该优惠券已过有效期，禁止使用！')

        self.write_off = True
        self.write_off_time = timezone.now()
        self.save()

    def cancelWriteOff(self):
        if not self.write_off:
            raise CustomError(u'该优惠券尚未核销，禁止撤销！')

        self.write_off = False
        self.write_off_time = None
        self.save()

    @staticmethod
    def getById(id):
        try:
            id = int(id)
        except:
            raise CustomError(u'无效的会员优惠券ID')

        instance = MemberCoupon.objects.filter(pk=id).first()
        if not instance:
            raise CustomError(u'未找到相应的会员优惠券')
        return instance

class ActivityCoupon(models.Model):
    activity = models.ForeignKey(Activity, verbose_name=u"活动", on_delete=models.PROTECT)
    coupon = models.ForeignKey(Coupon, verbose_name=u"优惠券", on_delete=models.PROTECT)

    class Meta:
        db_table = "activity_coupon"
        ordering = ['-id']
        verbose_name = u"活动优惠券"