121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
from django.db import models
|
|
from user.models import User
|
|
from django.contrib.contenttypes.fields import GenericForeignKey
|
|
from django.contrib.contenttypes.models import ContentType
|
|
import jsonfield
|
|
|
|
|
|
class Account(models.Model):
|
|
owner = models.ForeignKey(User, on_delete=models.CASCADE,
|
|
related_name="owned_accounts")
|
|
admin_users = models.ManyToManyField(User, related_name="admin_accounts",
|
|
blank=True)
|
|
view_users = models.ManyToManyField(User, related_name="view_accounts",
|
|
blank=True)
|
|
name = models.CharField(max_length=250)
|
|
|
|
@property
|
|
def qid(self):
|
|
return f"A{self.pk}"
|
|
|
|
def __str__(self):
|
|
return f"{self.name}"
|
|
|
|
|
|
class Institution(models.Model):
|
|
name = models.CharField(max_length=255)
|
|
|
|
@property
|
|
def qid(self):
|
|
return f"I{self.pk}"
|
|
|
|
def __str__(self):
|
|
return f"{self.name}"
|
|
|
|
|
|
class Bank(models.Model):
|
|
qrtr_account = models.ForeignKey(Account, on_delete=models.CASCADE)
|
|
connection = models.ForeignKey('connection.Connection',
|
|
on_delete=models.CASCADE)
|
|
institution = models.ForeignKey(Institution, on_delete=models.CASCADE,
|
|
related_name="banks")
|
|
nickname = models.CharField(max_length=250)
|
|
balance = models.DecimalField(decimal_places=3, max_digits=100)
|
|
ac_type = models.CharField(max_length=250, blank=True)
|
|
ac_subtype = models.CharField(max_length=250, blank=True)
|
|
|
|
@property
|
|
def qid(self):
|
|
return f"B{self.pk}"
|
|
|
|
def __str__(self):
|
|
return f"{self.nickname}"
|
|
|
|
|
|
class Slice(models.Model):
|
|
name = models.CharField(max_length=250)
|
|
icon = models.CharField(max_length=250)
|
|
budget = models.DecimalField(decimal_places=3, max_digits=100)
|
|
avail_parents = models.Q(
|
|
app_label='qrtr_account',
|
|
model='bank') | models.Q(
|
|
app_label='qrtr_account',
|
|
model='slice')
|
|
parent_type = models.ForeignKey(
|
|
ContentType,
|
|
limit_choices_to=avail_parents,
|
|
on_delete=models.CASCADE)
|
|
parent_id = models.PositiveIntegerField()
|
|
slice_of = GenericForeignKey('parent_type', 'parent_id')
|
|
|
|
@property
|
|
def qid(self):
|
|
return f"S{self.pk}"
|
|
|
|
def __str__(self):
|
|
return f"{self.name}"
|
|
|
|
|
|
class Schedule(models.Model):
|
|
name = models.CharField(max_length=255)
|
|
# TODO: Hook this up to an events system for Payday scheduling
|
|
|
|
|
|
class Rule(models.Model):
|
|
kinds = [("refill", "Refill"), ("increase", "Increase"), ("goal", "Goal")]
|
|
kind = models.CharField(choices=kinds, max_length=255)
|
|
when_to_run = models.ForeignKey(Schedule, on_delete=models.CASCADE)
|
|
amount_type = models.CharField(
|
|
choices=[
|
|
("quantity",
|
|
"Quantity"),
|
|
("round",
|
|
"Round"),
|
|
("percent",
|
|
"Percent")],
|
|
default="quantity",
|
|
max_length=20)
|
|
amount = models.DecimalField(decimal_places=3, max_digits=100)
|
|
source = models.ForeignKey(
|
|
Slice,
|
|
on_delete=models.CASCADE,
|
|
related_name="rule_source_set")
|
|
destination = models.ForeignKey(
|
|
Slice,
|
|
on_delete=models.CASCADE,
|
|
related_name="rule_destination_set")
|
|
|
|
|
|
class Transaction(models.Model):
|
|
datetime = models.DateTimeField()
|
|
Bank = models.ForeignKey(Bank, on_delete=models.CASCADE,
|
|
related_name='transactions')
|
|
details = jsonfield.JSONField()
|
|
|
|
@property
|
|
def qid(self):
|
|
return f"T{self.pk}"
|
|
|
|
def __str__(self):
|
|
return f"{self.Bank} - {self.datetime}"
|