afraca · June 12, 2018 10:58 · afraca · Jun 12, 2018
diff --git a/main.py b/main.py
 from selenium import webdriver
 from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
 from selenium.webdriver.remote.webelement import WebElement
 from selenium.common.exceptions import *
 from selenium.webdriver.firefox.options import Options
 import getpass
 from typing import Dict, Tuple, List
 import os
 import csv
 import re
 from difflib import SequenceMatcher
 import time
 import json
 import click

 # These type aliases makes the dictionary passing easier to understand
 ImdbId = str
 ImdbTitle = str
 ImdbRating = int

 # Netflix renders quickly enough, but finishes very slowly
 DRIVER_TIMEOUT = 15

 NETFLIX_LOGIN_URL = 'https://www.netflix.com/nl-en/login'

 # IMDB will set something (url?) required up in the "pre-login page"
 IMDB_PRE_LOGIN_URL = 'https://www.imdb.com/registration/signin'
 IMDB_LOGIN_URL = 'https://www.imdb.com/ap/signin?clientContext=133-8770098-9132729&openid.pape.max_auth_age=0&openid.return_to=https%3A%2F%2Fwww.imdb.com%2Fap-signin-handler&openid.identity=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0%2Fidentifier_select&openid.assoc_handle=imdb_us&openid.mode=checkid_setup&siteState=eyJvcGVuaWQuYXNzb2NfaGFuZGxlIjoiaW1kYl91cyIsInJlZGlyZWN0VG8iOiJodHRwczovL3d3dy5pbWRiLmNvbS8_cmVmXz1sb2dpbiJ9&openid.claimed_id=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0%2Fidentifier_select&openid.ns=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0&&tag=imdbtag_reg-20 '
 IMDB_TEMP_FILE_NAME = 'ratings.csv'

 RESULTS_FILE = 'results.json'
 RESULTS = {}
 # Poor man's ENUM
 RESULT_FOUND_VOTED_UP = 0
 RESULT_FOUND_VOTED_DOWN = 1
 RESULT_NOT_FOUND = 2
 RESULTS_SHOULD_BE_IGNORED = False


 def get_driver(headful: bool = False) -> webdriver.Firefox:
    options = Options()
    if not headful:
        options.headless = True
    profile = FirefoxProfile()
    profile.set_preference("browser.download.folderList", 2)
    profile.set_preference("browser.download.manager.showWhenStarting", False)
    profile.set_preference("browser.download.dir", os.getcwd())
    profile.set_preference("browser.helperApps.neverAsk.saveToDisk", 'text/csv')
    driver = webdriver.Firefox(firefox_options=options, firefox_profile=profile)
    driver.set_page_load_timeout(DRIVER_TIMEOUT)

    return driver


 ###################################################################################################


 def should_update_ratings() -> bool:
    answer = input('(Re-)download imdb ratings? (Y/N) (default: N):')
    if answer == 'Y':
        return True
    if answer == 'N' or answer == '':
        return False
    # Repeat question on invalid input
    return should_update_ratings()


 def login_to_imdb(driver: webdriver.Firefox, username: str, password: str):
    # As stated on global value, IMDB does something weird in login flow, so we need the 'pre-login' visit
    driver.get(IMDB_PRE_LOGIN_URL)
    login_button_elem = driver.find_element_by_partial_link_text('Sign in with IMDb')
    login_button_elem.click()
    user_elem = driver.find_element_by_id('ap_email')
    user_elem.send_keys(username)
    pass_elem = driver.find_element_by_id('ap_password')
    pass_elem.send_keys(password)
    submit = driver.find_element_by_id('signInSubmit')
    submit.click()


 def fetch_history(driver: webdriver.Firefox, download_csv: bool, user_id: str = None) -> \
        Dict[Tuple[ImdbId, ImdbTitle], ImdbRating]:
    if download_csv:
        if user_id is None:
            raise ValueError('user_id for imdb fetching is required')
        if os.path.exists(IMDB_TEMP_FILE_NAME):
            remove_history_file()
        history_url = 'https://imdb.com/user/{}/ratings/export'.format(user_id)
        # Download finishes quick, but somehow we never register an 'end',
        # so just set timeout and continue if file is there
        driver.set_page_load_timeout(5)
        try:
            driver.get(history_url)
        except TimeoutException:
            if not os.path.exists(IMDB_TEMP_FILE_NAME):
                raise
        driver.set_page_load_timeout(DRIVER_TIMEOUT)

    return read_history_file_to_dict()


 def read_history_file_to_dict() -> Dict[Tuple[ImdbId, ImdbTitle], ImdbRating]:
    ratings = {}
    with open(IMDB_TEMP_FILE_NAME, newline='') as csvfile:
        reader = csv.reader(csvfile, delimiter=',')
        # skip header
        next(reader, None)
        for row in reader:
            try:
                imdb_id, title, rating = csv_row_to_dict_entry(row)
                ratings[(imdb_id, title)] = rating
            except ValueError as e:
                print(e, row[0])
    print('Movie-To-Rating mapping created')
    return ratings


 def csv_row_to_dict_entry(row: List) -> Tuple[ImdbId, ImdbTitle, ImdbRating]:
    if len(row) < 10:
        raise ValueError('csv malformed')
    if row[5] != 'movie':
        raise ValueError('Not a movie')
    return str(row[0]), str(row[3]), int(row[1])


 def remove_history_file():
    print('Removing ratings file')
    os.remove(IMDB_TEMP_FILE_NAME)


 ###################################################################################################


 def login_to_netflix(driver: webdriver.Firefox, username: str, password: str):
    driver.get(NETFLIX_LOGIN_URL)
    user_elem = driver.find_element_by_id('email')
    user_elem.send_keys(username)
    pass_elem = driver.find_element_by_id('password')
    pass_elem.send_keys(password)
    submit = driver.find_element_by_css_selector('.btn.login-button.btn-submit')
    submit.click()
    print('Logged in to Netflix')


 def choose_netflix_profile(driver: webdriver.Firefox, profile_id: str):
    print('Choosing netflix profile.')
    driver.get('https://www.netflix.com/SwitchProfile?tkn={}'.format(profile_id))


 def vote_with_mapping(driver: webdriver.Firefox, mapping: Dict[Tuple[ImdbId, ImdbTitle], ImdbRating],
                      ignore_results: bool = False):
    for (imdb_id, title), rating in mapping.items():
        if not should_try_vote(imdb_id, rating, ignore_results):
            print('No netflix action required for {}'.format(title))
            continue

        try:
            driver.get('https://netflix.com/search?q={}'.format(title))
        except TimeoutException:
            RESULTS[imdb_id] = RESULT_NOT_FOUND
            print('Searched timed out for {}'.format(title))
            continue

        links_for_query = driver.find_elements_by_xpath('//a[@href][@aria-label]')
        # Last results are bad, exclude them, speeds it up
        if len(links_for_query) > 25:
            links_for_query = links_for_query[:25]
        ids_with_similarity = ids_with_similarity_all_links(links_for_query, title)
        if len(ids_with_similarity) == 0:
            print('No results found for {}'.format(title))
            RESULTS[imdb_id] = RESULT_NOT_FOUND
            continue
        best_matching_id = max(ids_with_similarity, key=lambda key: ids_with_similarity[key])
        if ids_with_similarity[best_matching_id] < 0.9:
            print('Could not find similar enough match for {}'.format(title))
            RESULTS[imdb_id] = RESULT_NOT_FOUND
            if ids_with_similarity[best_matching_id] > 0.75:
                print('(But it was really similar!...)')
            continue
        print('Closest id for {} was {}'.format(title, best_matching_id))
        vote_movie(driver, best_matching_id, imdb_id, rating)


 def should_try_vote(imdb_id: str, rating: int, ignore_results: bool = False) -> bool:
    if ignore_results:
        return True
    # Movie not yet processed
    if imdb_id not in RESULTS:
        return True
    # It was not on Netflix in previous processing
    # (So only way to accommodate Netflix changes is to rerun it ALL unfortunately)
    if RESULTS[imdb_id] == RESULT_NOT_FOUND:
        return False
    vote_up = rating >= 7
    # Movie was voted incorrectly
    if RESULTS[imdb_id] == RESULT_FOUND_VOTED_UP and not vote_up:
        return True
    # Movie was voted incorrectly
    if RESULTS[imdb_id] == RESULT_FOUND_VOTED_DOWN and vote_up:
        return True
    # Movie was voted correctly
    return False


 def ids_with_similarity_all_links(links: List[WebElement], title: str) -> Dict[str, float]:
    ids_with_similarity = {}
    for elem in links:
        label = elem.get_attribute('aria-label')
        href = elem.get_attribute('href')
        if label is None or href is None:
            print('XPath returned illegal element')
            continue
        maybe_id = re.search(r"\d{5,10}", href)
        if maybe_id is None:
            continue
        entry_id = maybe_id.group()
        ids_with_similarity[entry_id] = SequenceMatcher(None, label, title).ratio()

    return ids_with_similarity


 def vote_movie(driver: webdriver.Firefox, netflix_id: str, imdb_id: str, rating: int):
    # Netflix renders quick, completes slowly! Just try to get movie details couple of times
    tries = 3
    for i in range(tries):
        try:
            driver.get('https://netflix.com/title/{}'.format(netflix_id))
            break
        except TimeoutException:
            if i == (tries - 1):
                # Now we register it as really problematic
                raise

    try:
        driver.find_element_by_css_selector('.thumbs-component.rated')
        rated = True
    except NoSuchElementException:
        rated = False
    if rated:
        # remove old vote
        # Even though manual inspection only shows 1, we get 2 elements (1 for up 1 for down??)
        # Simply try them both for now...
        unvote_elements = driver.find_elements_by_css_selector('a.nf-svg-button.simpleround')

        def register_success():
            print('Removed old vote for {}'.format(id))

        try:
            unvote_elements[0].click()
            register_success()
        except ElementClickInterceptedException:
            unvote_elements[1].click()
            register_success()
    upvoting = rating >= 7
    data_rating_val = '"2"' if upvoting else '"1"'
    print('Rating was {} so voting {} for {}'.format(rating, 'up' if upvoting else 'down', imdb_id))
    if upvoting:
        RESULTS[imdb_id] = RESULT_FOUND_VOTED_UP
    else:
        RESULTS[imdb_id] = RESULT_FOUND_VOTED_DOWN
    selector = 'a.nf-svg-button[data-rating={}]'.format(data_rating_val)
    thumb = driver.find_element_by_css_selector(selector)
    # After "unvoting" we need to wait before we can register new vote. This SHOULD work with the
    # "expected conditions" of selenium (element_to_be_clickable), but did not work for me. Use stupid sleep...
    time.sleep(0.5)
    thumb.click()


 ###################################################################################################

 def print_statistics():
    # Just to know where we are at start and end of script
    num_not_found = sum(1 for result in RESULTS.items() if result[1] == RESULT_NOT_FOUND)
    num_found_voted_up = sum(1 for result in RESULTS.items() if result[1] == RESULT_FOUND_VOTED_UP)
    num_found_voted_down = sum(1 for result in RESULTS.items() if result[1] == RESULT_FOUND_VOTED_DOWN)
    print('Not found:{}'.format(num_not_found))
    print('Found, voted up: {}'.format(num_found_voted_up))
    print('Found, voted down: {}'.format(num_found_voted_down))


 def set_results_from_file():
    global RESULTS
    if os.path.exists(RESULTS_FILE):
        with open(RESULTS_FILE) as f:
            RESULTS = json.load(f)
        print('Loaded results from json')


 def set_file_from_results():
    with open(RESULTS_FILE, 'w') as f:
        f.write(json.dumps(RESULTS))
    print('Wrote results to json')


 @click.command()
 @click.argument('netflix-username')
 @click.argument('netflix-profile-id')
 @click.option('--imdb-username')
 @click.option('--imdb-id', default=None)
 @click.option('--headful', default=False, is_flag=True)
 @click.option('--ignore-results', default=False, is_flag=True)
 def main(netflix_username: str, netflix_profile_id: str, imdb_username: str, imdb_id: str, headful: bool,
         ignore_results: bool):
    driver = None
    set_results_from_file()
    print_statistics()

    def shutdown():
        nonlocal driver
        set_file_from_results()
        print_statistics()
        if driver is not None:
            driver.quit()

    download_csv = should_update_ratings()
    if download_csv:
        if imdb_username is None:
            raise ValueError('Missing --imdb-username value')
        if imdb_id is None:
            raise ValueError('Missing --imdb-id value')
        imdb_pass = getpass.getpass('IMDB Password:')
    else:
        imdb_pass = None

    netflix_pass = getpass.getpass('Netflix Password:')
    try:
        driver = get_driver(headful)
        if download_csv:
            login_to_imdb(driver, imdb_username, imdb_pass)
        login_to_netflix(driver, netflix_username, netflix_pass)
        mapping = fetch_history(driver, download_csv, imdb_id)
        choose_netflix_profile(driver, netflix_profile_id)
        vote_with_mapping(driver, mapping, ignore_results)
        print('All done!')
        shutdown()
        remove_history_file()
    except:
        shutdown()
        raise


 if __name__ == '__main__':
    main()
	from selenium import webdriver
	from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
	from selenium.webdriver.remote.webelement import WebElement
	from selenium.common.exceptions import *
	from selenium.webdriver.firefox.options import Options
	import getpass
	from typing import Dict, Tuple, List
	import os
	import csv
	import re
	from difflib import SequenceMatcher
	import time
	import json
	import click

	# These type aliases makes the dictionary passing easier to understand
	ImdbId = str
	ImdbTitle = str
	ImdbRating = int

	# Netflix renders quickly enough, but finishes very slowly
	DRIVER_TIMEOUT = 15

	NETFLIX_LOGIN_URL = 'https://www.netflix.com/nl-en/login'

	# IMDB will set something (url?) required up in the "pre-login page"
	IMDB_PRE_LOGIN_URL = 'https://www.imdb.com/registration/signin'
	IMDB_LOGIN_URL = 'https://www.imdb.com/ap/signin?clientContext=133-8770098-9132729&openid.pape.max_auth_age=0&openid.return_to=https%3A%2F%2Fwww.imdb.com%2Fap-signin-handler&openid.identity=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0%2Fidentifier_select&openid.assoc_handle=imdb_us&openid.mode=checkid_setup&siteState=eyJvcGVuaWQuYXNzb2NfaGFuZGxlIjoiaW1kYl91cyIsInJlZGlyZWN0VG8iOiJodHRwczovL3d3dy5pbWRiLmNvbS8_cmVmXz1sb2dpbiJ9&openid.claimed_id=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0%2Fidentifier_select&openid.ns=http%3A%2F%2Fspecs.openid.net%2Fauth%2F2.0&&tag=imdbtag_reg-20 '
	IMDB_TEMP_FILE_NAME = 'ratings.csv'

	RESULTS_FILE = 'results.json'
	RESULTS = {}
	# Poor man's ENUM
	RESULT_FOUND_VOTED_UP = 0
	RESULT_FOUND_VOTED_DOWN = 1
	RESULT_NOT_FOUND = 2
	RESULTS_SHOULD_BE_IGNORED = False


	def get_driver(headful: bool = False) -> webdriver.Firefox:
	options = Options()
	if not headful:
	options.headless = True
	profile = FirefoxProfile()
	profile.set_preference("browser.download.folderList", 2)
	profile.set_preference("browser.download.manager.showWhenStarting", False)
	profile.set_preference("browser.download.dir", os.getcwd())
	profile.set_preference("browser.helperApps.neverAsk.saveToDisk", 'text/csv')
	driver = webdriver.Firefox(firefox_options=options, firefox_profile=profile)
	driver.set_page_load_timeout(DRIVER_TIMEOUT)

	return driver


	###################################################################################################


	def should_update_ratings() -> bool:
	answer = input('(Re-)download imdb ratings? (Y/N) (default: N):')
	if answer == 'Y':
	return True
	if answer == 'N' or answer == '':
	return False
	# Repeat question on invalid input
	return should_update_ratings()


	def login_to_imdb(driver: webdriver.Firefox, username: str, password: str):
	# As stated on global value, IMDB does something weird in login flow, so we need the 'pre-login' visit
	driver.get(IMDB_PRE_LOGIN_URL)
	login_button_elem = driver.find_element_by_partial_link_text('Sign in with IMDb')
	login_button_elem.click()
	user_elem = driver.find_element_by_id('ap_email')
	user_elem.send_keys(username)
	pass_elem = driver.find_element_by_id('ap_password')
	pass_elem.send_keys(password)
	submit = driver.find_element_by_id('signInSubmit')
	submit.click()


	def fetch_history(driver: webdriver.Firefox, download_csv: bool, user_id: str = None) -> \
	Dict[Tuple[ImdbId, ImdbTitle], ImdbRating]:
	if download_csv:
	if user_id is None:
	raise ValueError('user_id for imdb fetching is required')
	if os.path.exists(IMDB_TEMP_FILE_NAME):
	remove_history_file()
	history_url = 'https://imdb.com/user/{}/ratings/export'.format(user_id)
	# Download finishes quick, but somehow we never register an 'end',
	# so just set timeout and continue if file is there
	driver.set_page_load_timeout(5)
	try:
	driver.get(history_url)
	except TimeoutException:
	if not os.path.exists(IMDB_TEMP_FILE_NAME):
	raise
	driver.set_page_load_timeout(DRIVER_TIMEOUT)

	return read_history_file_to_dict()


	def read_history_file_to_dict() -> Dict[Tuple[ImdbId, ImdbTitle], ImdbRating]:
	ratings = {}
	with open(IMDB_TEMP_FILE_NAME, newline='') as csvfile:
	reader = csv.reader(csvfile, delimiter=',')
	# skip header
	next(reader, None)
	for row in reader:
	try:
	imdb_id, title, rating = csv_row_to_dict_entry(row)
	ratings[(imdb_id, title)] = rating
	except ValueError as e:
	print(e, row[0])
	print('Movie-To-Rating mapping created')
	return ratings


	def csv_row_to_dict_entry(row: List) -> Tuple[ImdbId, ImdbTitle, ImdbRating]:
	if len(row) < 10:
	raise ValueError('csv malformed')
	if row[5] != 'movie':
	raise ValueError('Not a movie')
	return str(row[0]), str(row[3]), int(row[1])


	def remove_history_file():
	print('Removing ratings file')
	os.remove(IMDB_TEMP_FILE_NAME)


	###################################################################################################


	def login_to_netflix(driver: webdriver.Firefox, username: str, password: str):
	driver.get(NETFLIX_LOGIN_URL)
	user_elem = driver.find_element_by_id('email')
	user_elem.send_keys(username)
	pass_elem = driver.find_element_by_id('password')
	pass_elem.send_keys(password)
	submit = driver.find_element_by_css_selector('.btn.login-button.btn-submit')
	submit.click()
	print('Logged in to Netflix')


	def choose_netflix_profile(driver: webdriver.Firefox, profile_id: str):
	print('Choosing netflix profile.')
	driver.get('https://www.netflix.com/SwitchProfile?tkn={}'.format(profile_id))


	def vote_with_mapping(driver: webdriver.Firefox, mapping: Dict[Tuple[ImdbId, ImdbTitle], ImdbRating],
	ignore_results: bool = False):
	for (imdb_id, title), rating in mapping.items():
	if not should_try_vote(imdb_id, rating, ignore_results):
	print('No netflix action required for {}'.format(title))
	continue

	try:
	driver.get('https://netflix.com/search?q={}'.format(title))
	except TimeoutException:
	RESULTS[imdb_id] = RESULT_NOT_FOUND
	print('Searched timed out for {}'.format(title))
	continue

	links_for_query = driver.find_elements_by_xpath('//a[@href][@aria-label]')
	# Last results are bad, exclude them, speeds it up
	if len(links_for_query) > 25:
	links_for_query = links_for_query[:25]
	ids_with_similarity = ids_with_similarity_all_links(links_for_query, title)
	if len(ids_with_similarity) == 0:
	print('No results found for {}'.format(title))
	RESULTS[imdb_id] = RESULT_NOT_FOUND
	continue
	best_matching_id = max(ids_with_similarity, key=lambda key: ids_with_similarity[key])
	if ids_with_similarity[best_matching_id] < 0.9:
	print('Could not find similar enough match for {}'.format(title))
	RESULTS[imdb_id] = RESULT_NOT_FOUND
	if ids_with_similarity[best_matching_id] > 0.75:
	print('(But it was really similar!...)')
	continue
	print('Closest id for {} was {}'.format(title, best_matching_id))
	vote_movie(driver, best_matching_id, imdb_id, rating)


	def should_try_vote(imdb_id: str, rating: int, ignore_results: bool = False) -> bool:
	if ignore_results:
	return True
	# Movie not yet processed
	if imdb_id not in RESULTS:
	return True
	# It was not on Netflix in previous processing
	# (So only way to accommodate Netflix changes is to rerun it ALL unfortunately)
	if RESULTS[imdb_id] == RESULT_NOT_FOUND:
	return False
	vote_up = rating >= 7
	# Movie was voted incorrectly
	if RESULTS[imdb_id] == RESULT_FOUND_VOTED_UP and not vote_up:
	return True
	# Movie was voted incorrectly
	if RESULTS[imdb_id] == RESULT_FOUND_VOTED_DOWN and vote_up:
	return True
	# Movie was voted correctly
	return False


	def ids_with_similarity_all_links(links: List[WebElement], title: str) -> Dict[str, float]:
	ids_with_similarity = {}
	for elem in links:
	label = elem.get_attribute('aria-label')
	href = elem.get_attribute('href')
	if label is None or href is None:
	print('XPath returned illegal element')
	continue
	maybe_id = re.search(r"\d{5,10}", href)
	if maybe_id is None:
	continue
	entry_id = maybe_id.group()
	ids_with_similarity[entry_id] = SequenceMatcher(None, label, title).ratio()

	return ids_with_similarity


	def vote_movie(driver: webdriver.Firefox, netflix_id: str, imdb_id: str, rating: int):
	# Netflix renders quick, completes slowly! Just try to get movie details couple of times
	tries = 3
	for i in range(tries):
	try:
	driver.get('https://netflix.com/title/{}'.format(netflix_id))
	break
	except TimeoutException:
	if i == (tries - 1):
	# Now we register it as really problematic
	raise

	try:
	driver.find_element_by_css_selector('.thumbs-component.rated')
	rated = True
	except NoSuchElementException:
	rated = False
	if rated:
	# remove old vote
	# Even though manual inspection only shows 1, we get 2 elements (1 for up 1 for down??)
	# Simply try them both for now...
	unvote_elements = driver.find_elements_by_css_selector('a.nf-svg-button.simpleround')

	def register_success():
	print('Removed old vote for {}'.format(id))

	try:
	unvote_elements[0].click()
	register_success()
	except ElementClickInterceptedException:
	unvote_elements[1].click()
	register_success()
	upvoting = rating >= 7
	data_rating_val = '"2"' if upvoting else '"1"'
	print('Rating was {} so voting {} for {}'.format(rating, 'up' if upvoting else 'down', imdb_id))
	if upvoting:
	RESULTS[imdb_id] = RESULT_FOUND_VOTED_UP
	else:
	RESULTS[imdb_id] = RESULT_FOUND_VOTED_DOWN
	selector = 'a.nf-svg-button[data-rating={}]'.format(data_rating_val)
	thumb = driver.find_element_by_css_selector(selector)
	# After "unvoting" we need to wait before we can register new vote. This SHOULD work with the
	# "expected conditions" of selenium (element_to_be_clickable), but did not work for me. Use stupid sleep...
	time.sleep(0.5)
	thumb.click()


	###################################################################################################

	def print_statistics():
	# Just to know where we are at start and end of script
	num_not_found = sum(1 for result in RESULTS.items() if result[1] == RESULT_NOT_FOUND)
	num_found_voted_up = sum(1 for result in RESULTS.items() if result[1] == RESULT_FOUND_VOTED_UP)
	num_found_voted_down = sum(1 for result in RESULTS.items() if result[1] == RESULT_FOUND_VOTED_DOWN)
	print('Not found:{}'.format(num_not_found))
	print('Found, voted up: {}'.format(num_found_voted_up))
	print('Found, voted down: {}'.format(num_found_voted_down))


	def set_results_from_file():
	global RESULTS
	if os.path.exists(RESULTS_FILE):
	with open(RESULTS_FILE) as f:
	RESULTS = json.load(f)
	print('Loaded results from json')


	def set_file_from_results():
	with open(RESULTS_FILE, 'w') as f:
	f.write(json.dumps(RESULTS))
	print('Wrote results to json')


	@click.command()
	@click.argument('netflix-username')
	@click.argument('netflix-profile-id')
	@click.option('--imdb-username')
	@click.option('--imdb-id', default=None)
	@click.option('--headful', default=False, is_flag=True)
	@click.option('--ignore-results', default=False, is_flag=True)
	def main(netflix_username: str, netflix_profile_id: str, imdb_username: str, imdb_id: str, headful: bool,
	ignore_results: bool):
	driver = None
	set_results_from_file()
	print_statistics()

	def shutdown():
	nonlocal driver
	set_file_from_results()
	print_statistics()
	if driver is not None:
	driver.quit()

	download_csv = should_update_ratings()
	if download_csv:
	if imdb_username is None:
	raise ValueError('Missing --imdb-username value')
	if imdb_id is None:
	raise ValueError('Missing --imdb-id value')
	imdb_pass = getpass.getpass('IMDB Password:')
	else:
	imdb_pass = None

	netflix_pass = getpass.getpass('Netflix Password:')
	try:
	driver = get_driver(headful)
	if download_csv:
	login_to_imdb(driver, imdb_username, imdb_pass)
	login_to_netflix(driver, netflix_username, netflix_pass)
	mapping = fetch_history(driver, download_csv, imdb_id)
	choose_netflix_profile(driver, netflix_profile_id)
	vote_with_mapping(driver, mapping, ignore_results)
	print('All done!')
	shutdown()
	remove_history_file()
	except:
	shutdown()
	raise


	if __name__ == '__main__':
	main()