db: Give Discs a PlaylistState property

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

db/disc.py

@@ -8,6 +8,7 @@ from gi.repository import GObject
 from . import album
 from . import execute
 from . import objects
+from . import state
 
 class Disc(objects.Row):
     def __gt__(self, rhs): return self.number > rhs.number
@@ -29,6 +30,10 @@ class Disc(objects.Row):
     def number(self):
         return self.get_column("number")
 
+    @GObject.Property
+    def playlist_state(self):
+        return state.PlaylistState(self.get_column("plstateid"))
+
     @GObject.Property
     def subtitle(self):
         return self.get_column("subtitle")
@@ -42,17 +47,22 @@ class DiscTable(objects.Table):
         execute("CREATE TABLE IF NOT EXISTS discs "
                     "(discid INTEGER PRIMARY KEY, "
                     " albumid INTEGER, "
+                    " plstateid INTEGER NOT NULL, "
                     " number INTEGER, "
                     " subtitle TEXT, "
                     " FOREIGN KEY(albumid) REFERENCES albums(albumid), "
+                    " FOREIGN KEY(plstateid) REFERENCES playlist_states(plstateid), "
                     " UNIQUE(albumid, number))")
 
     def do_insert(self, album, number, subtitle):
         subtitle = subtitle if subtitle and len(subtitle) > 0 else None
-        return execute("INSERT INTO discs (albumid, number, subtitle) "
-                       "VALUES (?, ?, ?)", (int(album), number, subtitle))
+        plstate = state.Table.insert(random=False, loop=False)
+        return execute("INSERT INTO discs (albumid, plstateid, number, subtitle) "
+                       "VALUES (?, ?, ?, ?)",
+                       [ int(album), int(plstate), number, subtitle ])
 
     def do_delete(self, disc):
+        state.Table.delete(disc.playlist_state)
         return execute("DELETE FROM discs WHERE discid=?", [ int(disc) ])
 
     def do_get(self, rowid):

db/test_disc.py

@@ -11,7 +11,7 @@ class TestDiscTable(unittest.TestCase):
 
     def test_disc_table_init(self):
         self.assertIsInstance(db.disc.Table, db.disc.DiscTable)
-        db.execute("SELECT discid,albumid,number,subtitle FROM discs")
+        db.execute("SELECT discid,albumid,plstateid,number,subtitle FROM discs")
 
     def test_disc_table_insert(self):
         artist = db.artist.Table.insert("Test Artist", "Test Sort")
@@ -25,6 +25,7 @@ class TestDiscTable(unittest.TestCase):
         self.assertEqual(disc.number, 1)
         self.assertEqual(disc.subtitle, "subtitle")
         self.assertEqual(str(disc), "Test Album: subtitle")
+        self.assertIsInstance(disc.playlist_state, db.state.PlaylistState)
 
         disc2 = db.disc.Table.insert(album, 2, None)
         self.assertEqual(disc2.subtitle, None)
@@ -41,9 +42,11 @@ class TestDiscTable(unittest.TestCase):
         artist = db.artist.Table.find("Test Artist", "Test Sort")
         album  = db.album.Table.find(artist, "Test Album")
         disc   = db.disc.Table.find(album, 1, "subtitle")
+        state  = disc.playlist_state
 
         db.disc.Table.delete(disc)
         self.assertIsNone(db.disc.Table.lookup(album, 1))
+        self.assertIsNone(db.state.Table.get(int(state)))
 
     def test_disc_table_get(self):
         artist = db.artist.Table.insert("Test Artist", "Test Sort")