1 files changed, 6 insertions, 1 deletions
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