fixing pre-run script

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