diff options
author | Philip Sargent <philip.sargent@klebos.com> | 2020-06-07 16:16:35 +0100 |
---|---|---|
committer | Philip Sargent <philip.sargent@klebos.com> | 2020-06-07 16:16:35 +0100 |
commit | fcfda644d3a05c365217bff9f12a16bb343a57d5 (patch) | |
tree | 7409e039cb7e5a3a516d6317de95ff286fbe3802 /reset-django.py | |
parent | 75bac01f3a1cf9a5daedf66dc390b668ecd11620 (diff) | |
download | troggle-fcfda644d3a05c365217bff9f12a16bb343a57d5.tar.gz troggle-fcfda644d3a05c365217bff9f12a16bb343a57d5.tar.bz2 troggle-fcfda644d3a05c365217bff9f12a16bb343a57d5.zip |
Fix un-cleared db error (partial)
Diffstat (limited to 'reset-django.py')
-rw-r--r-- | reset-django.py | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/reset-django.py b/reset-django.py new file mode 100644 index 0000000..8b8f9b6 --- /dev/null +++ b/reset-django.py @@ -0,0 +1,74 @@ +import os
+import shutil
+
+from pprint import pprint
+
+# from https://groups.google.com/forum/#!topic/django-users/C8Q7CTpcChc
+# Just put it in the folder where manage.py file is and run it.
+
+folders = []
+base_dir = os.path.dirname(os.path.realpath(__file__))
+
+
+def get_directory_list():
+ global folders
+ global base_dir
+
+ for root, d_names, f_names in os.walk(base_dir):
+ for name in d_names:
+ folders.append(os.path.join(root, name))
+ folders = sorted(folders)
+
+ return folders
+
+
+def delete_pycache():
+ global folders
+
+ for folder in folders:
+ if folder.endswith("__pycache__"):
+ shutil.rmtree(folder)
+
+ print("All __pycache__ files deleted.")
+ return None
+
+
+def delete_migrations():
+ global folders
+
+ for folder in folders:
+ if folder.endswith("migrations"):
+ for item in os.listdir(folder):
+ if not item.endswith("__init__.py"):
+ os.remove(os.path.join(folder, item))
+
+ print("All migration files deleted.")
+ return None
+
+
+def delete_sqlite3():
+ global base_dir
+
+ db_file = os.path.join(base_dir, "troggle.sqlite3")
+
+ if os.path.exists(db_file):
+ os.remove(db_file)
+
+
+def main():
+ global folders
+
+ try:
+ get_directory_list()
+ delete_pycache()
+ delete_migrations()
+ delete_sqlite3()
+ print("All cleanup operations performed successfully.")
+
+ except Exception as e:
+ print("There was some error! Aaargh. \n")
+ raise
+
+
+if __name__ == "__main__":
+ main()
\ No newline at end of file |