emmental/lib/tag.py

# Copyright 2021 (c) Anna Schumaker.

class Tag:
    def __init__(self, name):
        self.name = name
        self.tracks = [ ]
        self.widgets = None

    def __getstate__(self):
        return { "name"   : self.name,
                 "tracks" : self.tracks }

    def __len__(self):
        return len(self.tracks)

    def __lt__(self, rhs):
        return self.name < rhs.name

    def __setstate__(self, state):
        self.name    = state["name"]
        self.tracks  = state["tracks"]
        self.widgets = None

    def add_track(self, track):
        self.tracks.append(track)

    def remove_track(self, track):
        self.tracks.remove(track)