bookshop_admin/apps/order/models.py

# coding=utf-8
from django.db import models
from django.conf import settings
from django.utils import timezone
from apps.commodity.models import Commodity
from apps.option.models import Config, Student,Classes
from apps.upload.models import Upload
from utils.exceptions import CustomError
from apps.base import Formater
from apps.account.models import CustomerWechat
from django.contrib.auth import get_user_model
User = get_user_model()

class Pay(models.Model):
    WAIT = 0
    PAY = 1
    UNDO = 2
    STATUS_CHOICES = (
        (WAIT, u'待付款'),
        (PAY, u'已付款'),
        (UNDO, u'已取消'),
    )
    CUSTOMER = 1
    TYPE_CHOICES = (
        (CUSTOMER, u'客户'),
    )

    pay_no = models.CharField(max_length=25, verbose_name='支付单号', unique=True, null=True)
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name='状态', default=WAIT)
    type = models.PositiveSmallIntegerField(choices=TYPE_CHOICES, verbose_name='类别', default=CUSTOMER)
    precreate_amount = models.BigIntegerField(verbose_name=u"预支付金额")
    create_time = models.DateTimeField(verbose_name=u"创建时间", default=timezone.now)
    amount = models.BigIntegerField(verbose_name=u"支付金额", null=True)
    user = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name=u'下单人', related_name='pay_user',
                             on_delete=models.PROTECT)
    pay_time = models.DateTimeField(verbose_name=u"创建时间", null=True)

    class Meta:
        db_table = "pay"
        verbose_name = u"支付信息"
        ordering = ('-id',)
        default_permissions = ()

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

        self.status = Pay.UNDO
        self.save()

    def paySuccess(self, pay_amount):
        if self.status != Pay.WAIT:
            return

        self.status = Pay.PAY
        self.amount = Formater.formatPrice(pay_amount)
        self.pay_time = timezone.now()
        self.save()
        order = Order.objects.filter(pay=self).first()
        if order:
            order.status = Order.FINISH
            order.actual_amount = self.amount
            order.save()

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

    # 小程序支付
    @staticmethod
    def wechatAppPay(user, amount, type=1):
        item = Pay._precreatePay(
            user,
            amount,
            type,
        )
        return item

    @staticmethod
    def _precreatePay(user, amount, type):

        pay_no = '{0}{1}'.format(user.id, timezone.now().strftime('%y%m%d%H%M%S'))
        pay = Pay.objects.create(
            pay_no=pay_no,
            status=Pay.WAIT,
            precreate_amount=amount,
            user=user,
            type=type,
        )
        return pay

class Order(models.Model):
    WAIT_PAY = 1
    FINISH = 2
    CANCEL = 3
    STATUS_CHOICES = (
        (WAIT_PAY, u'待付款'),
        (FINISH, u'已完成'),
        (CANCEL, u'已取消'),
    )

    ONLINE_PAY = 1
    PAYMENT_CHOICES = (
        (ONLINE_PAY, u'在线支付'),
    )

    no = models.CharField(max_length=50, verbose_name=u'订单号', blank=True)
    pay = models.ForeignKey(Pay, verbose_name='支付信息', on_delete=models.PROTECT, null=True, editable=False)

    payment = models.PositiveSmallIntegerField(choices=PAYMENT_CHOICES, verbose_name=u'付款方式', default=ONLINE_PAY)
    status = models.PositiveSmallIntegerField(choices=STATUS_CHOICES, verbose_name=u"订单状态", default=WAIT_PAY)
    notes = models.CharField(max_length=500, verbose_name=u"备注", null=True)
    create_user = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='order_create_user', verbose_name=u"添加人",
                                    on_delete=models.PROTECT, editable=False)
    create_time = models.DateTimeField(verbose_name=u'添加时间', auto_now_add=True, editable=False)
    student = models.ForeignKey(Student, verbose_name=u'学生', related_name='customer_order_student',
                                on_delete=models.PROTECT, null=True)

    commodity = models.ForeignKey(Commodity, verbose_name=u'商品', related_name='customer_order_commodity',
                                on_delete=models.PROTECT, null=True)
    name = models.CharField(max_length=20, verbose_name=u"名称", editable=False, null=True)
    begin_date = models.DateField(verbose_name=u'起始日期', editable=False, null=True)
    end_date = models.DateField(verbose_name=u'终止日期', editable=False, null=True)
    price = models.IntegerField(verbose_name=u'每餐单价', null=True, default=0, editable=False)
    days = models.IntegerField(verbose_name=u'天数', null=True, default=0, editable=False)
    total_amount = models.BigIntegerField(verbose_name=u'总金额', editable=False, default=0)
    balance_deduction = models.BigIntegerField(verbose_name=u'余额抵扣', editable=False, default=0)
    actual_amount = models.BigIntegerField(verbose_name=u'实付金额', editable=False, default=0)

    class Meta:
        db_table = "order"
        verbose_name = u"订单管理"
        ordering = ('-id',)
        default_permissions = ()
        permissions = [
            ('view_order', u'查看'),
            ('confirm_order', u'确认订单'),
            ('logistics_order', u'发货'),
        ]

    def get_no(self):
        now = timezone.now()
        no = '%s%s%s' % ('A', self.create_user.id, now.strftime('%Y%m%d%H%M%S%f'))
        return no

    @staticmethod
    def get_instance_by_id(id):
        try:
            id = int(id)
        except:
            raise CustomError('无效的订单ID')
        instance = Order.objects.filter(pk=id).first()
        if not instance:
            raise CustomError('未找到对应的订单')
        return instance