1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-18 06:55:09 +00:00

Add close method

This commit is contained in:
VarMonke 2022-05-18 23:46:12 +05:30
parent ff98977edb
commit 33309288a2
2 changed files with 7 additions and 2 deletions

View file

@ -372,7 +372,10 @@ class GHClient:
""":class:`float`: Returns the latency of the client."""
return await self.http.latency()
async def close(self) -> None:
"""Close the session."""
await self.http.session.close()
class Client(GHClient):
pass

View file

@ -18,6 +18,7 @@ from .exceptions import *
from .exceptions import GistNotFound, RepositoryAlreadyExists, MissingPermissions
from .exceptions import FileAlreadyExists
from .exceptions import ResourceAlreadyExists
from .exceptions import Ratelimited
from .objects import User, Gist, Repository, File, bytes_to_b64
from .urls import *
from . import __version__
@ -45,7 +46,8 @@ async def on_req_start(
) -> None:
"""Before-request hook to make sure we don't overrun the ratelimit."""
# print(repr(session), repr(ctx), repr(params))
pass
if session._rates.remaining in ('0', '1'): #type: ignore
raise Exception('Ratelimit exceeded')
async def on_req_end(session: aiohttp.ClientSession, ctx: SimpleNamespace, params: aiohttp.TraceRequestEndParams) -> None: