Add categories
This commit is contained in:
parent
3f73232312
commit
36be624f5c
3 changed files with 49 additions and 2 deletions
|
@ -1,6 +1,7 @@
|
|||
from django.contrib import admin
|
||||
|
||||
from .models import Transaction, Invoice
|
||||
from .models import Transaction, Invoice, Category
|
||||
|
||||
admin.site.register(Transaction)
|
||||
admin.site.register(Invoice)
|
||||
admin.site.register(Category)
|
||||
|
|
28
nummi/main/migrations/0002_category_transaction_category.py
Normal file
28
nummi/main/migrations/0002_category_transaction_category.py
Normal file
|
@ -0,0 +1,28 @@
|
|||
# Generated by Django 4.0.4 on 2022-05-20 11:34
|
||||
|
||||
from django.db import migrations, models
|
||||
import django.db.models.deletion
|
||||
import uuid
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('main', '0001_initial'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.CreateModel(
|
||||
name='Category',
|
||||
fields=[
|
||||
('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
|
||||
('name', models.CharField(max_length=256)),
|
||||
('parent', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='main.category')),
|
||||
],
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name='transaction',
|
||||
name='category',
|
||||
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='main.category'),
|
||||
),
|
||||
]
|
|
@ -3,15 +3,33 @@ from django.db import models
|
|||
from django.forms import ModelForm
|
||||
|
||||
|
||||
class Category(models.Model):
|
||||
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||||
name = models.CharField(max_length=256)
|
||||
parent = models.ForeignKey("self", on_delete=models.SET_NULL, blank=True, null=True)
|
||||
|
||||
def __str__(self):
|
||||
if self.parent:
|
||||
return f"{self.parent}>{self.name}"
|
||||
else:
|
||||
return f"{self.name}"
|
||||
|
||||
|
||||
class Transaction(models.Model):
|
||||
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
||||
name = models.CharField(max_length=256)
|
||||
description = models.TextField()
|
||||
value = models.DecimalField(max_digits=12, decimal_places=2)
|
||||
date = models.DateField()
|
||||
category = models.ForeignKey(
|
||||
Category, on_delete=models.SET_NULL, blank=True, null=True
|
||||
)
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.date} {self.name}: {self.value}€"
|
||||
res = f"{self.date} {self.name}: {self.value}€"
|
||||
if self.category:
|
||||
return f"{res} ({self.category})"
|
||||
return res
|
||||
|
||||
|
||||
class TransactionForm(ModelForm):
|
||||
|
|
Loading…
Reference in a new issue