yawaflua/Flask-Discord
1
0
Fork 0
mirror of https://github.com/yawaflua/Flask-Discord.git synced 2025-12-09 20:09:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

🔧 Fix unclosed OAuth2Session objects. Make use of the _make_session context manager.

Browse Source
This commit is contained in:
thec0sm0s
2020-10-04 21:14:58 +05:30
parent 5995e94b46
commit 5de64ebd58
2 changed files with 23 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
quart_discord/_http.py
View File

@@ -64,12 +64,12 @@ class DiscordOAuth2HttpClient(abc.ABC):
raise NotImplementedError
async def _fetch_token(self, state):
discord = await self._make_session(state=state)
return await discord.fetch_token(
configs.DISCORD_TOKEN_URL,
client_secret=self.__client_secret,
authorization_response=request.url
)
async with await self._make_session(state=state) as discord:
return await discord.fetch_token(
configs.DISCORD_TOKEN_URL,
client_secret=self.__client_secret,
authorization_response=request.url
)
async def _make_session(self, token: str = None, state: str = None, scope: list = None) -> OAuth2Session:
"""A low level method used for creating OAuth2 session.
@@ -136,20 +136,21 @@ class DiscordOAuth2HttpClient(abc.ABC):
"""
route = configs.DISCORD_API_BASE_URL + route
discord = await self._make_session()
async with (await discord.request(
method, route, data, **kwargs
) if oauth else aiohttp.request(method, route, data=data, **kwargs)) as response:
async with await self._make_session() as discord:
async with (
await discord.request(method, route, data, **kwargs)
if oauth else aiohttp.request(method, route, data=data, **kwargs)
) as response:
if response.status == 401:
raise exceptions.Unauthorized
if response.status == 429:
raise exceptions.RateLimited(response)
if response.status == 401:
raise exceptions.Unauthorized
if response.status == 429:
raise exceptions.RateLimited(response)
try:
return await response.json()
except aiohttp.ContentTypeError:
return await response.text()
try:
return await response.json()
except aiohttp.ContentTypeError:
return await response.text()
async def bot_request(self, route: str, method="GET", **kwargs) -> typing.Union[dict, str]:
"""Make HTTP request to specified endpoint with bot token as authorization headers.

7
quart_discord/client.py
View File

@@ -95,8 +95,8 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
state = jwt.encode(data or dict(), current_app.config["SECRET_KEY"]).decode(encoding="utf-8")
discord_session = await self._make_session(scope=scope, state=state)
authorization_url, state = discord_session.authorization_url(configs.DISCORD_AUTHORIZATION_BASE_URL)
async with await self._make_session(scope=scope, state=state) as discord_:
authorization_url, state = discord_.authorization_url(configs.DISCORD_AUTHORIZATION_BASE_URL)
self.__save_state(state)
@@ -174,7 +174,8 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
async def authorized(self):
"""A boolean indicating whether current session has authorization token or not."""
return (await self._make_session()).authorized
async with await self._make_session() as discord_:
return discord_.authorized
@staticmethod
async def fetch_user() -> models.User: