mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
84 lines
3.5 KiB
Python
84 lines
3.5 KiB
Python
import aiohttp
|
||
from utils.utils import get_local_proxy, get_bot
|
||
from utils.user_agent import get_user_agent
|
||
import asyncio
|
||
import platform
|
||
import aiofiles
|
||
from bs4 import BeautifulSoup
|
||
import os
|
||
|
||
if platform.system() == "Windows":
|
||
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
||
|
||
|
||
url = "https://github.com/Mrs4s/go-cqhttp/releases"
|
||
|
||
|
||
async def download_gocq_lasted(path: str):
|
||
async with aiohttp.ClientSession(headers=get_user_agent()) as session:
|
||
async with session.get(url, proxy=get_local_proxy()) as response:
|
||
soup = BeautifulSoup(await response.text(), "lxml")
|
||
a = soup.find("div", {"class": "release-header"}).find("a")
|
||
title = a.text
|
||
_url = a.get("href")
|
||
for file in os.listdir(path):
|
||
if file.endswith(".zip"):
|
||
if (
|
||
file == title + "-windows-amd64.zip"
|
||
or file == title + "_windows_amd64.zip"
|
||
):
|
||
return "gocqhttp没有更新！"
|
||
for file in os.listdir(path):
|
||
os.remove(path + file)
|
||
async with session.get(
|
||
"https://github.com" + _url, proxy=get_local_proxy()
|
||
) as res:
|
||
update_info = ""
|
||
soup = BeautifulSoup(await res.text(), "lxml")
|
||
info_div = soup.find("div", {"class": "markdown-body"})
|
||
for p in info_div.find_all("p"):
|
||
update_info += p.text.replace("<br>", "\n") + "\n"
|
||
div_all = soup.select(
|
||
"div.d-flex.flex-justify-between.flex-items-center.py-1.py-md-2.Box-body.px-2"
|
||
)
|
||
for div in div_all:
|
||
if (
|
||
div.find("a").find("span").text == title + "-windows-amd64.zip"
|
||
or div.find("a").find("span").text
|
||
== title + "-linux-arm64.tar.gz"
|
||
or div.find("a").find("span").text
|
||
== "go-cqhttp_windows_amd64.zip"
|
||
or div.find("a").find("span").text
|
||
== "go-cqhttp_linux_arm64.tar.gz"
|
||
):
|
||
file_url = div.find("a").get("href")
|
||
if div.find("a").find("span").text.find("windows") == -1:
|
||
tag = "-linux-arm64.tar.gz"
|
||
else:
|
||
tag = "-windows-amd64.zip"
|
||
async with session.get(
|
||
"https://github.com" + file_url, proxy=get_local_proxy()
|
||
) as res_file:
|
||
async with aiofiles.open(path + title + tag, "wb") as f:
|
||
await f.write(await res_file.read())
|
||
return update_info
|
||
|
||
|
||
async def upload_gocq_lasted(path, name, group_id):
|
||
bot = get_bot()
|
||
folder_id = 0
|
||
for folder in (await bot.get_group_root_files(group_id=group_id))["folders"]:
|
||
if folder["folder_name"] == "gocq":
|
||
folder_id = folder["folder_id"]
|
||
if not folder_id:
|
||
await bot.send_group_msg(group_id=group_id, message=f"请创建gocq文件夹后重试！")
|
||
for file in os.listdir(path):
|
||
os.remove(path + file)
|
||
else:
|
||
await bot.upload_group_file(
|
||
group_id=group_id, folder=folder_id, file=path + name, name=name
|
||
)
|
||
|
||
|
||
# asyncio.get_event_loop().run_until_complete(download_gocq_lasted())
|