mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-07-20 03:31:22 +02:00
Herculino Trotta
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
from import_export import fields, resources
|
|
from import_export.widgets import ForeignKeyWidget
|
|
|
|
from apps.export_app.widgets.foreign_key import AutoCreateForeignKeyWidget
|
|
from apps.export_app.widgets.many_to_many import AutoCreateManyToManyWidget
|
|
from apps.transactions.models import (
|
|
Transaction,
|
|
TransactionCategory,
|
|
TransactionTag,
|
|
TransactionEntity,
|
|
)
|
|
|
|
|
|
class TransactionResource(resources.ModelResource):
|
|
account = fields.Field(
|
|
attribute="account",
|
|
column_name="account",
|
|
widget=ForeignKeyWidget("accounts.Account", "name"),
|
|
)
|
|
|
|
category = fields.Field(
|
|
attribute="category",
|
|
column_name="category",
|
|
widget=AutoCreateForeignKeyWidget(TransactionCategory, "name"),
|
|
)
|
|
|
|
tags = fields.Field(
|
|
attribute="tags",
|
|
column_name="tags",
|
|
widget=AutoCreateManyToManyWidget(TransactionTag, field="name"),
|
|
)
|
|
|
|
entities = fields.Field(
|
|
attribute="entities",
|
|
column_name="entities",
|
|
widget=AutoCreateManyToManyWidget(TransactionEntity, field="name"),
|
|
)
|
|
|
|
class Meta:
|
|
model = Transaction
|
|
|
|
|
|
class TransactionTagResource(resources.ModelResource):
|
|
class Meta:
|
|
model = TransactionTag
|
|
|
|
|
|
class TransactionEntityResource(resources.ModelResource):
|
|
class Meta:
|
|
model = TransactionEntity
|
|
|
|
|
|
class TransactionCategoyResource(resources.ModelResource):
|
|
class Meta:
|
|
model = TransactionCategory
|