diff --git a/schmeckels/sort.py b/schmeckels/sort.py index 77cf728..be7f05f 100644 --- a/schmeckels/sort.py +++ b/schmeckels/sort.py @@ -36,17 +36,19 @@ def command(): print("Skipping") continue - tag = tag_lookup.get(select, None) - if not tag: - print(f"Creating new category '{select}'") - tag = create_tag(select, session) + for text in select.split(","): + tag = tag_lookup.get(text, None) + if not tag: + print(f"Creating new category '{text}'") + tag = create_tag(text, session) - tags = session.query(Tag).all() - tag_names = FuzzyWordCompleter([x.name for x in tags]) - tag_lookup = {tag.name: tag for tag in tags} + tags = session.query(Tag).all() + tag_names = FuzzyWordCompleter([x.name for x in tags]) + tag_lookup = {tag.name: tag for tag in tags} + + t.tags.append(tag) + session.add(t) - t.tags.append(tag) - session.add(t) session.commit() print("-" * 20)