add vlab_instances API endpoints

vreapis/virtual_labs/admin.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
-from virtual_labs.models import VirtualLab
+from virtual_labs.models import VirtualLab, VirtualLabInstance
 
 admin.site.register(VirtualLab)
+admin.site.register(VirtualLabInstance)

vreapis/virtual_labs/models.py

@@ -114,3 +114,8 @@ def __str__(self):
 
     class Meta:
         verbose_name = "KeyCloak Auth"
+
+
+class VirtualLabInstance(models.Model):
+    vlab = models.ForeignKey(VirtualLab, on_delete=models.CASCADE, null=True)
+    username = models.CharField(max_length=100, null=True)

vreapis/virtual_labs/serializers.py

@@ -1,8 +1,10 @@
 from dataclasses import fields
 from pyexpat import model
 from rest_framework import serializers
+from rest_framework.validators import UniqueTogetherValidator
 from django.contrib.auth.models import User
-from virtual_labs.models import VM, SDIAProvision, Topology, VLProfile, VirtualLab
+from virtual_labs.models import (VM, SDIAProvision, Topology, VLProfile,
+                                 VirtualLab, VirtualLabInstance)
 from workflows.models import Workflow
 from workflows.serializers import WorkflowSerializer
 
@@ -97,3 +99,25 @@ class Meta:
             'description',
             'endpoint'
         )
+
+
+class VirtualLabInstanceSerializer(serializers.ModelSerializer):
+
+    vlab = serializers.SlugRelatedField(
+        slug_field='slug',
+        queryset=VirtualLab.objects.all()
+        )
+
+    class Meta:
+        model = VirtualLabInstance
+        fields = (
+            'vlab',
+            'username',
+            )
+
+        validators = [
+            UniqueTogetherValidator(
+                queryset=VirtualLabInstance.objects.all(),
+                fields=['vlab', 'username'],
+                )
+            ]

vreapis/virtual_labs/views.py

@@ -1,5 +1,5 @@
 from sys import stdout
-from rest_framework import viewsets
+from rest_framework import mixins, viewsets
 from rest_framework.decorators import action
 from . import serializers
 from . import models
@@ -19,3 +19,20 @@ class VirtualLabViewSet(GetSerializerMixin, viewsets.ModelViewSet):
     }
 
 
+class VirtualLabInstanceViewSet(
+        GetSerializerMixin,
+        mixins.ListModelMixin,
+        mixins.CreateModelMixin,
+        viewsets.GenericViewSet,
+        ):
+
+    model = models.VirtualLabInstance
+    queryset = model.objects.all()
+    serializer_class = serializers.VirtualLabInstanceSerializer
+
+    def get_queryset(self):
+        vlab_slug = self.request.query_params.get('vlab_slug', None)
+        if vlab_slug:
+            return self.model.objects.filter(vlab__slug=vlab_slug)
+        else:
+            return self.model.objects.all()

vreapis/vreapis/urls.py

@@ -16,7 +16,7 @@
 from django.contrib import admin
 from django.urls import path, include
 from rest_framework import routers
-from virtual_labs.views import VirtualLabViewSet
+from virtual_labs.views import VirtualLabViewSet, VirtualLabInstanceViewSet
 from cells.views import CellsViewSet
 from workflows.views import WorkflowViewSet
 from data_products.views import DataProductsViewSet, GeoDataProductsViewSet
@@ -25,13 +25,11 @@
 from vreapis.settings.base import BASE_PATH
 
 
-
-from vreapis.settings.base import BASE_PATH
-
 admin.site.site_header = 'Virtual Labs Administration'
 
 router = routers.DefaultRouter()
 router.register(r'vlabs', VirtualLabViewSet)
+router.register(r'vlab_instances', VirtualLabInstanceViewSet)
 router.register(r'workflows', WorkflowViewSet)
 router.register(r'cells', CellsViewSet)
 router.register(r'dataprods', DataProductsViewSet)