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