curds: Normalize node names before setting

This makes them easier to look up

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
Anna Schumaker 2019-04-16 12:13:30 -04:00
parent 0994e8986e
commit d266218b7c
2 changed files with 5 additions and 2 deletions

View File

@ -4,7 +4,7 @@ class PlaylistNode:
def __init__(self, name="", icon=""):
self.children = [ ]
self.icon = icon
self.name = name
self.name = self.__normalize__(name)
self.next = None
self.parent = None
self.prev = None
@ -20,6 +20,9 @@ class PlaylistNode:
def __lt__(self, rhs):
return self.name < rhs.name
def __normalize__(self, name):
return name.title()
def append_child(self, child):
prev = None
if len(self.children) > 0:

View File

@ -11,7 +11,7 @@ class TestPlaylistNode(unittest.TestCase):
self.assertIsNone(n.next)
self.assertIsNone(n.prev)
n = node.PlaylistNode("Test Node", "test-icon")
n = node.PlaylistNode("test node", "test-icon")
self.assertEqual(n.icon, "test-icon")
self.assertEqual(n.name, "Test Node")