refactor: use aiohttp for requests

This commit is contained in:
Ryan 2025-01-22 14:49:29 -05:00
parent 640e750e3d
commit 8ef4c85bd8
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
3 changed files with 15 additions and 10 deletions

View File

@ -1,7 +1,7 @@
import re
import aiohttp
import disnake
import requests
import arguments
import commands
@ -31,8 +31,9 @@ async def lookup(message):
return
if args.application:
response = requests.get(
f"https://discord.com/api/v9/applications/{args.id}/rpc"
session = aiohttp.ClientSession()
response = await (
await session.get(f"https://discord.com/api/v9/applications/{args.id}/rpc")
).json()
if "code" in response.keys():
await utils.reply(message, "application not found!")

View File

@ -115,7 +115,9 @@ async def fast_forward(message):
seconds = args.seconds
if args.sponsorblock:
video = sponsorblock.get_segments(players[message.guild.id].current.player.id)
video = await sponsorblock.get_segments(
players[message.guild.id].current.player.id
)
if not video:
await utils.reply(
message, "no sponsorblock segments were found for this video!"

View File

@ -1,21 +1,23 @@
import hashlib
import requests
import aiohttp
from state import sponsorblock_cache
def get_segments(videoId: str):
async def get_segments(videoId: str):
if videoId in sponsorblock_cache:
return sponsorblock_cache[videoId]
hashPrefix = hashlib.sha256(videoId.encode()).hexdigest()[:4]
response = requests.get(
session = aiohttp.ClientSession()
response = await session.get(
f'https://sponsor.ajay.app/api/skipSegments/{hashPrefix}?categories=["music_offtopic"]'
)
print(response.text)
if response.status_code == 200 and (
results := list(filter(lambda v: videoId == v["videoID"], response.json()))
if response.status == 200 and (
results := list(
filter(lambda v: videoId == v["videoID"], await response.json())
)
):
sponsorblock_cache[videoId] = results[0]
return results[0]