def _compare_external_links(page_source, movie, external_url_base, site_name): movie_details_page = BeautifulSoup(page_source, 'html.parser') external_links = movie_details_page.find(id='info-wrapper').find('ul', class_='external').find_all('a') for link in external_links: if external_url_base in link['href']: if site_name == 'imdb': return IMDB.normalize_imdb_id(movie['imdb']['id']) == \ IMDB.normalize_imdb_id(link['href'].split('/')[-1]) return movie[site_name]['id'] == link['href'].split('/')[-1] return False
class IMDBSiteTest(TestCase): def setUp(self): self.site = IMDB(None) def test_login(self): self.assertIn('imdb.com', self.site.browser.current_url) self.assertNotIn('Sign in', self.site.browser.page_source) def tearDown(self): self.site.kill_browser()
def _compare_external_links(page_source, movie, external_url_base, site_name): movie_details_page = BeautifulSoup(page_source, 'html.parser') if movie_details_page.find(class_='ratingstable'): external_links = movie_details_page.find( class_='ratingstable').find_all('a') for link in external_links: if external_url_base in link['href']: link_href = link['href'].strip('/') return IMDB.normalize_imdb_id(movie[site_name]['id']) == \ IMDB.normalize_imdb_id(link_href.split('/')[-1]) return False
def _compare_external_links(page_source, movie, external_url_base, site_name): movie_details_page = BeautifulSoup(page_source, 'html.parser') try: external_links = movie_details_page.find( id='fi_info_ext').find_all('a') except AttributeError: return False for link in external_links: if external_url_base in link['href']: return IMDB.normalize_imdb_id(movie[site_name]['id']) == \ IMDB.normalize_imdb_id(link['href'].rstrip('/').split('/')[-1]) return False
def _compare_external_links(page_source, movie, external_url_base, site_name): movie_details_page = BeautifulSoup(page_source, "html.parser") external_links = (movie_details_page.find(id="info-wrapper").find( "ul", class_="external").find_all("a")) for link in external_links: if external_url_base in link["href"]: if site_name == "imdb": return IMDB.normalize_imdb_id( movie["imdb"]["id"]) == IMDB.normalize_imdb_id( link["href"].split("/")[-1]) return movie[site_name]["id"] == link["href"].split("/")[-1] return False
def setUp(self): self.site = IMDB(None)
def __init__(self, args): super(IMDBRatingsParser, self).__init__(IMDB(args), args) self.downloaded_file_name = 'ratings.csv'
def __init__(self, args): super(IMDBRatingsParser, self).__init__(IMDB(args), args)
def __init__(self, args): super(IMDBRatingsParser, self).__init__(IMDB(args), args) self.USERID = self._get_user_id() self.downloaded_file_name = 'ratings.csv'