From c49b5c169fafd50275f64f904e763a3047317b10 Mon Sep 17 00:00:00 2001
From: benjamyn <benjamyn@bendev.lovelynet.net>
Date: Tue, 20 Oct 2020 03:57:14 -0400
Subject: [PATCH] Moved django over to rest_framework

---
 migratorapi/api/serializers.py      | 20 ++++++++++++++++++++
 migratorapi/api/urls.py             |  9 ++++++---
 migratorapi/api/views.py            |  8 ++++++++
 migratorapi/migratorapi/settings.py |  1 +
 4 files changed, 35 insertions(+), 3 deletions(-)
 create mode 100644 migratorapi/api/serializers.py

diff --git a/migratorapi/api/serializers.py b/migratorapi/api/serializers.py
new file mode 100644
index 0000000..e3697e0
--- /dev/null
+++ b/migratorapi/api/serializers.py
@@ -0,0 +1,20 @@
+from rest_framework import serializers
+from .models import Migration
+
+
+class MigrationSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Migration
+        fields = ['submit_time',
+                  'domain',
+                  'booked_time',
+                  'original_server',
+                  'new_server',
+                  'username',
+                  'notes',
+                  'brand',
+                  'ticket_id',
+                  'migration_status',
+                  'agent_booked',
+                  'additional_domains',
+                  'migration_type']
diff --git a/migratorapi/api/urls.py b/migratorapi/api/urls.py
index eff5cee..ea09495 100644
--- a/migratorapi/api/urls.py
+++ b/migratorapi/api/urls.py
@@ -1,7 +1,10 @@
-from django.urls import path
+from django.urls import include, path
 from . import views
+from rest_framework import routers
+
+router = routers.DefaultRouter()
+router.register('migrations', views.MigrationViewSet)
 
-app_name = 'api'
 urlpatterns = [
-    path('', views.IndexView.as_view(), name='index')
+    path('', include(router.urls))
 ]
diff --git a/migratorapi/api/views.py b/migratorapi/api/views.py
index c4afcf0..2b9cd3f 100644
--- a/migratorapi/api/views.py
+++ b/migratorapi/api/views.py
@@ -3,10 +3,18 @@ from django.http import HttpResponseRedirect
 from django.views import generic
 from django.urls import reverse
 from django.utils import timezone
+from rest_framework import status, viewsets
+from .serializers import MigrationSerializer
 
 from .models import Migration
 
 
+class MigrationViewSet(viewsets.ModelViewSet):
+    queryset = Migration.objects.all()
+    serializer_class = MigrationSerializer
+    # permission_classes = (AllowAny,)
+
+
 class IndexView(generic.ListView):
     template_name = 'api/index.html'
     context_object_name = 'latest_migrations'
diff --git a/migratorapi/migratorapi/settings.py b/migratorapi/migratorapi/settings.py
index 48be50e..f066776 100644
--- a/migratorapi/migratorapi/settings.py
+++ b/migratorapi/migratorapi/settings.py
@@ -38,6 +38,7 @@ INSTALLED_APPS = [
     'django.contrib.sessions',
     'django.contrib.messages',
     'django.contrib.staticfiles',
+    'rest_framework',
 ]
 
 MIDDLEWARE = [