1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
import requests,json,re,time
from bs4 import BeautifulSoup
s = requests.Session()
def login(userinfo):
'''登录，获取token'''
print('开始登录...')
login_index_url = 'https://wallhaven.cc/login'
response = s.get(login_index_url)
html = response.text
bf = BeautifulSoup(html, 'lxml')
hidden = bf.find_all('input', {'type':'hidden'})
for i in hidden:
_token = i['value']
data = {
'_token' : _token,
}
data.update(userinfo)
login_url = 'https://wallhaven.cc/auth/login'
response = s.post(login_url, data=data, allow_redirects=False)
username = response.headers['location'].split('/')[-1] # 获取跳转URL
response = s.get(response.headers['location'])
if response.status_code == 200:
print("login success")
token = re.findall('<input type="hidden" name="_token" value="(.*?)">',response.text)[0]
return username,token
else:
print('login failed')
return response.status_code
def AddtoFav(pic_hashid,token,collectionID):
"""构造加入收藏的链接并请求"""
s.headers.update({"referer": f"https://wallhaven.cc/w/{pic_hashid}"})
favLink = f'https://wallhaven.cc/favorites/add?wallHashid={pic_hashid}&collectionId={collectionID}&_token={token}'
r = s.get(favLink)
if r.status_code==200:
print('成功加入收藏')
return 200
elif r.status_code==429:
print(r.text)
return 429
else:
print(r.text)
return 3
def GetCollectionPicIDs(apikey, collectionID, nickname):
'''返回相应ID在线收藏夹中图片ID列表'''
# 第一步获取改ID收藏夹的图片总条数
params = { "apikey": apikey }
r = requests.get('https://wallhaven.cc/api/v1/collections', params=params)
for i in r.json()['data']:
if i['id']==collectionID:
count = i['count']
print(f'正在获取在线收藏夹图片ID，约等待 {(count//24+2)*1.5} 秒.')
ids = []
pages = count//24+2
for page in range(1,pages):
print('Remain Time:\t',pages*1.5)
params = { "apikey": apikey,
"page": page,
}
r = requests.get(f'https://wallhaven.cc/api/v1/collections/{nickname}/{collectionID}', params=params)
ids += [record['id'] for record in r.json()['data']]
time.sleep(1.5) # 限制每分钟45次，QOS=1.3333
pages-=1
return ids
def main(userinfo:dict,pic_ids,collectionID,apikey):
headers = {'User-Agent' : 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.122 Safari/537.36'}
s.headers.update(headers)
nickname,token = login(userinfo)
headers = {
"accept": "*/*",
"accept-encoding": "gzip, deflate, br",
"accept-language": "zh-CN,zh;q=0.9,en-CN;q=0.8,en;q=0.7,zh-HK;q=0.6",
"cache-control": "no-cache",
"dnt": "1",
"pragma": "no-cache",
"sec-fetch-dest": "empty",
"sec-fetch-mode": "cors",
"sec-fetch-site": "same-origin",
"user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36",
"x-requested-with": "XMLHttpRequest",
}
s.headers.update(headers)
OnlinePicIDs = GetCollectionPicIDs(apikey, collectionID, nickname)
PicIDsReady2Add = [pic_id for pic_id in pic_ids if pic_id not in OnlinePicIDs]
length = len(PicIDsReady2Add)
if len(token)==40:
for picid in PicIDsReady2Add:
print('Still Remain:\t',length,end='\t')
time.sleep(1.5) # 休眠1.5秒可保证服务器不提示请求过于频繁
code = AddtoFav(picid,token,collectionID)
if code == 429:
time.sleep(3) # 429错误时，暂停1分钟再重试
AddtoFav(picid,token,collectionID)
print(picid)
elif code == 1:
time.sleep(3) # 防止报429 Too Many Requests 错误，间隔5秒请求
continue
elif code ==3: # 其他错误时打印图片ID，供下次继续
print(picid)
length -= 1
print('All Done!')
if __name__ == '__main__':
with open('bookmarks.json','r') as f:
bookmarks = json.load(f)
pic_ids = [bookmark["n"] for bookmark in bookmarks]
apikey = "***********************"
userinfo = {'username': '******', # 账号-为邮箱
'password': '++KwN9NGC@F34s3Ed++'} # 密码
collectionID = 972651
main(userinfo,pic_ids,collectionID,apikey)
|
