emmental/lib/tag.py

# Copyright 2021 (c) Anna Schumaker.
from . import publisher
import threading

class Tag:
    def __init__(self, name, sort=None):
        self.name = name
        self.sort = sort.lower() if sort else str(name).lower()
        self.tracks = [ ]
        self.widgets = None
        self.lock = threading.Lock()
        self.TrackAdded = publisher.Publisher()
        self.TrackRemoved = publisher.Publisher()

    def __getstate__(self):
        with self.lock:
            return { "name"   : self.name,
                     "sort"   : self.sort,
                     "tracks" : [ t.trackid for t in self.tracks ] }

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

    def __lt__(self, rhs):
        if not isinstance(rhs, SuperTag):
            return self.sort < rhs.sort
        if self == rhs.parent:
            return True
        return self.sort < rhs.parent.sort

    def __setstate__(self, state):
        self.name    = state["name"]
        self.sort    = state["sort"]
        self.tracks  = state["tracks"]
        self.widgets = None
        self.lock    = threading.Lock()
        self.TrackAdded   = publisher.Publisher()
        self.TrackRemoved = publisher.Publisher()

    def __str__(self):
        return self.name

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

    def get_header(self):
        return self.sort[0].upper() if len(self.sort) > 0 else ""

    def init_track(self, track):
        with self.lock:
            try:
                i = self.tracks.index(track.trackid)
                self.tracks[i] = track
            except Exception as e:
                pass

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


class SuperTag(Tag):
    def __init__(self, parent, name, sort=None):
        Tag.__init__(self, name, sort)
        self.parent = parent

    def __getstate__(self):
        with self.lock:
            return { "name"   : self.name,
                     "sort"   : self.sort,
                     "tracks" : [ t.trackid for t in self.tracks ],
                     "parent" : self.parent }

    def __lt__(self, rhs):
        if not isinstance(rhs, SuperTag):
            return self.parent.sort < rhs.sort
        if self.parent != rhs.parent:
            return self.parent < rhs.parent
        return self.sort < rhs.sort

    def __setstate__(self, state):
        Tag.__setstate__(self, state)
        self.parent = state["parent"]