mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-02-23 07:54:52 +01:00
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
from django.conf import settings
|
|
from django.core.exceptions import ValidationError
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from apps.common.middleware.thread_local import get_current_user
|
|
from apps.common.models import SharedObject, SharedObjectManager
|
|
from apps.transactions.models import Transaction
|
|
|
|
|
|
class AccountGroup(SharedObject):
|
|
name = models.CharField(max_length=255, verbose_name=_("Name"))
|
|
|
|
objects = SharedObjectManager()
|
|
all_objects = models.Manager() # Unfiltered manager
|
|
|
|
class Meta:
|
|
verbose_name = _("Account Group")
|
|
verbose_name_plural = _("Account Groups")
|
|
db_table = "account_groups"
|
|
unique_together = (("owner", "name"),)
|
|
ordering = ["name", "id"]
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Account(SharedObject):
|
|
name = models.CharField(max_length=255, verbose_name=_("Name"))
|
|
group = models.ForeignKey(
|
|
AccountGroup,
|
|
on_delete=models.SET_NULL,
|
|
verbose_name=_("Account Group"),
|
|
blank=True,
|
|
null=True,
|
|
)
|
|
currency = models.ForeignKey(
|
|
"currencies.Currency",
|
|
verbose_name=_("Currency"),
|
|
on_delete=models.PROTECT,
|
|
related_name="accounts",
|
|
)
|
|
exchange_currency = models.ForeignKey(
|
|
"currencies.Currency",
|
|
verbose_name=_("Exchange Currency"),
|
|
on_delete=models.SET_NULL,
|
|
related_name="exchange_accounts",
|
|
null=True,
|
|
blank=True,
|
|
help_text=_("Default currency for exchange calculations"),
|
|
)
|
|
|
|
is_asset = models.BooleanField(
|
|
default=False,
|
|
verbose_name=_("Asset account"),
|
|
help_text=_(
|
|
"Asset accounts count towards your Net Worth, but not towards your month."
|
|
),
|
|
)
|
|
is_archived = models.BooleanField(
|
|
default=False,
|
|
verbose_name=_("Archived"),
|
|
help_text=_("Archived accounts don't show up nor count towards your net worth"),
|
|
)
|
|
untracked_by = models.ManyToManyField(
|
|
settings.AUTH_USER_MODEL,
|
|
blank=True,
|
|
related_name="untracked_accounts",
|
|
)
|
|
|
|
objects = SharedObjectManager()
|
|
all_objects = models.Manager() # Unfiltered manager
|
|
|
|
class Meta:
|
|
verbose_name = _("Account")
|
|
verbose_name_plural = _("Accounts")
|
|
unique_together = (("owner", "name"),)
|
|
ordering = ["name", "id"]
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def is_untracked_by(self):
|
|
user = get_current_user()
|
|
return self.untracked_by.filter(pk=user.pk).exists()
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
if self.exchange_currency == self.currency:
|
|
raise ValidationError(
|
|
{
|
|
"exchange_currency": _(
|
|
"Exchange currency cannot be the same as the account's main currency."
|
|
)
|
|
}
|
|
)
|