mirror of
				https://github.com/RGBCube/GitHubWrapper
				synced 2025-10-31 14:02:46 +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
	
	 sudosnok
						sudosnok