1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-17 14:35:09 +00:00

Adjustments and an init

This commit is contained in:
sudosnok 2022-03-26 21:35:52 +00:00
parent 05d522eb1e
commit b4045c70bd

View file

@ -4,12 +4,32 @@ __all__ = (
'Github',
)
from getpass import getpass
import aiohttp
from . import http
from .objects import User
class Github:
_auth = None
_state = aiohttp.ClientSession
def __init__(
self,
*,
using_auth: bool = False,
custom_headers: dict[str, str | int] = {}
):
"""The main client, used to start most use-cases."""
self._headers = custom_headers
if using_auth:
username = input('Enter your username: ') #these two lines are blocking, but it's probably not a problem
auth_token = getpass.getpass('Enter your token: ')
self._auth = aiohttp.BasicAuth(username, auth_token)
async def start(self):
"""Main entry point to the wrapper, this creates the ClientSession."""
headers = {}
self._session = http.make_session(headers=headers)
self._state = await http.make_session(headers=self._headers, authorization=self._auth)
async def get_user(self, username: str):
"""Fetch a Github user from their username."""
return User(await http.get_user(self._state, username), self._state)