20 lines
491 B
Python
20 lines
491 B
Python
|
# Copyright 2021 (c) Anna Schumaker.
|
||
|
from . import tag
|
||
|
|
||
|
class TagStore:
|
||
|
def __init__(self):
|
||
|
self.store = dict()
|
||
|
|
||
|
def add(self, name, track):
|
||
|
name = name.strip()
|
||
|
if (t := self.store.get(name)) == None:
|
||
|
t = tag.Tag(name)
|
||
|
self.store[name] = t
|
||
|
t.add_track(track)
|
||
|
return t
|
||
|
|
||
|
def remove(self, t, track):
|
||
|
t.remove_track(track)
|
||
|
if len(t) == 0 and t in self.store.values():
|
||
|
self.store.pop(t.name)
|