(fix): use .get() to avoid KeyError

TagStudioDev · Aug 9, 2024 · 6ca902e · 6ca902e
1 parent ce87b11
commit 6ca902e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tagstudio/src/qt/modals/folders_to_tags.py b/tagstudio/src/qt/modals/folders_to_tags.py
@@ -61,7 +61,7 @@ def add_folders_to_tree(items: list[str]) -> Tag:
                 library.add_tag_to_library(new_tag)
                 branch["dirs"][folder] = dict(dirs={}, tag=new_tag)
             branch = branch["dirs"][folder]
-        return branch["tag"]
+        return branch.get("tag")
 
     for tag in library.tags:
         reversed_tag = reverse_tag(library, tag, None)