Merge branch 'master' of github.com:arkavidia-hmif/arkavidia-backend

Author: hashshura

arkav/arkalogica/admin.py

@@ -3,11 +3,20 @@
 from arkav.arkalogica.models import Choice, ChoiceImage
 from arkav.arkalogica.models import Answer, Submission
 from arkav.arkalogica.models import QuestionImage
+from arkav.arkalogica.services import ArkalogicaService
 
 
 @admin.register(Session)
 class SessionAdmin(admin.ModelAdmin):
     list_display = ['title', 'description', 'start_time', 'end_time']
+    actions = ['create_csv']
+
+    def create_csv(self, request, queryset):
+        q = queryset[0]
+        container = ArkalogicaService().get_submissions(q)
+        return ArkalogicaService().create_csv(container, 'Submissions-%s' % q)
+
+    create_csv.short_description = 'Export Submissions to CSV'
 
 
 class ChoiceInline(admin.TabularInline):

arkav/arkalogica/migrations/0002_submission_session.py

@@ -1,4 +1,4 @@
-# Generated by Django 2.2.13 on 2021-01-25 12:05
+# Generated by Django 2.2.13 on 2021-02-01 06:52
 
 from django.db import migrations, models
 import django.db.models.deletion
@@ -14,10 +14,10 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='submission',
             name='session',
-            field=models.OneToOneField(
-                default=1, on_delete=django.db.models.deletion.CASCADE,
-                related_name='submission', to='arkalogica.Session'
-            ),
-            preserve_default=False,
+            field=models.ForeignKey(
+                default=2,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name='submission',
+                to='arkalogica.Session'),
         ),
     ]

arkav/arkalogica/models.py

@@ -28,7 +28,7 @@ class Submission(models.Model):
     start = models.DateTimeField(auto_now_add=True, editable=False)
     end = models.DateTimeField(auto_now=True)
     user = models.OneToOneField(to=User, related_name='submission', on_delete=models.CASCADE, unique=True)
-    session = models.OneToOneField(to=Session, related_name='submission', on_delete=models.CASCADE, unique=True)
+    session = models.ForeignKey(to=Session, related_name='submission', on_delete=models.CASCADE, default=2)
 
     def __str__(self):
         return self.user.full_name + "'s Submission"

arkav/arkalogica/services.py

@@ -7,6 +7,8 @@
 from rest_framework import status
 from django.db import transaction
 from arkav.utils.exceptions import ArkavAPIException
+import csv
+from django.http import HttpResponse
 
 
 class ArkalogicaService:
@@ -34,6 +36,46 @@ def get_submission(self, requested_user):
         submission, state = Submission.objects.get_or_create(user=requested_user)
         return submission
 
+    @transaction.atomic
+    def get_submissions(self, session):
+        content = []
+
+        questions_id = []
+        # Create header (line 1)
+        header = []
+        header.append("user")
+        if session.questions.count() > 0:
+            for q in session.questions.all():
+                questions_id.append(q.id)
+                header.append('%s' % q)
+
+        content.append(header)
+
+        # Create content (line 2-...)
+        all_submissions = session.submission.all()
+        for sub in all_submissions:
+            user_submission = []
+            user_submission.append(sub.user.full_name)
+            for q in questions_id:
+                ans = Answer.objects.filter(submission__user__id=sub.user.id, question=q).first()
+                if ans:
+                    user_submission.append(ans.tag)
+                else:
+                    user_submission.append('-')
+            content.append(user_submission)
+
+        return content
+
+    def create_csv(self, content, fileName):
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename="%s.csv"' % fileName
+
+        writer = csv.writer(response)
+        for c in content:
+            writer.writerow(c)
+
+        return response
+
     @transaction.atomic
     def submit_answer(self, answer_data, requested_user):
         session = Session.objects.first()