mirror of
https://github.com/RGBCube/GitHubWrapper
synced 2025-05-18 15:05:08 +00:00
Added more functionality and allowed a smoother entry point to the program
This commit is contained in:
parent
e4aedcce97
commit
3f0326a19e
1 changed files with 19 additions and 1 deletions
|
@ -8,11 +8,12 @@ from getpass import getpass
|
||||||
import aiohttp
|
import aiohttp
|
||||||
|
|
||||||
from . import http
|
from . import http
|
||||||
|
from .exceptions import AlreadyStarted, NotStarted
|
||||||
from .objects import User
|
from .objects import User
|
||||||
|
|
||||||
class Github:
|
class Github:
|
||||||
_auth = None
|
_auth = None
|
||||||
session = aiohttp.ClientSession
|
has_started = False
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
*,
|
*,
|
||||||
|
@ -26,9 +27,22 @@ class Github:
|
||||||
auth_token = getpass.getpass('Enter your token: ')
|
auth_token = getpass.getpass('Enter your token: ')
|
||||||
self._auth = aiohttp.BasicAuth(username, auth_token)
|
self._auth = aiohttp.BasicAuth(username, auth_token)
|
||||||
|
|
||||||
|
def __await__(self):
|
||||||
|
return self.start().__await__()
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f'<Github Client; has_auth={bool(self._auth)}>'
|
return f'<Github Client; has_auth={bool(self._auth)}>'
|
||||||
|
|
||||||
|
def check_limits(self, as_dict: bool = False) -> dict[str, str | int] | list[str]:
|
||||||
|
if not self.has_started:
|
||||||
|
raise NotStarted
|
||||||
|
if not as_dict:
|
||||||
|
output = []
|
||||||
|
for key, value in self.session._rates._asdict().items():
|
||||||
|
output.append(f'{key} : {value}')
|
||||||
|
return output
|
||||||
|
return self.session._rates._asdict()
|
||||||
|
|
||||||
def update_auth(self) -> None:
|
def update_auth(self) -> None:
|
||||||
"""Allows you to input auth information after instantiating the client."""
|
"""Allows you to input auth information after instantiating the client."""
|
||||||
username = input('Enter your username: ')
|
username = input('Enter your username: ')
|
||||||
|
@ -37,7 +51,11 @@ class Github:
|
||||||
|
|
||||||
async def start(self) -> None:
|
async def start(self) -> None:
|
||||||
"""Main entry point to the wrapper, this creates the ClientSession."""
|
"""Main entry point to the wrapper, this creates the ClientSession."""
|
||||||
|
if self.has_started:
|
||||||
|
raise AlreadyStarted
|
||||||
self.session = await http.make_session(headers=self._headers, authorization=self._auth)
|
self.session = await http.make_session(headers=self._headers, authorization=self._auth)
|
||||||
|
self.has_started = True
|
||||||
|
return self
|
||||||
|
|
||||||
async def get_user(self, username: str) -> User:
|
async def get_user(self, username: str) -> User:
|
||||||
"""Fetch a Github user from their username."""
|
"""Fetch a Github user from their username."""
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue