Add view to download the current set of submissions

Author: PeterJCLaw

code_submitter/server.py

@@ -1,5 +1,6 @@
 import io
 import zipfile
+import datetime
 
 import databases
 from sqlalchemy.sql import select
@@ -13,7 +14,7 @@
 from starlette.datastructures import UploadFile
 from starlette.middleware.authentication import AuthenticationMiddleware
 
-from . import auth, config
+from . import auth, utils, config
 from .auth import User
 from .tables import Archive, ChoiceHistory
 
@@ -119,9 +120,28 @@ async def upload(request: Request) -> Response:
         status_code=302,
     )
 
+
+@requires(['authenticated', 'blueshirt'])
+async def download_submissions(request: Request) -> Response:
+    buffer = io.BytesIO()
+    with zipfile.ZipFile(buffer, mode='w') as zf:
+        await utils.collect_submissions(database, zf)
+
+    filename = 'submissions-{now}.zip'.format(
+        now=datetime.datetime.now(datetime.timezone.utc),
+    )
+
+    return Response(
+        buffer.getvalue(),
+        headers={'Content-Disposition': f'attachment; filename="{filename}"'},
+        media_type='application/zip',
+    )
+
+
 routes = [
     Route('/', endpoint=homepage, methods=['GET']),
     Route('/upload', endpoint=upload, methods=['POST']),
+    Route('/download-submissions', endpoint=download_submissions, methods=['GET']),
 ]
 
 middleware = [

tests/tests_app.py

@@ -275,3 +275,48 @@ def test_upload_archive_without_robot_py(self) -> None:
             self.database.fetch_all(ChoiceHistory.select()),
         )
         self.assertEqual([], choices, "Should not have created a choice")
+
+    def test_download_submissions_requires_blueshirt(self) -> None:
+        response = self.session.get(self.url_for('download_submissions'))
+        self.assertEqual(403, response.status_code)
+
+    def test_download_submissions_when_none(self) -> None:
+        self.session.auth = ('blueshirt', 'blueshirt')
+
+        response = self.session.get(self.url_for('download_submissions'))
+        self.assertEqual(200, response.status_code)
+
+        with zipfile.ZipFile(io.BytesIO(response.content)) as zf:
+            self.assertCountEqual(
+                ['summary.txt'],
+                zf.namelist(),
+            )
+
+    def test_download_submissions(self) -> None:
+        self.session.auth = ('blueshirt', 'blueshirt')
+
+        self.await_(self.database.execute(
+            Archive.insert().values(
+                id=8888888888,
+                content=b'',
+                username='someone_else',
+                team='ABC',
+                created=datetime.datetime(2020, 8, 8, 12, 0),
+            ),
+        ))
+        self.await_(self.database.execute(
+            ChoiceHistory.insert().values(
+                archive_id=8888888888,
+                username='test_user',
+                created=datetime.datetime(2020, 9, 9, 12, 0),
+            ),
+        ))
+
+        response = self.session.get(self.url_for('download_submissions'))
+        self.assertEqual(200, response.status_code)
+
+        with zipfile.ZipFile(io.BytesIO(response.content)) as zf:
+            self.assertCountEqual(
+                ['summary.txt', 'ABC.zip'],
+                zf.namelist(),
+            )