import time
import json
from selenium import webdriver

USERNAME=''
PASSWORD=''

driver = webdriver.Chrome()  # Optional argument, if not specified will search path.
driver.get('https://embed.music.apple.com/us/album/seriously-single/1397849421?app=music');
time.sleep(5) # Let the user actually see something!

sign_in_button = driver.find_element_by_css_selector('button.song__heading__auth')
sign_in_button.click()


#window_before = driver.window_handles[0]
#driver.switch_to.frame(0)
#window_after = driver.window_handles[1]
time.sleep(10)
print driver.window_handles
driver.switch_to.window(driver.window_handles[1])
driver.switch_to.frame(0)

time.sleep(10)

account_name_text_field = driver.find_element_by_id('account_name_text_field')
account_name_text_field.send_keys(USERNAME)
final_signin_button = driver.find_element_by_id('sign-in')
final_signin_button.click()

time.sleep(5)

password_text_field = driver.find_element_by_id('password_text_field')
password_text_field.send_keys(PASSWORD)


final_signin_button = driver.find_element_by_id('sign-in')
final_signin_button.click()

print 'please enter 2 factor auth code'
for i in xrange(0,180):
    print i
    time.sleep(1)


driver.switch_to_window(driver.window_handles[0])
cookies = driver.get_cookies()
print cookies
with open('cookie.json', 'w+') as f:
    f.write(json.dumps(cookies))


print 'saved cookies'

#search_box.send_keys('ChromeDriver')
#search_box.submit()
time.sleep(5) # Let the user actually see something!
driver.quit()