emmental/lib/tagstore.py

# Copyright 2021 (c) Anna Schumaker.
from . import tag

class TagStore:
    def __init__(self):
        self.store = dict()

    def add(self, name, track=None):
        name = name.strip()
        if (t := self.store.get(name)) == None:
            t = tag.Tag(name)
            self.store[name] = t
        if track:
            t.add_track(track)
        return t

    def remove(self, t, track=None):
        if track:
            t.remove_track(track)
        if (track==None or len(t) == 0) and t in self.store.values():
            self.store.pop(t.name)