2021-10-07 20:48:26 -04:00
|
|
|
# Copyright 2021 (c) Anna Schumaker.
|
|
|
|
from gi.repository import GObject
|
|
|
|
from . import sql
|
|
|
|
from . import state
|
2021-10-09 13:40:17 -04:00
|
|
|
from . import table
|
2021-10-07 20:48:26 -04:00
|
|
|
|
|
|
|
class Playlist(GObject.GObject):
|
2021-10-13 17:04:26 -04:00
|
|
|
def __init__(self, row, icon_name):
|
2021-10-07 20:48:26 -04:00
|
|
|
GObject.GObject.__init__(self)
|
2021-10-14 13:34:33 -04:00
|
|
|
self._rowkey = row.keys()[0]
|
|
|
|
self._rowid = row[self._rowkey]
|
2021-10-07 20:48:26 -04:00
|
|
|
self._plstate = state.Table.get(row["plstateid"])
|
2021-10-13 17:04:26 -04:00
|
|
|
self._icon_name = icon_name
|
2021-10-07 20:48:26 -04:00
|
|
|
|
2021-10-16 17:37:23 -04:00
|
|
|
def delete(self): raise NotImplementedError
|
|
|
|
|
2021-10-07 20:48:26 -04:00
|
|
|
def has_children(self): return False
|
|
|
|
|
2021-10-14 17:29:00 -04:00
|
|
|
def get_n_tracks(self):
|
|
|
|
cur = sql.execute(f"SELECT COUNT(*) FROM tracks "
|
|
|
|
f"WHERE {self._rowkey}=?", [ self._rowid ])
|
|
|
|
return cur.fetchone()[0]
|
|
|
|
|
2021-10-17 16:43:09 -04:00
|
|
|
def get_track(self, n):
|
|
|
|
order = ', '.join(self.plist_state.sort)
|
|
|
|
row = sql.execute(f"SELECT * FROM tracks "
|
|
|
|
f"INNER JOIN artists USING(artistid) "
|
|
|
|
f"INNER JOIN albums USING(albumid) "
|
|
|
|
f"INNER JOIN discs USING(discid) "
|
|
|
|
f"INNER JOIN years USING(yearid) "
|
|
|
|
f"WHERE tracks.{self._rowkey}=? "
|
|
|
|
f"ORDER BY {order} LIMIT 1 OFFSET ?",
|
|
|
|
[ self.rowid, n ]).fetchone()
|
|
|
|
return track.Table.factory(row)
|
|
|
|
|
2021-10-07 20:48:26 -04:00
|
|
|
@GObject.Property
|
|
|
|
def name(self): raise NotImplementedError
|
|
|
|
|
2021-10-13 17:04:26 -04:00
|
|
|
@GObject.Property
|
|
|
|
def icon_name(self): return self._icon_name
|
|
|
|
|
2021-10-07 20:48:26 -04:00
|
|
|
@GObject.Property
|
|
|
|
def plist_state(self): return self._plstate
|
|
|
|
|
|
|
|
@GObject.Property
|
|
|
|
def rowid(self): return self._rowid
|
|
|
|
|
2021-10-14 13:34:33 -04:00
|
|
|
@GObject.Property
|
|
|
|
def rowkey(self): return self._rowkey
|
|
|
|
|
2021-10-07 20:48:26 -04:00
|
|
|
|
2021-10-15 10:14:58 -04:00
|
|
|
class MappedPlaylist(Playlist):
|
|
|
|
def __init__(self, row, icon_name, map_table):
|
|
|
|
Playlist.__init__(self, row, icon_name)
|
|
|
|
self._map_table = map_table
|
|
|
|
|
|
|
|
@GObject.Property
|
|
|
|
def map_table(self): return self._map_table
|
|
|
|
|
2021-10-15 16:06:02 -04:00
|
|
|
def add_track(self, track):
|
|
|
|
return sql.execute(f"INSERT OR IGNORE INTO {self.map_table} "
|
|
|
|
f"({self.rowkey}, trackid) VALUES (?, ?)",
|
|
|
|
[ self.rowid, track.rowid ]).rowcount == 1
|
|
|
|
|
2021-10-16 18:33:00 -04:00
|
|
|
def clear(self):
|
|
|
|
sql.execute(f"DELETE FROM {self.map_table} "
|
|
|
|
f"WHERE {self.rowkey}=?", [ self.rowid ])
|
|
|
|
|
2021-10-14 17:29:00 -04:00
|
|
|
def get_n_tracks(self):
|
|
|
|
cur = sql.execute(f"SELECT COUNT(*) FROM {self.map_table} "
|
|
|
|
f"WHERE {self._rowkey}=?", [ self._rowid ])
|
|
|
|
return cur.fetchone()[0]
|
|
|
|
|
2021-10-17 16:43:09 -04:00
|
|
|
def get_track(self, n):
|
|
|
|
order = ', '.join(self.plist_state.sort)
|
|
|
|
row = sql.execute(f"SELECT * FROM tracks "
|
|
|
|
f"INNER JOIN {self.map_table} USING(trackid) "
|
|
|
|
f"INNER JOIN artists USING(artistid) "
|
|
|
|
f"INNER JOIN albums USING(albumid) "
|
|
|
|
f"INNER JOIN discs USING(discid) "
|
|
|
|
f"INNER JOIN years USING(yearid) "
|
|
|
|
f"WHERE {self._rowkey}=? "
|
|
|
|
f"ORDER BY {order} LIMIT 1 OFFSET ?",
|
|
|
|
[ self._rowid, n ]).fetchone()
|
|
|
|
return track.Table.factory(row)
|
|
|
|
|
2021-10-15 16:06:02 -04:00
|
|
|
def remove_track(self, track):
|
|
|
|
return sql.execute(f"DELETE FROM {self.map_table} "
|
|
|
|
f"WHERE {self.rowkey}=? AND trackid=?",
|
|
|
|
[ self.rowid, track.rowid ]).rowcount == 1
|
|
|
|
|
2021-10-15 10:14:58 -04:00
|
|
|
|
2021-10-07 20:48:26 -04:00
|
|
|
class ParentPlaylist(Playlist):
|
|
|
|
def has_children(self): return True
|
|
|
|
def get_child_table(self): raise NotImplementedError
|
|
|
|
|
|
|
|
def get_n_children(self):
|
|
|
|
return self.get_child_table().get_n_children(self)
|
|
|
|
|
|
|
|
def get_child(self, n):
|
|
|
|
return self.get_child_table().get_child(self, n)
|
|
|
|
|
|
|
|
def get_child_index(self, child):
|
|
|
|
return self.get_child_table().get_child_index(self, child)
|
|
|
|
|
|
|
|
def find_child(self, *args):
|
|
|
|
if (res := self.lookup_child(*args)) == None:
|
|
|
|
res = self.get_child_table().insert(self, *args)
|
|
|
|
self.emit("children-changed", self.get_child_index(res), 0, 1)
|
|
|
|
return res
|
|
|
|
|
|
|
|
def lookup_child(self, *args):
|
|
|
|
return self.get_child_table().lookup(self, *args)
|
|
|
|
|
|
|
|
@GObject.Signal(arg_types=(int,int,int))
|
|
|
|
def children_changed(self, pos, rm, add): pass
|
2021-10-09 13:40:17 -04:00
|
|
|
|
|
|
|
|
|
|
|
class Model(table.Model):
|
|
|
|
def insert(self, *args, **kwargs):
|
2021-10-10 13:49:19 -04:00
|
|
|
loop = kwargs.pop("loop", False)
|
2021-10-14 14:18:36 -04:00
|
|
|
sort = kwargs.pop("sort", state.DefaultSort)
|
|
|
|
return super().insert(state.Table.insert(loop=loop,sort=sort), *args)
|
2021-10-09 13:40:17 -04:00
|
|
|
|
|
|
|
def delete(self, plist):
|
|
|
|
state.Table.delete(plist.plist_state)
|
|
|
|
return super().delete(plist)
|
|
|
|
|
|
|
|
|
|
|
|
class ChildModel(table.Child):
|
|
|
|
def insert(self, *args, **kwargs):
|
|
|
|
return super().insert(state.Table.insert(), *args)
|
|
|
|
|
|
|
|
def delete(self, plist):
|
|
|
|
state.Table.delete(plist.plist_state)
|
|
|
|
return super().delete(plist)
|
2021-10-17 16:43:09 -04:00
|
|
|
|
|
|
|
|
|
|
|
from . import track
|