From 2ab67258e9cec570e0471f93ede5d41ed14f3fac Mon Sep 17 00:00:00 2001
From: Anna Schumaker <Anna@NoWheyCreamery.com>
Date: Fri, 5 Nov 2021 15:50:04 -0400
Subject: [PATCH] scanner: Create an EnableSwitch

For changing a library path's "enabled" property

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
---
 scanner/__init__.py     |  3 +++
 scanner/test_scanner.py |  2 ++
 scanner/test_widgets.py | 21 +++++++++++++++++++++
 scanner/widgets.py      | 14 ++++++++++++++
 4 files changed, 40 insertions(+)

diff --git a/scanner/__init__.py b/scanner/__init__.py
index a4a5acc..cddad61 100644
--- a/scanner/__init__.py
+++ b/scanner/__init__.py
@@ -22,6 +22,9 @@ def UpdateAllButton():
 def RemoveButton(library):
     return widgets.RemoveButton(library, Queue)
 
+def EnableSwitch(library):
+    return widgets.EnableSwitch(library, Queue)
+
 
 def commit():
     Queue.push(task.CommitTask())
diff --git a/scanner/test_scanner.py b/scanner/test_scanner.py
index 65f5280..d859b01 100644
--- a/scanner/test_scanner.py
+++ b/scanner/test_scanner.py
@@ -49,3 +49,5 @@ class TestScanner(unittest.TestCase):
                               scanner.widgets.UpdateAllButton)
         self.assertIsInstance(scanner.RemoveButton(lib),
                               scanner.widgets.RemoveButton)
+        self.assertIsInstance(scanner.EnableSwitch(lib),
+                              scanner.widgets.EnableSwitch)
diff --git a/scanner/test_widgets.py b/scanner/test_widgets.py
index 2dd9ec7..6b6dfcb 100644
--- a/scanner/test_widgets.py
+++ b/scanner/test_widgets.py
@@ -168,3 +168,24 @@ class TestScannerRemoveButton(unittest.TestCase):
         self.assertEqual(len(q.tasks), 1)
         self.assertIsInstance(q.tasks[0], task.RemoveLibrarySchedulerTask)
         q.tasks_finished()
+
+
+class TestEnableSwitch(unittest.TestCase):
+    def test_init(self):
+        lib = db.library.Table.find("/a/b/c")
+        q   = queue.TaskQueue()
+        es  = widgets.EnableSwitch(lib, q)
+
+        self.assertIsInstance(es, Gtk.Switch)
+        self.assertEqual(es.library, lib)
+        self.assertEqual(es.queue, q)
+        self.assertEqual(es.get_halign(), Gtk.Align.CENTER)
+        self.assertEqual(es.get_valign(), Gtk.Align.CENTER)
+        self.assertTrue(es.get_active())
+
+        es.set_active(False)
+        self.assertFalse(es.library.enabled)
+        self.assertFalse(widgets.EnableSwitch(lib, q).get_active())
+
+        es.set_active(True)
+        self.assertTrue(lib.enabled)
diff --git a/scanner/widgets.py b/scanner/widgets.py
index 1896067..b132fd6 100644
--- a/scanner/widgets.py
+++ b/scanner/widgets.py
@@ -124,3 +124,17 @@ class RemoveButton(ScannerButton):
 
     def do_clicked(self):
         self.queue.push(task.RemoveLibrarySchedulerTask(self.library))
+
+
+class EnableSwitch(Gtk.Switch):
+    def __init__(self, library, queue):
+        Gtk.Switch.__init__(self)
+        self.set_active(library.enabled)
+        self.set_halign(Gtk.Align.CENTER)
+        self.set_valign(Gtk.Align.CENTER)
+        self.connect("notify::active", self.switched)
+        self.library = library
+        self.queue = queue
+
+    def switched(self, switch, param):
+        self.library.enabled = switch.get_active()