refactor: use aiohttp for requests
This commit is contained in:
parent
640e750e3d
commit
8ef4c85bd8
@ -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!")
|
||||
|
@ -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!"
|
||||
|
@ -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]
|
||||
|
Loading…
x
Reference in New Issue
Block a user