84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
|
|
from DrissionPage import ChromiumPage, ChromiumOptions
|
|
from requests.cookies import RequestsCookieJar
|
|
from retry import retry
|
|
import time
|
|
import os
|
|
import pickle
|
|
|
|
|
|
# 随机端口
|
|
# co = ChromiumOptions().auto_port()
|
|
# page1 = ChromiumPage(co)
|
|
# page = ChromiumPage(co)
|
|
|
|
@retry(tries=50, delay=1)
|
|
def get_cookies(page):
|
|
tab = page.new_tab(url='https://www.douyin.com/user/self')
|
|
|
|
i = 1
|
|
while True:
|
|
if i == 1:
|
|
print(f'第{i}次尝试登录...')
|
|
# tab.wait.eles_loaded('x://div[@data-e2e="user-detail"]//h1')
|
|
if tab.ele('x://div[@data-e2e="user-detail"]//h1').text != '未登录':
|
|
print('登录成功')
|
|
# print('未登录-登录成功')
|
|
break
|
|
# tab.wait.eles_loaded('x://*[text()="编辑资料"]')
|
|
if tab.ele('x://*[text()="编辑资料"]'):
|
|
print('登录成功')
|
|
# print('编辑资料-登录成功')
|
|
break
|
|
i += 1
|
|
print(f'第{i}次尝试登录...')
|
|
time.sleep(1)
|
|
|
|
title = tab.title
|
|
turl = tab.ele('x://link[@rel="canonical"]').attr('href')
|
|
|
|
Douyin_account = tab.ele('x://*[text()="抖音号："]').text
|
|
Douyin_account = Douyin_account.replace('抖音号：', '')
|
|
# print(title)
|
|
# print(turl)
|
|
filename = title.replace('的抖音 - 抖音', '')
|
|
if filename == "":
|
|
filename = tab.ele('x://div[@data-e2e="user-detail"]//h1').text
|
|
cookies = tab.cookies()
|
|
cookie_jar = RequestsCookieJar()
|
|
for cookie in cookies:
|
|
cookie_jar.set(cookie['name'], cookie['value'], domain=cookie['domain'])
|
|
turl = turl.replace('https://www.douyin.com/user/', '')
|
|
cntext = f'{turl}|{filename}|{Douyin_account}\r'
|
|
if os.path.isfile('cookie_name.txt'):
|
|
with open('cookie_name.txt', 'rb') as fp:
|
|
key = fp.read().decode('utf-8')
|
|
# 分割字符串得到列表
|
|
klist = key.split('\r')
|
|
|
|
seen = set()
|
|
unique_klist = [x for x in klist if not (x in seen or seen.add(x))]
|
|
cntext += '\r'.join(unique_klist)
|
|
|
|
with open('./cookie_name.txt', 'w', encoding='utf-8') as f:
|
|
f.write(cntext)
|
|
with open(f'./{turl}.pkl', 'wb') as f:
|
|
pickle.dump(cookie_jar, f)
|
|
print('登录成功')
|
|
|
|
|
|
|
|
def get_cookie():
|
|
co = ChromiumOptions().auto_port()
|
|
page = ChromiumPage(co)
|
|
get_cookies(page)
|
|
page.quit()
|
|
return True
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
get_cookie()
|