From 38ad8709390128af6fb0b657ab05588ecbc3262b Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <yohann.danello@gmail.com>
Date: Wed, 11 Mar 2020 01:03:15 +0100
Subject: [PATCH] Add one missing model to the API
---
apps/note/api/serializers.py | 13 ++++++++++++-
apps/note/api/urls.py | 3 ++-
apps/note/api/views.py | 14 ++++++++++++--
3 files changed, 26 insertions(+), 4 deletions(-)
diff --git a/apps/note/api/serializers.py b/apps/note/api/serializers.py
index 1696bfee..61257ec4 100644
--- a/apps/note/api/serializers.py
+++ b/apps/note/api/serializers.py
@@ -5,7 +5,7 @@ from rest_framework import serializers
from rest_polymorphic.serializers import PolymorphicSerializer
from ..models.notes import Note, NoteClub, NoteSpecial, NoteUser, Alias
-from ..models.transactions import TransactionTemplate, Transaction, MembershipTransaction
+from ..models.transactions import TransactionTemplate, Transaction, MembershipTransaction, TemplateCategory
class NoteSerializer(serializers.ModelSerializer):
@@ -78,6 +78,17 @@ class NotePolymorphicSerializer(PolymorphicSerializer):
}
+class TemplateCategorySerializer(serializers.ModelSerializer):
+ """
+ REST API Serializer for Transaction templates.
+ The djangorestframework plugin will analyse the model `TemplateCategory` and parse all fields in the API.
+ """
+
+ class Meta:
+ model = TemplateCategory
+ fields = '__all__'
+
+
class TransactionTemplateSerializer(serializers.ModelSerializer):
"""
REST API Serializer for Transaction templates.
diff --git a/apps/note/api/urls.py b/apps/note/api/urls.py
index 54218796..5e176ec5 100644
--- a/apps/note/api/urls.py
+++ b/apps/note/api/urls.py
@@ -2,7 +2,7 @@
# SPDX-License-Identifier: GPL-3.0-or-later
from .views import NotePolymorphicViewSet, AliasViewSet, \
- TransactionViewSet, TransactionTemplateViewSet, MembershipTransactionViewSet
+ TemplateCategoryViewSet, TransactionViewSet, TransactionTemplateViewSet, MembershipTransactionViewSet
def register_note_urls(router, path):
@@ -12,6 +12,7 @@ def register_note_urls(router, path):
router.register(path + '/note', NotePolymorphicViewSet)
router.register(path + '/alias', AliasViewSet)
+ router.register(path + '/transaction/category', TemplateCategoryViewSet)
router.register(path + '/transaction/transaction', TransactionViewSet)
router.register(path + '/transaction/template', TransactionTemplateViewSet)
router.register(path + '/transaction/membership', MembershipTransactionViewSet)
diff --git a/apps/note/api/views.py b/apps/note/api/views.py
index abcf915b..4fbb9481 100644
--- a/apps/note/api/views.py
+++ b/apps/note/api/views.py
@@ -6,9 +6,9 @@ from rest_framework import viewsets
from .serializers import NoteSerializer, NotePolymorphicSerializer, NoteClubSerializer, NoteSpecialSerializer, \
NoteUserSerializer, AliasSerializer, \
- TransactionTemplateSerializer, TransactionSerializer, MembershipTransactionSerializer
+ TemplateCategorySerializer, TransactionTemplateSerializer, TransactionSerializer, MembershipTransactionSerializer
from ..models.notes import Note, NoteClub, NoteSpecial, NoteUser, Alias
-from ..models.transactions import TransactionTemplate, Transaction, MembershipTransaction
+from ..models.transactions import TransactionTemplate, Transaction, MembershipTransaction, TemplateCategory
class NoteViewSet(viewsets.ModelViewSet):
@@ -131,6 +131,16 @@ class AliasViewSet(viewsets.ModelViewSet):
return queryset
+class TemplateCategoryViewSet(viewsets.ModelViewSet):
+ """
+ REST API View set.
+ The djangorestframework plugin will get all `TemplateCategory` objects, serialize it to JSON with the given serializer,
+ then render it on /api/note/transaction/category/
+ """
+ queryset = TemplateCategory.objects.all()
+ serializer_class = TemplateCategorySerializer
+
+
class TransactionTemplateViewSet(viewsets.ModelViewSet):
"""
REST API View set.
--
GitLab