refactor using Pathlib (AI written)

core/TESTS/test_drawings.py

@@ -0,0 +1,60 @@
+import os
+import pathlib
+import tempfile
+
+from django.test import TestCase
+
+import settings
+from troggle.parsers import drawings
+from troggle.core.models.survex import DrawingFile
+
+
+class DrawingsPathlibTests(TestCase):
+    def test_load_drawings_creates_expected_entries(self):
+        with tempfile.TemporaryDirectory() as td:
+            # create a small tree
+            p = pathlib.Path(td)
+            (p / 'one.pdf').write_text('pdf')
+            (p / 'two.txt').write_text('txt')
+            sub = p / 'dir'
+            sub.mkdir()
+            (sub / 'three.png').write_text('png')
+            sub2 = p / 'dir2'
+            sub2.mkdir()
+            (sub2 / 'abc.th2').write_text('th2')
+            (sub2 / 'abc.th').write_text('th')
+
+            # point the module at our tempdir
+            settings.DRAWINGS_DATA = td
+
+            drawings.load_drawings_files()
+
+            # all files should be present
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='one.pdf').exists())
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='two.txt').exists())
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='dir/three.png').exists())
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='dir2/abc.th2').exists())
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='dir2/abc.th').exists())
+
+    def test_hidden_and_backup_skipped(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = pathlib.Path(td)
+            (p / '.hidden').write_text('hid')
+            (p / 'file~').write_text('bak')
+            settings.DRAWINGS_DATA = td
+
+            drawings.load_drawings_files()
+
+            # Should not import hidden or backup files
+            self.assertFalse(DrawingFile.objects.filter(dwgpath='.hidden').exists())
+            self.assertFalse(DrawingFile.objects.filter(dwgpath='file~').exists())
+
+    def test_no_extension_file(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = pathlib.Path(td)
+            (p / 'noext').write_text('data')
+            settings.DRAWINGS_DATA = td
+
+            drawings.load_drawings_files()
+
+            self.assertTrue(DrawingFile.objects.filter(dwgpath='noext').exists())