def get_albums(artist):
artist_page = Page(artist)
discog_section = artist_page.get_section("Discography")
album_page_names = parse_discog_section(discog_section)
albums = {}
for album_name in album_page_names:
album_page_name = album_page_names[album_name]
album_page = Page(album_page_name)
track_section = album_page.get_section("Track listing")
albums[album_name] = parse_tracklist_section(track_section)
return albums
if is_list_item(line):
album_page,album_name = handle_list_item(line)
if album_page:
album_pages.append(album_page)
return album_pages
if __name__=="__main__":
import article
article.DEBUG = True
# name = "Godsmack"
name = "OSI_(band)"
# name = "Depswa"
artist_page = Page(name)
discog_section = artist_page.get_section("Discography")
album_page_names = parse_discog_section(discog_section)
albums = {}
for album_page_name in album_page_names:
album_page = Page(album_page_name)
track_section = album_page.get_section("Track listing")
albums[album_page_name] = parse_track_names(track_section)
for album in albums:
print album.upper()
for track_name in albums[album]:
print track_name
print
