xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

disable ssl check if debug mode is enabled

Browse source
This commit is contained in:
yzhh 2023-02-08 14:25:35 +08:00
parent dfc61032c2
commit b9556afc09
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
core/utils/http.py
View file

@ -7,6 +7,7 @@ from typing import Union
import aiohttp
import filetype as ft
from aiofile import async_open
from aiohttp import TCPConnector
from tenacity import retry, wait_fixed, stop_after_attempt
from config import Config
@ -15,6 +16,8 @@ from .cache import random_cache_path
from ..exceptions import NoReportException
logging_resp = False
debug = Config('debug')
proxy = ''
def private_ip_check(url: str):
@ -53,9 +56,11 @@ async def get_url(url: str, status_code: int = False, headers: dict = None, fmt=
if not Config('allow_request_private_ip') and not request_private_ip:
private_ip_check(url)
async with aiohttp.ClientSession(headers=headers) as session:
async with aiohttp.ClientSession(headers=headers,
connector=TCPConnector(verify_ssl=False) if debug else None, ) as session:
try:
async with session.get(url, timeout=aiohttp.ClientTimeout(total=timeout), headers=headers) as req:
async with session.get(url, timeout=aiohttp.ClientTimeout(total=timeout), headers=headers,
proxy=proxy) as req:
Logger.debug(f'[{req.status}] {url}')
if logging_resp:
Logger.debug(await req.read())
@ -98,10 +103,12 @@ async def post_url(url: str, data: any = None, status_code: int = False, headers
if not Config('allow_request_private_ip') and not request_private_ip:
private_ip_check(url)
async with aiohttp.ClientSession(headers=headers) as session:
async with aiohttp.ClientSession(headers=headers,
connector=TCPConnector(verify_ssl=False) if debug else None, ) as session:
try:
async with session.post(url, data=data, headers=headers,
timeout=aiohttp.ClientTimeout(total=timeout)) as req:
timeout=aiohttp.ClientTimeout(total=timeout),
proxy=proxy) as req:
Logger.debug(f'[{req.status}] {url}')
if logging_resp:
Logger.debug(await req.read())

2
modules/summary/__init__.py
View file

@ -43,6 +43,6 @@ async def _(msg: Bot.MessageSession):
output += remove_suffix(await post_url('https://chat-simplifier.imzbb.cc/api/generate', data=json.dumps(
{'prompt': f'''{prompt}<|chat_start|>
{post_texts}
<|chat_end|>'''}), headers={'Content-Type': 'application/json'}, timeout=9999999), '<|im_end|>')
<|chat_end|>'''}), headers={'Content-Type': 'application/json'}, status_code=200, timeout=9999999), '<|im_end|>')
await wait_msg.delete()
await msg.finish(output, disable_secret_check=True)