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:
parent
0994e8986e
commit
d266218b7c
|
@ -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:
|
||||
|
|
|
@ -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")
|
||||
|
||||
|
|
Loading…
Reference in New Issue