feat: export (WIP)

app/apps/export_app/resources/transactions.py

@@ -1,13 +1,17 @@
 from import_export import fields, resources
 from import_export.widgets import ForeignKeyWidget
 
+from apps.accounts.models import Account
 from apps.export_app.widgets.foreign_key import AutoCreateForeignKeyWidget
 from apps.export_app.widgets.many_to_many import AutoCreateManyToManyWidget
+from apps.export_app.widgets.string import EmptyStringToNoneField
 from apps.transactions.models import (
     Transaction,
     TransactionCategory,
     TransactionTag,
     TransactionEntity,
+    RecurringTransaction,
+    InstallmentPlan,
 )
 
 
@@ -15,7 +19,58 @@ class TransactionResource(resources.ModelResource):
     account = fields.Field(
         attribute="account",
         column_name="account",
-        widget=ForeignKeyWidget("accounts.Account", "name"),
+        widget=ForeignKeyWidget(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"),
+    )
+
+    internal_id = EmptyStringToNoneField(
+        column_name="internal_id", attribute="internal_id"
+    )
+
+    class Meta:
+        model = Transaction
+
+    def get_queryset(self):
+        return Transaction.all_objects.all()
+
+
+class TransactionTagResource(resources.ModelResource):
+    class Meta:
+        model = TransactionTag
+
+
+class TransactionEntityResource(resources.ModelResource):
+    class Meta:
+        model = TransactionEntity
+
+
+class TransactionCategoyResource(resources.ModelResource):
+    class Meta:
+        model = TransactionCategory
+
+
+class RecurringTransactionResource(resources.ModelResource):
+    account = fields.Field(
+        attribute="account",
+        column_name="account",
+        widget=ForeignKeyWidget(Account, "name"),
     )
 
     category = fields.Field(
@@ -37,19 +92,33 @@ class TransactionResource(resources.ModelResource):
     )
 
     class Meta:
-        model = Transaction
+        model = RecurringTransaction
 
 
-class TransactionTagResource(resources.ModelResource):
+class InstallmentPlanResource(resources.ModelResource):
+    account = fields.Field(
+        attribute="account",
+        column_name="account",
+        widget=ForeignKeyWidget(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 = TransactionTag
-
-
-class TransactionEntityResource(resources.ModelResource):
-    class Meta:
-        model = TransactionEntity
-
-
-class TransactionCategoyResource(resources.ModelResource):
-    class Meta:
-        model = TransactionCategory
+        model = InstallmentPlan