uncloud/uncloud/uncloud_pay/models.py

140 lines
4.6 KiB
Python
Raw Normal View History

2020-02-27 10:21:38 +00:00
from django.db import models
from django.contrib.auth import get_user_model
from django.core.validators import MinValueValidator
2020-02-27 10:21:38 +00:00
import uuid
2020-02-27 10:21:38 +00:00
AMOUNT_MAX_DIGITS=10
AMOUNT_DECIMALS=2
2020-02-27 10:21:38 +00:00
class Bill(models.Model):
owner = models.ForeignKey(get_user_model(),
2020-02-27 14:50:46 +00:00
on_delete=models.CASCADE)
2020-02-27 10:21:38 +00:00
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
2020-02-27 14:50:46 +00:00
return 20
2020-02-27 10:21:38 +00:00
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)
2020-02-27 14:50:46 +00:00
def charge(self, amount):
pass
class Meta:
unique_together = [['owner', 'primary']]
2020-02-27 10:21:38 +00:00
class Payment(models.Model):
uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
owner = models.ForeignKey(get_user_model(),
2020-02-27 14:50:46 +00:00
on_delete=models.CASCADE)
2020-02-27 10:21:38 +00:00
amount = models.DecimalField(
default=0.0,
max_digits=AMOUNT_MAX_DIGITS,
decimal_places=AMOUNT_DECIMALS,
2020-02-27 10:21:38 +00:00
validators=[MinValueValidator(0)])
source = models.CharField(max_length=256,
choices = (
('wire', 'Wire Transfer'),
2020-02-27 11:21:52 +00:00
('stripe', 'Stripe'),
2020-02-27 10:21:38 +00:00
('voucher', 'Voucher'),
('referral', 'Referral'),
('unknown', 'Unknown')
),
default='unknown')
timestamp = models.DateTimeField(editable=False)
2020-02-27 10:36:50 +00:00
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