From 100d78a53b89a76ceee1b1d376f1fa26fa452131 Mon Sep 17 00:00:00 2001
From: Kseninia Mikhaylova <k.mikhailova@svs-tech.pro>
Date: Tue, 28 May 2024 16:04:27 +0300
Subject: [PATCH] add element viewset

---
 .vscode/PythonImportHelper-v2-Completion.json | 13 +++++++++++--
 back/api/urls.py                              |  1 +
 back/inventory/views.py                       | 13 ++++++++++---
 3 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/.vscode/PythonImportHelper-v2-Completion.json b/.vscode/PythonImportHelper-v2-Completion.json
index 65977b9..2baab60 100644
--- a/.vscode/PythonImportHelper-v2-Completion.json
+++ b/.vscode/PythonImportHelper-v2-Completion.json
@@ -1821,7 +1821,7 @@
         "kind": 5,
         "importPath": "back.api.urls",
         "description": "back.api.urls",
-        "peekOfCode": "router = routers.DefaultRouter()\nrouter.register(r'partner', views.PartnerViewSet)\nurlpatterns = [\n    path('', include(router.urls)),\n    path('admin/', admin.site.urls),\n]",
+        "peekOfCode": "router = routers.DefaultRouter()\nrouter.register(r'partner', views.PartnerViewSet)\nrouter.register(r'element', views.ElementViewSet)\nurlpatterns = [\n    path('', include(router.urls)),\n    path('admin/', admin.site.urls),\n]",
         "detail": "back.api.urls",
         "documentation": {}
     },
@@ -1902,7 +1902,16 @@
         "kind": 6,
         "importPath": "back.inventory.views",
         "description": "back.inventory.views",
-        "peekOfCode": "class PartnerViewSet(viewsets.ModelViewSet):\n    \"\"\"\n    API endpoint that allows partners to be viewed or edited.\n    \"\"\"\n    queryset = Partner.objects.all()\n    serializer_class = PartnerSerializer",
+        "peekOfCode": "class PartnerViewSet(viewsets.ModelViewSet):\n    \"\"\"\n    API endpoint that allows partners to be viewed or edited.\n    \"\"\"\n    queryset = Partner.objects.all()\n    serializer_class = PartnerSerializer\nclass ElementViewSet(viewsets.ModelViewSet):\n    \"\"\"\n    API endpoint that allows elements to be viewed or edited.\n    \"\"\"",
+        "detail": "back.inventory.views",
+        "documentation": {}
+    },
+    {
+        "label": "ElementViewSet",
+        "kind": 6,
+        "importPath": "back.inventory.views",
+        "description": "back.inventory.views",
+        "peekOfCode": "class ElementViewSet(viewsets.ModelViewSet):\n    \"\"\"\n    API endpoint that allows elements to be viewed or edited.\n    \"\"\"\n    queryset = Element.objects.all()\n    serializer_class = ElementSerializer",
         "detail": "back.inventory.views",
         "documentation": {}
     },
diff --git a/back/api/urls.py b/back/api/urls.py
index 2badacb..3818a53 100644
--- a/back/api/urls.py
+++ b/back/api/urls.py
@@ -22,6 +22,7 @@ from inventory import views
 
 router = routers.DefaultRouter()
 router.register(r'partner', views.PartnerViewSet)
+router.register(r'element', views.ElementViewSet)
 
 urlpatterns = [
     path('', include(router.urls)),
diff --git a/back/inventory/views.py b/back/inventory/views.py
index 3957b4e..db8cdd4 100644
--- a/back/inventory/views.py
+++ b/back/inventory/views.py
@@ -1,9 +1,8 @@
 from rest_framework import viewsets
-from .models import Partner
-from .serializers import PartnerSerializer
+from .models import Element, Partner
+from .serializers import PartnerSerializer, ElementSerializer
 
 
-# Create your views here.
 class PartnerViewSet(viewsets.ModelViewSet):
     """
     API endpoint that allows partners to be viewed or edited.
@@ -11,3 +10,11 @@ class PartnerViewSet(viewsets.ModelViewSet):
 
     queryset = Partner.objects.all()
     serializer_class = PartnerSerializer
+    
+class ElementViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows elements to be viewed or edited.
+    """
+
+    queryset = Element.objects.all()
+    serializer_class = ElementSerializer