raoul/igitar
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make Song.parse() take the file content instead of a file name

Browse Source
This commit is contained in:
Raoul Snyman 2021-07-30 14:32:53 -07:00
parent 61b1cda0f6
commit 80e74a77d4
1 changed files with 26 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/chordpro/base.py
View File

@ -190,33 +190,33 @@ class Song(object):
def __init__(self, filename=None):
self.filename = filename
if self.filename:
self.parse(self.filename)
with open(filename) as song_file:
self.parse(song_file.read())
def parse(self, filename):
def parse(self, text):
self.metadata = Metadata()
self.verses = []
self.verse_order = []
with open(filename) as song_file:
is_verse = False
current_verse = None
for line_number, line in enumerate(song_file):
if Directive.is_directive(line):
self.metadata.add(Directive(line))
elif Verse.is_start_of_verse(line):
is_verse = True
current_verse = Verse.parse(line)
self.verse_order.append(current_verse)
elif Verse.is_end_of_verse(line):
if not Verse.is_end_of_verse(line, current_verse.type_):
raise MismatchedVerseType(line_number, current_verse.type_)
self.verses.append(current_verse)
is_verse = False
current_verse = None
elif is_verse:
current_verse.add_line(line.strip())
elif Verse.is_chorus_marker(line):
chorus_name = Verse.get_chorus_from_marker(line)
for verse in self.verses[::-1]:
if verse.title == chorus_name or verse.type_ == "chorus":
self.verse_order.append(verse)
break
is_verse = False
current_verse = None
for line_number, line in enumerate(text.splitlines()):
if Directive.is_directive(line):
self.metadata.add(Directive(line))
elif Verse.is_start_of_verse(line):
is_verse = True
current_verse = Verse.parse(line)
self.verse_order.append(current_verse)
elif Verse.is_end_of_verse(line):
if not Verse.is_end_of_verse(line, current_verse.type_):
raise MismatchedVerseType(line_number, current_verse.type_)
self.verses.append(current_verse)
is_verse = False
current_verse = None
elif is_verse:
current_verse.add_line(line.strip())
elif Verse.is_chorus_marker(line):
chorus_name = Verse.get_chorus_from_marker(line)
for verse in self.verses[::-1]:
if verse.title == chorus_name or verse.type_ == "chorus":
self.verse_order.append(verse)
break