Modify tokenizer to parse notes

smnp/token/tokenizers/note.py

@@ -1,37 +1,44 @@
 import re
-from smnp.token.type import TokenType
+
+from smnp.note.model import Note
 from smnp.token.model import Token
+from smnp.token.type import TokenType
+
 
 def tokenizeNote(input, current, line):
     consumedChars = 0
-    value = ''
+    notePitch = None
+    octave = None
+    duration = None
+    dot = False
     if input[current] == '@':
         consumedChars += 1
-        value += input[current]
         if input[current+consumedChars] in ('C', 'c', 'D', 'd', 'E', 'e', 'F', 'f', 'G', 'g', 'A', 'a', 'H', 'h', 'B', 'b'):
-            value += input[current+consumedChars]
+            notePitch = input[current+consumedChars]
             consumedChars += 1
             
             if current+consumedChars < len(input) and input[current+consumedChars] in ('b', '#'):                        
-                value += input[current+consumedChars]
+                notePitch += input[current+consumedChars]
                 consumedChars += 1
                 
             if current+consumedChars < len(input) and re.match(r'\d', input[current+consumedChars]):            
-                value += input[current+consumedChars]
+                octave = input[current+consumedChars]
                 consumedChars += 1
                 
-            if current+consumedChars < len(input) and input[current+consumedChars] == '.':            
-                duration = input[current+consumedChars]
+            if current+consumedChars < len(input) and input[current+consumedChars] == '.':
+                duration = ''
                 consumedChars += 1
                 while current+consumedChars < len(input) and re.match(r'\d', input[current+consumedChars]):
                     duration += input[current+consumedChars]        
-                    consumedChars += 1  
-                if current+consumedChars < len(input) and input[current+consumedChars] == 'd':
-                    duration += input[current+consumedChars]
                     consumedChars += 1
-                if len(duration) > 1:
-                    value += duration
-                else:
-                    consumedChars -= 1
+                if len(duration) == 0:
+                    return (0, None)
+                dot = (current+consumedChars) < len(input) and input[current+consumedChars] == 'd'
+                consumedChars += 1
+
+            octave = int(octave) if octave is not None else None
+            duration = int(duration) if duration is not None else None
+            value = Note(notePitch, octave, duration, dot)
+
             return (consumedChars, Token(TokenType.NOTE, value, (line, current)))
     return (0, None)