unfollow_util.py 文件源码

def unfollow_users(browser, amount):
  """Unfollows given amount of users"""

  unfollowed = 0
  last_length = 0

  #Click on the view all button on the main page to load all the recommended accounts
  browser.get('https://twitter.com/following')

  body_elem = browser.find_element_by_tag_name('body')

  timeline = browser.find_elements_by_xpath(
    '//div[@class = "ProfileCard-actions"]//span[contains(@class, "user-actions-follow-button js-follow-btn follow-button")]')

  while len(timeline) < amount and len(timeline) > last_length:
    last_length = len(timeline)
    body_elem.send_keys(Keys.END)
    sleep(1)
    body_elem.send_keys(Keys.HOME)
    sleep(1)

    timeline = browser.find_elements_by_xpath(
      '//div[@class = "ProfileCard-actions"]//span[contains(@class, "user-actions-follow-button js-follow-btn follow-button")]')

  if len(timeline) > amount:
    unfollowed = amount
  else:
    unfollowed = len(timeline)

  action_chain = Actions(browser)

  for index, button in enumerate(timeline[:unfollowed]):
    action_chain.move_to_element(button)
    action_chain.wait(1)
    action_chain.move_to_element(button)
    action_chain.click()
    action_chain.wait(1)
    action_chain.print_it(str(index + 1) + '/' + str(unfollowed))

  action_chain.perform()

  return unfollowed