automation.py 文件源码

def login(username,password):
    print('logging in..')
    element = browser.find_element_by_name('username')
    element.clear()
    element.send_keys(username)

    element = browser.find_element_by_name('password')
    element.clear()
    element.send_keys(password)

    element.send_keys(Keys.RETURN)

    #check if the username and password are correct
    loginTime=time.time()
    while True:
        if browser.current_url=='https://www.instagram.com/':
            break
        elif time.time()-loginTime>10:                  #if login takes more than 10 seconds, tell user that his internet is slow
                print('Your Internet seems slow!\n',e)
                browser.close()
                sys.exit()
        else:
            try:
                element=browser.find_element_by_xpath("//*[@id='slfErrorAlert']")
                print('\n'+element.text)
                print('Enter your credentials again!\n')
                browser.close()
                main()
                sys.exit()
            except:
                #do nothing
                pass

    print('login complete')