db: Give Libraries a PlaylistState property

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>

db/library.py

@@ -1,14 +1,15 @@
 # Copyright 2021 (c) Anna Schumaker.
 #
 # Table: libraries
-# +-----------+------+---------+
-# | libraryid | path | enabled |
-# +-----------+------+---------+
+# +-----------+-----------+---------+------+
+# | libraryid | plstateid | enabled | path |
+# +-----------+-----------+---------+------+
 import pathlib
 from gi.repository import GObject
 from . import commit
 from . import execute
 from . import objects
+from . import state
 from . import track
 
 class Library(objects.Row):
@@ -25,6 +26,10 @@ class Library(objects.Row):
                           "WHERE libraryid=?", [ self.rowid ])
         return [ track.Track(row["trackid"]) for row in cursor.fetchall() ]
 
+    @GObject.Property
+    def playlist_state(self):
+        return state.PlaylistState(self.get_column("plstateid"))
+
     @GObject.Property
     def path(self):
         return pathlib.Path(self.get_column("path"))
@@ -48,16 +53,21 @@ class LibraryTable(objects.Table):
     def do_create(self):
         execute("CREATE TABLE IF NOT EXISTS libraries "
                     "(libraryid INTEGER PRIMARY KEY, "
+                    " plstateid INTEGER NOT NULL, "
+                    " enabled INTEGER DEFAULT 1, "
                     " path TEXT UNIQUE, "
-                    " enabled INTEGER DEFAULT 1)")
+                    " FOREIGN KEY (plstateid) REFERENCES playlist_states(plstateid))")
 
     def do_insert(self, path):
-        return execute("INSERT INTO libraries (path) "
-                       "VALUES (?)", (str(path),))
+        plstate = state.Table.insert(random=False, loop=False)
+        return execute("INSERT INTO libraries (plstateid, path) "
+                       "VALUES (?, ?)",
+                       [ int(plstate), str(path) ])
 
     def do_delete(self, library):
         for t in library.tracks():
             track.Table.delete(t)
+        state.Table.delete(library.playlist_state)
         return execute("DELETE FROM libraries WHERE libraryid=?", [ int(library) ])
 
     def do_get(self, rowid):

db/test_library.py

@@ -12,7 +12,7 @@ class TestLibraryTable(unittest.TestCase):
 
     def test_library_table_init(self):
         self.assertIsInstance(db.library.Table, db.library.LibraryTable)
-        db.execute("SELECT libraryid,path,enabled FROM libraries")
+        db.execute("SELECT libraryid,plstateid,enabled,path FROM libraries")
 
     def test_library_table_insert(self):
         library = db.library.Table.insert(pathlib.Path("/a/b/c"))
@@ -22,17 +22,20 @@ class TestLibraryTable(unittest.TestCase):
         self.assertEqual(library.path, pathlib.Path("/a/b/c"))
         self.assertEqual(str(library), "/a/b/c")
         self.assertTrue(library.enabled)
+        self.assertIsInstance(library.playlist_state, db.state.PlaylistState)
 
         with self.assertRaises(sqlite3.IntegrityError):
             db.library.Table.insert(pathlib.Path("/a/b/c"))
 
     def test_library_table_delete(self):
         library = db.library.Table.find(pathlib.Path("/a/b/c"))
+        state = library.playlist_state
         track = db.make_fake_track(1, 1.234, "Test 1", "/a/b/c/1.ext",
                                    lib=library.path)
 
         db.library.Table.delete(library)
         self.assertIsNone(db.library.Table.lookup(pathlib.Path("/a/b/c")))
+        self.assertIsNone(db.state.Table.get(int(state)))
         self.assertIsNone(db.track.Table.get(int(track)))
 
     def test_library_table_get(self):