|
1 | 1 | import logging |
2 | | -import os |
3 | | -import platform |
4 | 2 | import random |
5 | | -import sys |
| 3 | +import getpass |
6 | 4 | import time |
7 | 5 |
|
8 | 6 | from selenium import webdriver |
9 | 7 | from selenium.webdriver.chrome.options import Options |
10 | 8 |
|
11 | | -argv = sys.argv |
12 | | -argv_length = len(argv) |
13 | | - |
14 | | -username = argv[1] if argv_length > 1 else input("请输入登录账号\n") |
15 | | -password = argv[2] if argv_length > 2 else input("请输入登录密码\n") |
16 | | -role = int(argv[3]) if argv_length > 3 else input("请输入Role\n") |
17 | | - |
18 | | -if platform.system() == "Linux": |
19 | | - os.system("ps -ef | grep chrome | awk '{print $2}' | xargs kill -9") |
| 9 | +import jsloader |
| 10 | +from logger import log |
20 | 11 |
|
21 | 12 |
|
22 | 13 | def sleep(min_seconds=3, max_seconds=10): |
23 | 14 | """随机休眠""" |
24 | 15 | time.sleep(random.randint(min_seconds, max_seconds)) |
25 | 16 |
|
26 | 17 |
|
27 | | -logging.basicConfig(level='INFO', filename='robot.log', |
28 | | - format='%(asctime)s %(filename)s[%(lineno)d] %(name)s (%(levelname)s): %(message)s') |
29 | | -# 无头模式 |
30 | | -options = Options() |
31 | | -options.add_argument('--headless') |
32 | | -options.add_argument('log-level=3') |
33 | | -options.add_argument('--no-sandbox') |
34 | | -options.add_argument('--disable-gpu') |
35 | | -options.add_argument('window-size=1366x728') |
36 | | -options.add_argument('--disable-dev-shm-usage') |
37 | | -options.add_argument('blink-settings=imagesEnabled=false') |
38 | | -driver = webdriver.Chrome(chrome_options=options) |
39 | | - |
40 | | -# 登录简书 |
| 18 | +log.Logger('robot.log') |
41 | 19 | jianshu = "https://www.jianshu.com" |
42 | | -driver.get("{}/sign_in".format(jianshu)) |
43 | | -driver.maximize_window() |
44 | | - |
45 | | -# 豆瓣登录 |
46 | | -driver.find_element_by_xpath("/html/body/div[1]/div[2]/div/div/ul/li[4]").click() |
47 | | -driver.find_element_by_class_name("douban").click() |
48 | | -sleep() |
49 | | - |
50 | | -# 切换窗口 |
51 | | -driver.close() |
52 | | -window_handles = driver.window_handles |
53 | | -driver.switch_to.window(window_handles[-1]) |
54 | | -driver.find_element_by_id("inp-alias").send_keys(username) |
55 | | -driver.find_element_by_id("inp-pwd").send_keys(password) |
56 | | - |
57 | | -try: |
58 | | - driver.find_element_by_xpath("/html/body/div/div[2]/form[2]/div[4]/input[1]").click() |
59 | | -except: |
60 | | - logging.error("{} : 登录失败".format(role)) |
61 | | -else: |
62 | | - sleep() |
63 | | - |
64 | | - page_title = driver.title |
65 | | - if page_title.find("简书") == -1: |
66 | | - logging.error("{} ：{}".format(role, page_title)) |
67 | | - else: |
| 20 | +cookie_data = jsloader.load_cookie() |
| 21 | +for role, cookie in cookie_data.items(): |
| 22 | + # 无头模式 |
| 23 | + chrome_dir = r"--user-data-dir=C:\Users\{}\AppData\Local\Google\Chrome\User Data".format(getpass.getuser()) |
| 24 | + options = Options() |
| 25 | + options.add_argument(chrome_dir) |
| 26 | + options.add_argument('--headless') |
| 27 | + driver = webdriver.Chrome(options=options) |
| 28 | + try: |
68 | 29 | # 1. 收益 |
69 | 30 | driver.get('{}/mobile/fp?read_mode=night'.format(jianshu)), sleep(1, 2) |
70 | 31 | elements = driver.find_elements_by_class_name("order") |
@@ -95,6 +56,14 @@ def sleep(min_seconds=3, max_seconds=10): |
95 | 56 | except: |
96 | 57 | logging.error("Button_Element : {}".format(button_element)) |
97 | 58 | pass |
98 | | -finally: |
99 | | - driver.close() |
100 | | - driver.quit() |
| 59 | + except: |
| 60 | + logging.error("{} : 异常".format(role)) |
| 61 | + else: |
| 62 | + # 重新获取 cookie |
| 63 | + driver.add_cookie({"name": "remember_user_token", |
| 64 | + "value": ""}) |
| 65 | + pass |
| 66 | + finally: |
| 67 | + driver.delete_all_cookies() |
| 68 | + driver.close() |
| 69 | + driver.quit() |
0 commit comments