mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
74 lines
3.3 KiB
Python
74 lines
3.3 KiB
Python
import aiohttp
|
||
from util.utils import get_local_proxy, get_bot
|
||
from util.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())
|
||
|