mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
77 lines
3.4 KiB
Python
77 lines
3.4 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':
|
|||
|
|
path = os.getcwd() + '/resources/gocqhttp_file/'
|
|||
|
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
|||
|
|
else:
|
|||
|
|
path = r'/home/hibiki/hibikibot/resources/gocqhttp_file/'
|
|||
|
|
|
|||
|
|
|
|||
|
|
url = 'https://github.com/Mrs4s/go-cqhttp/releases'
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def download_gocq_lasted():
|
|||
|
|
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(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())
|
|||
|
|
|