PagerMaid_Plugins/bingwall.py
2022-01-18 16:47:20 +08:00

61 lines
1.9 KiB
Python

import json
import random
from requests import get
from pagermaid import version
from pagermaid.listener import listener
from pagermaid.utils import alias_command
from os import remove
def get_url(num):
json_url = f"https://www.bing.com/HPImageArchive.aspx?format=js&mkt=zh-CN&n=1&idx={str(num)}"
req = get(json_url)
url = " "
copyright = " "
if req.status_code == 200:
data = json.loads(req.text)
url = data['images'][0]['url']
copyright = data['images'][0]['copyright']
return url, copyright
@listener(is_plugin=True, outgoing=True, command=alias_command("bingwall"),
description="获取Bing每日壁纸")
async def bingwall(context):
await context.edit("获取壁纸中 . . .")
status = False
for _ in range(20): # 最多重试20次
website = random.randint(0, 0)
num = random.randint(1, 7)
url, copyright = get_url(num)
image_url = f"https://www.bing.com{url}"
filename = "wallpaper" + str(random.random())[2:] + ".jpg"
try:
if website == 0 and image_url != " ":
img = get(image_url)
else:
continue
if img.status_code == 200:
with open(filename, 'wb') as f:
f.write(img.content)
await context.edit("上传中 . . .")
await context.client.send_file(context.chat_id, filename, caption=f"#bing wallpaper\n{str(copyright)}")
status = True
break # 成功了就赶紧结束啦!
except:
try:
remove(filename)
except:
pass
continue
try:
remove(filename)
except:
pass
try:
await context.delete()
except:
pass
if not status:
await context.client.send_message(context.chat_id, "出错了呜呜呜 ~ 试了好多好多次都无法访问到服务器 。")