twitter.py 文件源码

def twitter_tweet(self):
    print("1. write tweet")
    print("2. copy from clipboard")

    if six.PY2:
        choice = raw_input('Enter choice:')
    else:
        choice = input('Enter choice:')

    if int(choice) == 1:
        if six.PY2:
            tweet = raw_input('Enter tweet here:')
        else:
            tweet = input('Enter tweet here:')
    else:
        tweet = pyperclip.paste()

    driver = twitter_login(self)
    post_box = driver.find_element_by_id('tweet-box-home-timeline')
    post_box.send_keys(tweet)
    submit = driver.find_element_by_css_selector(
        'button.tweet-action.EdgeButton.EdgeButton--primary.js-tweet-btn')
    submit.click()
    print("Posted")

    return driver