Display CSV files as simple tables

Displays CSV files as a table, horizontally and verically scrollable.
No pagination, sorting or filtering as yet.

Author: rebkwok

airlock/templates/file_browser/csv.html

@@ -0,0 +1,25 @@
+{% load airlocktags %}
+{% as_csv_data contents as csv_data %}
+<div class="inline-block min-w-full align-middle max-w-full">
+    {% #table %}
+      {% #table_head class="bg-slate-200" id="csvTable" %}
+        {% #table_row %}
+          {% for header in csv_data.headers %}
+            {% #table_header %}{{ header }}{% /table_header %}
+          {% endfor %}
+        {% /table_row %}
+      {% /table_head %}
+
+      {% #table_body %}
+        {% for row in csv_data.rows %}
+            {% #table_row class="even:bg-slate-50" %}
+            {% for cell in row %}
+                {% #table_cell %}
+                    {{ cell }}
+                {% /table_cell %}
+            {% endfor %}
+            {% /table_row %}
+        {% endfor %}
+      {% /table_body %}
+    {% /table %}
+  </div>

airlock/templates/file_browser/index.html

@@ -182,6 +182,8 @@
                 style="width: 100%;"
             >
             </iframe>
+            {% elif path_item.suffix == '.csv' %}
+              {% include "file_browser/csv.html" with contents=path_item.contents %}
             {% else %}
             <pre>{{ path_item.contents }}</pre>
             {% endif %}

airlock/templatetags/__init__.py

@@ -0,0 +1,15 @@
+import csv
+
+from django import template
+
+
+register = template.Library()
+
+
+@register.simple_tag
+def as_csv_data(file_content):
+    reader = csv.reader(file_content.splitlines(keepends=True))
+    return {
+        "headers": next(reader),
+        "rows": list(reader),
+    }

airlock/templatetags/airlocktags.py

@@ -84,6 +84,13 @@ def test_workspace_view_with_html_file(client_with_permission):
     assert "foobar" in response.rendered_content
 
 
+def test_workspace_view_with_csv_file(client_with_permission):
+    factories.write_workspace_file("workspace", "file.csv", "header1,header2\nFoo,Bar")
+    response = client_with_permission.get("/workspaces/view/workspace/file.csv")
+    for content in ["<table", "Foo", "Bar"]:
+        assert content in response.rendered_content
+
+
 def test_workspace_view_with_404(client_with_permission):
     factories.create_workspace("workspace")
     response = client_with_permission.get("/workspaces/view/workspace/no_such_file.txt")

tests/integration/test_views.py

@@ -0,0 +1,9 @@
+from airlock.templatetags.airlocktags import as_csv_data
+
+
+def test_file_content_as_csv_data():
+    content = "Header1,Header2,Header3\n" "One,Two,Three\n" "Four,Five,Six\n"
+    assert as_csv_data(content) == {
+        "headers": ["Header1", "Header2", "Header3"],
+        "rows": [["One", "Two", "Three"], ["Four", "Five", "Six"]],
+    }