uncloud-mravi/uncloud/uncloud_pay/models.py

from django.db import models
from django.contrib.auth import get_user_model
from django.core.validators import MinValueValidator

import uuid

AMOUNT_MAX_DIGITS=10
AMOUNT_DECIMALS=2

class Bill(models.Model):
    owner = models.ForeignKey(get_user_model(),
            on_delete=models.CASCADE,
            editable=False)

    creation_date = models.DateTimeField()
    starting_date = models.DateTimeField()
    ending_date = models.DateTimeField()
    due_date = models.DateField()

    paid = models.BooleanField(default=False)
    valid = models.BooleanField(default=True)

    @property
    def amount(self):
        # iterate over all related orders
        pass


class Order(models.Model):
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    owner = models.ForeignKey(get_user_model(),
                              on_delete=models.CASCADE,
                              editable=False)

    creation_date = models.DateTimeField()
    starting_date = models.DateTimeField()
    ending_date = models.DateTimeField(blank=True,
                                       null=True)

    bill = models.ManyToManyField(Bill,
                                  editable=False,
                                  blank=True,
                                  null=True)


    recurring_price = models.FloatField(editable=False)
    one_time_price = models.FloatField(editable=False)

    recurring_period = models.CharField(max_length=32,
                              choices = (
                                  ('onetime', 'Onetime'),
                                  ('per_year', 'Per Year'),
                                  ('per_month', 'Per Month'),
                                  ('per_week', 'Per Week'),
                                  ('per_day', 'Per Day'),
                                  ('per_hour', 'Per Hour'),
                                  ('per_minute', 'Per Minute'),
                                  ('per_second', 'Per Second'),
                              ),
                              default='onetime'

    )

    # def amount(self):
    #     amount = recurring_price
    #     if recurring and first_month:
    #         amount += one_time_price

    #     return amount # you get the picture

class PaymentMethod(models.Model):
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    owner = models.ForeignKey(get_user_model(),
            on_delete=models.CASCADE,
            editable=False)
    source = models.CharField(max_length=256,
            choices = (
                ('stripe', 'Stripe'),
                ('unknown', 'Unknown'),
                ),
            default='stripe')
    description = models.TextField()
    primary = models.BooleanField(default=True)

    class Meta:
        unique_together = [['owner', 'primary']]

class Payment(models.Model):
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    owner = models.ForeignKey(get_user_model(),
            on_delete=models.CASCADE,
            editable=False)

    amount = models.DecimalField(
            default=0.0,
            max_digits=AMOUNT_MAX_DIGITS,
            decimal_places=AMOUNT_DECIMALS,
            validators=[MinValueValidator(0)])

    source = models.CharField(max_length=256,
                              choices = (
                                  ('wire', 'Wire Transfer'),
                                  ('stripe', 'Stripe'),
                                  ('voucher', 'Voucher'),
                                  ('referral', 'Referral'),
                                  ('unknown', 'Unknown')
                              ),
                              default='unknown')
    timestamp = models.DateTimeField(editable=False)




class Product(models.Model):
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    owner = models.ForeignKey(get_user_model(),
                              on_delete=models.CASCADE,
                              editable=False)

    description = ""

    status = models.CharField(max_length=256,
                              choices = (
                                  ('pending', 'Pending'),
                                  ('being_created', 'Being created'),
                                  ('active', 'Active'),
                                  ('deleted', 'Deleted')
                              ),
                              default='pending'
    )

    order = models.ForeignKey(Order,
                              on_delete=models.CASCADE,
                              editable=False)

    class Meta:
        abstract = True