emmental/db/album.py

# Copyright 2021 (c) Anna Schumaker.
#
# Table: albums
# +---------+----------+------+------+
# | albumid | artistid | name | sort |
# +---------+----------+------+------+
#
# Index: album_index
# +-----------------------------+
# | (artistid, name) -> albumid |
# +-----------------------------+
from gi.repository import GObject
from . import artist
from . import disc
from . import execute
from . import objects

class Album(objects.Tag):
    def do_get_column(self, column):
        return execute(f"SELECT {column} FROM albums "
                        "WHERE albumid=?", [ self.rowid ])

    @GObject.Property
    def artist(self):
        return artist.Artist(self.get_column("artistid"))

    def discs(self):
        cursor = execute(f"SELECT discid FROM discs "
                          "WHERE albumid=?", [ self.rowid ])
        return [ disc.Disc(row["discid"]) for row in cursor.fetchall() ]


class AlbumTable(objects.Table):
    def __init__(self):
        objects.Table.__init__(self, "albums", Album)

    def do_create(self):
        execute("CREATE TABLE IF NOT EXISTS albums "
                    "(albumid INTEGER PRIMARY KEY, "
                    " artistid INTEGER, "
                    " name TEXT, "
                    " sort TEXT, "
                    " FOREIGN KEY(artistid) REFERENCES artists(artistid), "
                    " UNIQUE(artistid, name))")
        execute("CREATE INDEX IF NOT EXISTS artist_index "
                "ON albums(artistid, name)")

    def do_insert(self, artist, name):
        return execute("INSERT INTO albums (artistid, name, sort) "
                       "VALUES (?, ?, ?)", (int(artist), name, name.casefold()))

    def do_delete(self, album):
        return execute("DELETE FROM albums WHERE albumid=?", [ int(album) ])

    def do_get(self, rowid):
        return execute("SELECT albumid FROM albums "
                       "WHERE albumid=?", [ rowid ])

    def do_lookup(self, artist, name):
        return execute("SELECT albumid FROM albums "
                       "WHERE (artistid=? AND name=?)", [ int(artist), name ])


Table = AlbumTable()