cleaning files for Array columns (#176)

piccolo_api/media/base.py

@@ -2,6 +2,7 @@
 
 import abc
 import asyncio
+import itertools
 import logging
 import pathlib
 import string
@@ -259,7 +260,11 @@ async def get_file_keys_from_db(self) -> t.List[str]:
         Returns the file key for each file we have in the database.
         """
         table = self.column._meta.table
-        return await table.select(self.column).output(as_list=True)
+        response = await table.select(self.column).output(as_list=True)
+        if isinstance(self.column, Array):
+            return [i for i in itertools.chain(*response)]
+        else:
+            return response
 
     async def get_unused_file_keys(self) -> t.List[str]:
         """

tests/media/test_base.py

@@ -138,7 +138,7 @@ def setUp(self):
     def tearDown(self):
         Movie.alter().drop_table().run_sync()
 
-    def test_get_file_keys_from_db(self):
+    def test_varchar(self):
         Movie.insert(
             Movie(poster="image-1.jpg"),
             Movie(poster="image-2.jpg"),
@@ -153,6 +153,23 @@ def test_get_file_keys_from_db(self):
             sorted(response), ["image-1.jpg", "image-2.jpg", "image-3.jpg"]
         )
 
+    def test_array(self):
+        Movie.insert(
+            Movie(screenshots=["image-1.jpg", "image-2.jpg"]),
+            Movie(screenshots=["image-3.jpg", "image-4.jpg"]),
+        ).run_sync()
+
+        storage = LocalMediaStorage(
+            column=Movie.screenshots, media_path="/tmp/"
+        )
+
+        response = asyncio.run(storage.get_file_keys_from_db())
+
+        self.assertListEqual(
+            sorted(response),
+            ["image-1.jpg", "image-2.jpg", "image-3.jpg", "image-4.jpg"],
+        )
+
 
 class TestDeleteUnusedFiles(TestCase):
     def setUp(self):