From 321f912083633417b7bc050f8c139fdfd724357e Mon Sep 17 00:00:00 2001
From: Philip Sargent <philip.sargent@gmail.com>
Date: Thu, 12 Dec 2024 17:08:00 +0000
Subject: fixing pre-run script

---
 reset-django.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'reset-django.py')

diff --git a/reset-django.py b/reset-django.py
index d822d07..c363927 100644
--- a/reset-django.py
+++ b/reset-django.py
@@ -1,5 +1,6 @@
 import os
 import shutil
+from pathlib import Path
 
 """Cleans all django-created files and compiled python. Used by the
 pre-run.sh script which cleans and initialises everything before
@@ -47,7 +48,11 @@ def delete_migrations():
         if folder.endswith("migrations"):
             for item in os.listdir(folder):
                 if not item.endswith("__init__.py"):
-                    os.remove(os.path.join(folder, item))
+                    fullitem = Path(folder, item)
+                    if fullitem.is_dir():
+                        print(f"__ directory {item} in {folder} not deleted")
+                    else:
+                        os.remove(os.path.join(folder, item))
 
     print("All migration files deleted.")
     return None
-- 
cgit v1.2.3