db: Create a TrackTable and Track object
Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
parent
f519830095
commit
ea555a428c
|
|
@ -1,10 +1,11 @@
|
|||
# Copyright 2021 (c) Anna Schumaker.
|
||||
import lib
|
||||
import pathlib
|
||||
import sqlite3
|
||||
|
||||
File = lib.data.emmental_data / "emmental.sqlite3"
|
||||
|
||||
Connection = sqlite3.connect(File)
|
||||
Connection = sqlite3.connect(File, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
Connection.row_factory = sqlite3.Row
|
||||
|
||||
commit = Connection.commit
|
||||
|
|
@ -18,9 +19,20 @@ from . import genre
|
|||
from . import decade
|
||||
from . import year
|
||||
from . import library
|
||||
from . import track
|
||||
|
||||
def make_fake_track(trackno, length, title, path, lib="/a/b/c", art="Test Artist",
|
||||
alb="Test Album", disk=1, subtitle=None, yeer=2021):
|
||||
lib = library.Table.find(pathlib.Path(lib))
|
||||
art = artist.Table.find(art, art)
|
||||
alb = album.Table.find(art, alb)
|
||||
disk = disc.Table.find(alb, disk, subtitle)
|
||||
yeer = year.Table.find(yeer)
|
||||
return track.Table.insert(lib, art, alb, disk, yeer,trackno,
|
||||
length, title, pathlib.Path(path))
|
||||
|
||||
def reset():
|
||||
mods = [ artist, album, disc, genre, decade, year, library ]
|
||||
mods = [ artist, album, disc, genre, decade, year, library, track ]
|
||||
|
||||
for mod in mods: mod.Table.drop()
|
||||
for mod in mods: mod.Table.do_create()
|
||||
|
|
|
|||
|
|
@ -0,0 +1,72 @@
|
|||
# Copyright 2021 (c) Anna Schumaker.
|
||||
import datetime
|
||||
import db
|
||||
import pathlib
|
||||
import sqlite3
|
||||
import unittest
|
||||
from gi.repository import GObject
|
||||
|
||||
|
||||
class TestTrackTable(unittest.TestCase):
|
||||
def setUp(self):
|
||||
db.reset()
|
||||
|
||||
def test_track_table_init(self):
|
||||
self.assertIsInstance(db.track.Table, db.track.TrackTable)
|
||||
db.execute("SELECT trackid,libraryid,artistid,albumid,discid,decadeid,yearid FROM tracks")
|
||||
db.execute("SELECT number,playcount,lastplayed,length,title,path FROM tracks")
|
||||
|
||||
def test_track_table_insert(self):
|
||||
library = db.library.Table.find(pathlib.Path("/a/b/c"))
|
||||
artist = db.artist.Table.find("Test Artist", "test artist")
|
||||
album = db.album.Table.find(artist, "Test Album")
|
||||
disc = db.disc.Table.find(album, 1, None)
|
||||
year = db.year.Table.find(2021)
|
||||
track = db.track.Table.insert(library, artist, album, disc, year,
|
||||
1, 1.234, "Test Title", pathlib.Path("/a/b/c/d.efg"))
|
||||
|
||||
self.assertIsInstance(track, db.track.Track)
|
||||
self.assertIsInstance(track, db.objects.Row)
|
||||
|
||||
self.assertEqual(track.library, library)
|
||||
self.assertEqual(track.artist, artist)
|
||||
self.assertEqual(track.album, album)
|
||||
self.assertEqual(track.disc, disc)
|
||||
self.assertEqual(track.decade, year.decade)
|
||||
self.assertEqual(track.year, year)
|
||||
|
||||
self.assertEqual(track.number, 1)
|
||||
self.assertEqual(track.playcount, 0)
|
||||
self.assertEqual(track.lastplayed, None)
|
||||
self.assertEqual(track.length, 1.234)
|
||||
self.assertEqual(track.title, "Test Title")
|
||||
self.assertEqual(track.path, "/a/b/c/d.efg")
|
||||
|
||||
with self.assertRaises(sqlite3.IntegrityError):
|
||||
db.track.Table.insert(library, artist, album, disc, year,
|
||||
1, 1.234, "Test Title", pathlib.Path("/a/b/c/d.efg"))
|
||||
|
||||
def test_track_table_delete(self):
|
||||
track = db.make_fake_track(1, 1.234, "Test Title", "/a/b/c/d.efg")
|
||||
db.track.Table.delete(track)
|
||||
self.assertIsNone(db.track.Table.lookup(pathlib.Path("/a/b/c/d.efg")))
|
||||
|
||||
def test_track_table_get(self):
|
||||
track = db.make_fake_track(1, 1.234, "Test Title", "/a/b/c/d.efg")
|
||||
self.assertEqual(db.track.Table.get(1), track)
|
||||
self.assertIsNone(db.track.Table.get(2))
|
||||
|
||||
def test_track_table_lookup(self):
|
||||
track = db.make_fake_track(1, 1.234, "Test Title", "/a/b/c/d.efg")
|
||||
self.assertEqual(db.track.Table.lookup(pathlib.Path("/a/b/c/d.efg")), track)
|
||||
self.assertIsNone(db.library.Table.lookup(pathlib.Path("/a/b/d/h.ijk")))
|
||||
|
||||
def test_track_table_find(self):
|
||||
with self.assertRaises(NotImplementedError):
|
||||
db.track.Table.find(pathlib.Path("/a/b/c/d.efg"))
|
||||
|
||||
def test_track_played(self):
|
||||
track = db.make_fake_track(1, 1.234, "Test Title", "/a/b/c/d.efg")
|
||||
track.played()
|
||||
self.assertEqual(track.playcount, 1)
|
||||
self.assertEqual(track.lastplayed, datetime.date.today())
|
||||
|
|
@ -0,0 +1,135 @@
|
|||
# Copyright 2021 (c) Anna Schumaker.
|
||||
#
|
||||
# Table: tracks
|
||||
# +---------+-----------+------------+---------+--------+--------+
|
||||
# | trackid | libraryid | artistid | albumid | discid | yearid |
|
||||
# +---------+-----------+------------+---------+--------+--------+
|
||||
# | number | playcount | lastplayed | length | title | path |
|
||||
# +---------+-----------+------------+---------+--------+--------|
|
||||
#
|
||||
# Index: track_index
|
||||
# +-----------------+
|
||||
# | path -> trackid |
|
||||
# +-----------------+
|
||||
import datetime
|
||||
import pathlib
|
||||
from gi.repository import GObject
|
||||
from . import artist
|
||||
from . import album
|
||||
from . import commit
|
||||
from . import decade
|
||||
from . import disc
|
||||
from . import execute
|
||||
from . import library
|
||||
from . import objects
|
||||
from . import year
|
||||
|
||||
class Track(objects.Row):
|
||||
def do_get_column(self, column):
|
||||
return execute(f"SELECT {column} FROM tracks "
|
||||
"WHERE trackid=?", [ self.rowid ])
|
||||
|
||||
@GObject.Property
|
||||
def library(self):
|
||||
return library.Library(self.get_column("libraryid"))
|
||||
|
||||
@GObject.Property
|
||||
def artist(self):
|
||||
return artist.Artist(self.get_column("artistid"))
|
||||
|
||||
@GObject.Property
|
||||
def album(self):
|
||||
return album.Album(self.get_column("albumid"))
|
||||
|
||||
@GObject.Property
|
||||
def disc(self):
|
||||
return disc.Disc(self.get_column("discid"))
|
||||
|
||||
@GObject.Property
|
||||
def decade(self):
|
||||
return decade.Decade(self.get_column("decadeid"))
|
||||
|
||||
@GObject.Property
|
||||
def year(self):
|
||||
return year.Year(self.get_column("yearid"))
|
||||
|
||||
@GObject.Property
|
||||
def number(self):
|
||||
return self.get_column("number")
|
||||
|
||||
@GObject.Property
|
||||
def playcount(self):
|
||||
return self.get_column("playcount")
|
||||
|
||||
@GObject.Property
|
||||
def lastplayed(self):
|
||||
return self.get_column("lastplayed")
|
||||
|
||||
@GObject.Property
|
||||
def length(self):
|
||||
return self.get_column("length")
|
||||
|
||||
@GObject.Property
|
||||
def title(self):
|
||||
return self.get_column("title")
|
||||
|
||||
@GObject.Property
|
||||
def path(self):
|
||||
return self.get_column("path")
|
||||
|
||||
def played(self):
|
||||
execute("UPDATE tracks "
|
||||
"SET playcount=playcount+1, lastplayed=? "
|
||||
"WHERE trackid=?", [ datetime.date.today(), self.rowid ])
|
||||
commit()
|
||||
|
||||
|
||||
class TrackTable(objects.Table):
|
||||
def __init__(self):
|
||||
objects.Table.__init__(self, "tracks", Track)
|
||||
|
||||
def do_create(self):
|
||||
execute("CREATE TABLE IF NOT EXISTS tracks "
|
||||
"(trackid INTEGER PRIMARY KEY, "
|
||||
" libraryid INTEGER, "
|
||||
" artistid INTEGER, "
|
||||
" albumid INTEGER, "
|
||||
" discid INTEGER, "
|
||||
" decadeid INTEGER, "
|
||||
" yearid INTEGER, "
|
||||
" number INTEGER, "
|
||||
" playcount INTEGER DEFAULT 0,"
|
||||
" lastplayed DATE DEFAULT NULL, "
|
||||
" length REAL, "
|
||||
" title TEXT, "
|
||||
" path TEXT UNIQUE, "
|
||||
" FOREIGN KEY(libraryid) REFERENCES libraries(libraryid), "
|
||||
" FOREIGN KEY(artistid) REFERENCES artists(artistid), "
|
||||
" FOREIGN KEY(albumid) REFERENCES albums(albumid), "
|
||||
" FOREIGN KEY(discid) REFERENCES discs(discid), "
|
||||
" FOREIGN KEY(yearid) REFERENCES years(yearid))")
|
||||
execute("CREATE INDEX IF NOT EXISTS track_index ON tracks(path)")
|
||||
|
||||
def do_insert(self, library, artist, album, disc, year, number, length, title, path):
|
||||
return execute("INSERT INTO tracks (libraryid, artistid, albumid, discid, "
|
||||
"decadeid, yearid, number, length, title, path) "
|
||||
"VALUES (?,?,?,?,?,?,?,?,?,?)",
|
||||
[ int(library), int(artist), int(album), int(disc),
|
||||
int(year.decade), int(year), number, length, title, str(path) ])
|
||||
|
||||
def do_delete(self, track):
|
||||
return execute("DELETE FROM tracks WHERE trackid=?", [ int(track) ])
|
||||
|
||||
def do_get(self, rowid):
|
||||
return execute("SELECT trackid FROM tracks "
|
||||
"WHERE trackid=?", [ rowid ])
|
||||
|
||||
def do_lookup(self, path):
|
||||
return execute("SELECT trackid FROM tracks "
|
||||
"WHERE path=?", [ str(path) ])
|
||||
|
||||
def find(self, *args):
|
||||
raise NotImplementedError
|
||||
|
||||
|
||||
Table = TrackTable()
|
||||
Loading…
Reference in New Issue