import requests
import base64
from datetime import datetime
from datetime import timedelta
from datetime import timezone
from bs4 import BeautifulSoup
import time
import ddddocr
import os
baidu_cookie = os.environ[“BAIDU_COOKIE”]
SHA_TZ = timezone(
timedelta(hours=8),
name=’Asia/Shanghai’,
)
utc_now = datetime.utcnow().replace(tzinfo=timezone.utc)
beijing_now = utc_now.astimezone(SHA_TZ)
# 定义headers
headers = {
‘User-Agent’: ‘Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.110 Safari/537.36’,
}
# 定义baidu_headers,cookie需要 BAIDUID BDUSS ,STOKEN
baidu_headers = {
“user-agent”: “Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/”
“97.0.4692.99 Safari/537.36 Edg/97.0.1072.69”,
“Referer”: “https://pan.baidu.com/disk/home?from=newversion&stayAtHome=true”,
“cookie”: baidu_cookie,
‘Connection’: ‘close’
}
# 获取bing每日一图
def get_bing_pic():
url = ‘https://cn.bing.com/’
r = requests.get(url, headers=headers)
soup = BeautifulSoup(r.text, ‘html.parser’)
# 寻找link标签,preloadBg图片的url
link = soup.find(‘link’, id=’preloadBg’)
# 获取图片url
img_url = link[‘href’]
# 分割url通过&分割
img_url = img_url.split(‘&’)[0]
# 判断url是否有’1920×1080′ 如果有则替换为’UHD’,UHD为4K图片标识
if ‘1920×1080’ in img_url:
img_url = img_url.replace(‘1920×1080’, ‘UHD’)
img_url = “https://s.cn.bing.net” + img_url
# 获取图片名称
img_name = img_url.split(‘=’)[1]
# 返回图片url
return img_url, img_name
# 判断网盘是否登录
def is_login():
url = ‘https://pan.baidu.com/disk/home?from=newversion&stayAtHome=true#/all?path=%2F&vmode=list’
r = requests.get(url, headers=baidu_headers)
# 通过网页源码判断是否登录
if ‘百度网盘-全部文件’ in r.text:
print(‘网盘登录成功’)
return True
else:
print(“更换COOKIE”)
return False
# 获取logid,bdstoken
def get_logid_bdstoken():
# 获取bdstoken的url
get_bdstoken_url = “http://pan.baidu.com/api/gettemplatevariable?clienttype=0&app_id=250528&web=1&fields=[%22bdstoken%22]”
bdstoken = requests.get(get_bdstoken_url, headers=baidu_headers).json().get(‘result’).get(‘bdstoken’)
# 获取logid
logid = str(base64.b64encode(
requests.get(get_bdstoken_url, headers=baidu_headers).request.headers.get(‘cookie’).split(‘;’)[0].strip(
“BAIDUID=”).encode(
‘utf-8’))).replace(“b'”, “”).replace(“‘”, “”)
return logid, bdstoken
# 验证网盘目标目录文件夹是否存在
def is_dir_exist(logid, bdstoken):
url = f”https://pan.baidu.com/api/list?clienttype=0&app_id=250528&web=1&dp-logid={logid}&order=time&desc=1&dir=/图片&num=100&page=1″
url1 = f”https://pan.baidu.com/api/create?a=commit&channel=chunlei&web=1&app_id=250528&bdstoken={bdstoken}&logid={logid}&clienttype=0″
data = {
“path”: “/图片/” + str(beijing_now.date()),
“isdir”: 1,
‘block_list’: []
}
r = requests.get(url, headers=baidu_headers)
for i in r.json().get(‘list’):
if str(beijing_now.date()) in i.get(‘server_filename’):
return True
else:
requests.post(url1, headers=baidu_headers, data=data)
return is_dir_exist(logid, bdstoken)
def is_file_exist(logid, img_name):
url = f”https://pan.baidu.com/api/list?clienttype=0&app_id=250528&web=1&dp-logid={logid}&order=time&desc=1&dir=/图片/{str(beijing_now.date())}&num=100&page=1″
# 发送请求
r = requests.get(url=url, headers=baidu_headers)
# 判断是否下载成功
if len(r.json().get(“list”)) == 0:
print(“今日未下载”)
return True
for i in r.json().get(‘list’):
if img_name in i.get(‘path’):
print(‘今天图片下载完成’)
return False
# 网盘离线下载,参数为logid,bdstoken,图片url
def download_offline(logid, bdstoken, img_url):
# 拼接网盘离线下载url
download_offline_url = f”https://pan.baidu.com/rest/2.0/services/cloud_dl?channel=chunlei&web=1&app_id=250528&bdstoken={bdstoken}&logid={logid}&clienttype=0″
# 定义data
data = {
“method”: “add_task”,
“app_id”: “250528”,
“source_url”: img_url,
“save_path”: f”/图片/{str(beijing_now.date())}” # 网盘目录
}
# 发送请求
r = requests.post(download_offline_url, headers=baidu_headers, data=data)
if “Invalid source url” in r.json():
print(“重新下载 下载地址无效，请核对”)
download_offline(logid, bdstoken, img_url)
if ‘vcode’ in r.json():
print(“需要验证码调用”)
download_offline_captcha(logid, bdstoken, img_url, r)
# 需要验证码的离线下载
def download_offline_captcha(logid, bdstoken, img_url, request):
# 拼接网盘离线下载url
download_offline_url = f”https://pan.baidu.com/rest/2.0/services/cloud_dl?channel=chunlei&web=1&app_id=250528&bdstoken={bdstoken}&logid={logid}&clienttype=0″
# 通过ddddocr库识别验证码
ocr = ddddocr.DdddOcr()
# 验证码图片
captcha_img = requests.get(request.json()[‘img’], headers=baidu_headers).content
# 验证码ocr
captcha = ocr.classification(captcha_img)
print(captcha)
# 定义data
data = {
“method”: “add_task”,
“app_id”: “250528”,
“source_url”: img_url,
“save_path”: f”/图片/{str(beijing_now.date())}”, # 网盘目录
“input”: captcha,
“vcode”: request.json()[‘vcode’]
}
# 发送请求
r = requests.post(download_offline_url, headers=baidu_headers, data=data)
print(r.json())
if ‘vcode’ in r.json():
time.sleep(3)
print(‘重新验证’)
download_offline_captcha(logid, bdstoken, img_url, request)
# 判断是否下载成功
def is_download_success(logid, img_name):
time.sleep(2)
# 拼接判断是否下载成功的url
url = f”https://pan.baidu.com/api/list?clienttype=0&app_id=250528&web=1&dp-logid={logid}&order=time&desc=1&dir=/图片/{str(beijing_now.date())}&num=100&page=1″
# 发送请求
r = requests.get(url, headers=baidu_headers)
# 判断是否下载成功
if len(r.json().get(“list”)) == 0:
print(“未下载完成”)
time.sleep(2)
return is_download_success(logid, img_name)
for ii in r.json().get(‘list’):
if img_name in ii.get(‘path’):
print(‘今天图片下载完成’)
return False
for i in r.json().get(‘list’):
if “th.jpg” in i.get(‘path’):
print(‘下载完成’)
return True
else:
print(i)
print(“有其他图片,继续下载”)
return is_download_success(logid, img_name)
def rechristen(logid, bdstoken, img_name):
# 拼接改名url
name_url = f”https://pan.baidu.com/api/filemanager?opera=rename&async=2&onnest=fail&channel=chunlei&web=1&app_id=250528&bdstoken={bdstoken}&logid={logid}&clienttype=0″
# 定义data
data = {
“filelist”: ‘[{“path”:”‘ + f”/图片/{str(beijing_now.date())}” + ‘/th.jpg”,”newname”:”‘ + img_name + ‘”}]’
}
# 发送请求
requests.post(name_url, data, headers=baidu_headers)
print(“改名完成”)
if __name__ == ‘__main__’:
img_url, img_name = get_bing_pic()
if is_login():
logid, bdstoken = get_logid_bdstoken()
if is_dir_exist(logid, bdstoken) and is_file_exist(logid, img_name):
download_offline(logid, bdstoken, img_url)
if is_download_success(logid, img_name):
rechristen(logid, bdstoken, img_name)
相关文章
