emmental/curds/playlist/user.py

26 lines
818 B
Python

# Copyright 2019 (c) Anna Schumaker.
from . import new
from . import playlist
from .. import tree
class UserNode(tree.ETree):
def __init__(self):
sort = [ "artist", "date", "album", "discnumber", "tracknumber" ]
self.starred = playlist.Playlist("Starred", "starred", sort)
self.new = new.NewPlaylist()
tree.ETree.__init__(self, "Playlists", "audio-x-generic")
self.insert_child(self.new)
self.insert_child(self.starred)
def __setstate__(self, state):
self.new = state["new"]
self.starred = state["starred"]
tree.ETree.__setstate__(self, state)
def reset(self):
tree.ETree.reset(self)
self.new.reset()
self.starred.reset()
self.insert_child(self.new)
self.insert_child(self.starred)