login.py 1.0 KB

123456789101112131415161718192021222324252627282930313233
  1. from selenium.common.exceptions import NoSuchElementException
  2. import json
  3. import time
  4. def login(driver, cookie_path):
  5. driver.get('https://www.mcbbs.net/forum-chat-1.html')
  6. driver.delete_all_cookies()
  7. with open(cookie_path, 'r') as cookief:
  8. cookieslist = json.load(cookief)
  9. for cookie in cookieslist:
  10. driver.add_cookie(cookie)
  11. driver.get('https://www.mcbbs.net/home.php?mod=spacecp')
  12. time.sleep(2)
  13. try:
  14. element = driver.find_element_by_xpath("//div[@class='alert_info']/p")
  15. except NoSuchElementException:
  16. print('已登录')
  17. else:
  18. if '需要先登录' in element.text:
  19. return False
  20. else:
  21. return True
  22. try:
  23. element = driver.find_element_by_xpath("//div[@class='alert_error']")
  24. except NoSuchElementException:
  25. return True
  26. else:
  27. if '空间已被锁定无法访问' in element.text:
  28. print(cookie_path + '被禁言')
  29. return False
  30. else:
  31. return True