emmental/lib/tag.py

# Copyright 2021 (c) Anna Schumaker.
import threading

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

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

    def __len__(self):
        with self.lock:
            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
        self.lock    = threading.Lock()

    def __str__(self):
        return self.name

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

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