61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
#
|
|
# Table: artists
|
|
# +----------+-----------+------+------+
|
|
# | artistid | plstateid | name | sort |
|
|
# +----------+-----------+------+------+
|
|
#
|
|
# Index: artist_index
|
|
# +------------------+
|
|
# | name -> artistid |
|
|
# +------------------+
|
|
from gi.repository import GObject
|
|
from . import album
|
|
from . import playlist
|
|
from . import sql
|
|
|
|
class Artist(playlist.ParentPlaylist):
|
|
def __init__(self, row):
|
|
playlist.ParentPlaylist.__init__(self, row, "avatar-default-symbolic")
|
|
self._name = row["name"]
|
|
|
|
@GObject.Property
|
|
def name(self): return self._name
|
|
|
|
def delete(self): Table.delete(self)
|
|
|
|
def find_album(self, name): return self.find_child(name)
|
|
|
|
def get_child_table(self): return album.Table
|
|
|
|
|
|
class ArtistTable(playlist.Model):
|
|
def __init__(self):
|
|
playlist.Model.__init__(self, "artists", "sort")
|
|
|
|
def do_create(self):
|
|
sql.execute("CREATE TABLE IF NOT EXISTS artists "
|
|
"(artistid INTEGER PRIMARY KEY, "
|
|
" plstateid INTEGER NOT NULL, "
|
|
" name TEXT UNIQUE, "
|
|
" sort TEXT, "
|
|
" FOREIGN KEY(plstateid) REFERENCES playlist_states(plstateid))")
|
|
sql.execute("CREATE INDEX IF NOT EXISTS artist_index ON artists(name)")
|
|
|
|
def do_factory(self, row):
|
|
return Artist(row)
|
|
|
|
def do_insert(self, plstate, name, sort):
|
|
return sql.execute("INSERT INTO artists (plstateid, name, sort) "
|
|
"VALUES (?, ?, ?)",
|
|
[ plstate.rowid, name, sort.casefold() ])
|
|
|
|
def do_lookup(self, name):
|
|
return sql.execute("SELECT * FROM artists WHERE name=?", [ name ])
|
|
|
|
def find(self, name, sort):
|
|
return res if (res := self.lookup(name)) else self.insert(name, sort)
|
|
|
|
|
|
Table = ArtistTable()
|