From c79086f7152db570ed615ca291855019e033fb4d Mon Sep 17 00:00:00 2001
From: Travis Abendshien <lvnvtravis@gmail.com>
Date: Thu, 4 Jul 2024 17:40:19 -0700
Subject: [PATCH] Fix collation data not clearing on library close

---
 tagstudio/src/core/library.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/tagstudio/src/core/library.py b/tagstudio/src/core/library.py
index 107b88cfb..7eedb8b82 100644
--- a/tagstudio/src/core/library.py
+++ b/tagstudio/src/core/library.py
@@ -843,24 +843,22 @@ def save_library_backup_to_disk(self) -> str:
 
     def clear_internal_vars(self):
         """Clears the internal variables of the Library object."""
+
+        # Reset Directory Data =================================================
         self.library_dir = None
-        self.is_legacy_library = False
 
+        # Reset Entries ========================================================
         self.entries.clear()
         self._next_entry_id = 0
-        # self.filtered_entries.clear()
         self._entry_id_to_index_map.clear()
-
-        self._collation_id_to_index_map.clear()
-
         self.missing_matches = {}
         self.dir_file_count = -1
         self.files_not_in_library.clear()
         self.missing_files.clear()
         self.fixed_files.clear()
         self.filename_to_entry_id_map: dict[Path, int] = {}
-        self.ext_list = self.default_ext_exclude_list
 
+        # Reset Tags ===========================================================
         self.tags.clear()
         self._next_tag_id = 1000
         self._tag_strings_to_id_map = {}
@@ -868,6 +866,13 @@ def clear_internal_vars(self):
         self._tag_id_to_index_map = {}
         self._tag_entry_ref_map.clear()
 
+        # Reset Collations =====================================================
+        self.collations.clear()
+        self._collation_id_to_index_map.clear()
+
+        # Reset Extension List =================================================
+        self.ext_list = self.default_ext_exclude_list
+
     def refresh_dir(self) -> Generator:
         """Scans a directory for files, and adds those relative filenames to internal variables."""