mirror of
				https://github.com/RGBCube/GitHubWrapper
				synced 2025-10-31 05:52:45 +00:00 
			
		
		
		
	Moved objects into their own place in a folder for readability
This commit is contained in:
		
							parent
							
								
									9a2889702b
								
							
						
					
					
						commit
						d5ba5bc58d
					
				
					 6 changed files with 93 additions and 73 deletions
				
			
		
							
								
								
									
										5
									
								
								Github/objects/__init__.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								Github/objects/__init__.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,5 @@ | |||
| #== __init__.py ==# | ||||
| 
 | ||||
| from .objects import * | ||||
| from .user import * | ||||
| from .repo import * | ||||
							
								
								
									
										29
									
								
								Github/objects/objects.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								Github/objects/objects.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,29 @@ | |||
| #== objects.py ==# | ||||
| 
 | ||||
| from datetime import datetime | ||||
| import aiohttp | ||||
| 
 | ||||
| __all__ = ( | ||||
|     'APIOBJECT', | ||||
| ) | ||||
| 
 | ||||
| def dt_formatter(time_str): | ||||
|     if time_str is not None: | ||||
|         return datetime.strptime(time_str, r"%Y-%m-%dT%H:%M:%SZ") | ||||
|     return None | ||||
| 
 | ||||
| def repr_dt(time_str): | ||||
|     return time_str.strftime(r'%d-%m-%Y, %H:%M:%S') | ||||
| 
 | ||||
| class APIOBJECT: | ||||
|     __slots__ = ( | ||||
|         '_response', | ||||
|         'session' | ||||
|     ) | ||||
| 
 | ||||
|     def __init__(self, response: dict[str, str | int], session: aiohttp.ClientSession) -> None: | ||||
|         self._response = response | ||||
|         self.session = session | ||||
| 
 | ||||
|     def __repr__(self) -> str: | ||||
|         return f'<{self.__class__.__name__}>' | ||||
							
								
								
									
										49
									
								
								Github/objects/repo.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								Github/objects/repo.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,49 @@ | |||
| #== repo.py ==# | ||||
| 
 | ||||
| import aiohttp | ||||
| 
 | ||||
| from .objects import APIOBJECT, dt_formatter | ||||
| from . import PartialUser | ||||
| from .. import http | ||||
| 
 | ||||
| __all__ = ( | ||||
|     'Repository', | ||||
| ) | ||||
| 
 | ||||
| class Repository(APIOBJECT): | ||||
|     __slots__ = ( | ||||
|         'id', | ||||
|         'name', | ||||
|         'owner', | ||||
|         'size' | ||||
|         'created_at', | ||||
|         'url', | ||||
|         'html_url', | ||||
|         'archived', | ||||
|         'disabled', | ||||
|         'updated_at', | ||||
|         'open_issues_count', | ||||
|         'default_branch', | ||||
|         'clone_url', | ||||
|         'stargazers_count', | ||||
|         'watchers_count', | ||||
|         'forks', | ||||
|         'license' | ||||
|         ) | ||||
|     def __init__(self, response: dict, session: aiohttp.ClientSession) -> None: | ||||
|         super().__init__(response, session) | ||||
|         tmp = self.__slots__ + APIOBJECT.__slots__ | ||||
|         keys = {key: value for key,value in self.response.items() if key in tmp} | ||||
|         for key, value in key.items(): | ||||
|             if key == 'owner': | ||||
|                 self.owner = PartialUser(value, self.session) | ||||
|                 return | ||||
| 
 | ||||
|             if '_at' in key and value is not None: | ||||
|                 setattr(self, key, dt_formatter(value)) | ||||
|                 return | ||||
| 
 | ||||
|             setattr(self, key, value) | ||||
| 
 | ||||
|     def __repr__(self) -> str: | ||||
|         return f'<Repository; id: {self.id}, name: {self.name}, owner: {self.owner}, created_at: {dt_formatter(self.created_at)}, default_branch: {self.default_branch}, license: {self.license}, >' | ||||
							
								
								
									
										81
									
								
								Github/objects/user.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										81
									
								
								Github/objects/user.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,81 @@ | |||
| #== user.py ==# | ||||
| 
 | ||||
| import aiohttp | ||||
| 
 | ||||
| from .objects import APIOBJECT, dt_formatter | ||||
| from .. import http | ||||
| 
 | ||||
| __all__ = ( | ||||
|     'User', | ||||
|     'PartialUser' | ||||
| ) | ||||
| 
 | ||||
| class _BaseUser(APIOBJECT): | ||||
|     __slots__ = ( | ||||
|         'login', | ||||
|         'id', | ||||
|         ) | ||||
|     def __init__(self, response: dict, session: aiohttp.ClientSession) -> None: | ||||
|         super().__init__(response, session) | ||||
|         self.login = response.get('login') | ||||
|         self.id = response.get('id') | ||||
| 
 | ||||
|     def __repr__(self) -> str: | ||||
|         return f'<{self.__class__.__name__}; id = {self.id}, login = {self.login}>' | ||||
| 
 | ||||
| class User(_BaseUser):  | ||||
|     __slots__ = ( | ||||
|         'login', | ||||
|         'id', | ||||
|         'avatar_url', | ||||
|         'html_url', | ||||
|         'public_repos', | ||||
|         'public_gists', | ||||
|         'followers', | ||||
|         'following', | ||||
|         'created_at', | ||||
|         ) | ||||
|     def __init__(self, response: dict, session: aiohttp.ClientSession) -> None: | ||||
|         super().__init__(response, session) | ||||
|         tmp = self.__slots__ + _BaseUser.__slots__ | ||||
|         keys = {key: value for key,value in self._response.items() if key in tmp} | ||||
|         for key, value in keys.items(): | ||||
|             if '_at' in key and value is not None: | ||||
|                 setattr(self, key, dt_formatter(value)) | ||||
|                 continue | ||||
|             else: | ||||
|                 setattr(self, key, value) | ||||
|                 continue | ||||
| 
 | ||||
|             setattr(self, key, value) | ||||
| 
 | ||||
|     def __repr__(self): | ||||
|         return f'<User; login: {self.login}, id: {self.id}, created_at: {dt_formatter(self.created_at)}>' | ||||
| 
 | ||||
|     @classmethod | ||||
|     async def get_user(cls, session: aiohttp.ClientSession, username: str) -> 'User': | ||||
|         """Returns a User object from the username, with the mentions slots.""" | ||||
|         response = await http.get_user(session, username) | ||||
|         return User(response, session) | ||||
| 
 | ||||
| class PartialUser(_BaseUser): | ||||
|     __slots__ = ( | ||||
|         'site_admin', | ||||
|         'html_url', | ||||
|         'created_at', | ||||
|         ) + _BaseUser.__slots__ | ||||
| 
 | ||||
|     def __init__(self, response: dict, session: aiohttp.ClientSession) -> None: | ||||
|         super().__init__(response, session) | ||||
|         self.site_admin = response.get('site_admin') | ||||
|         self.html_url = response.get('html_url') | ||||
|         self.avatar_url = response.get('avatar_url') | ||||
| 
 | ||||
| 
 | ||||
|     def __repr__(self): | ||||
|         return f'<PartialUser; login: {self.login}, id: {self.id}, site_admin: {self.site_admin}, html_url: {self.html_url}, created_at: {dt_formatter(self.created_at)}>' | ||||
| 
 | ||||
|     async def _get_user(self): | ||||
|         """Upgrades the PartialUser to a User object."""  | ||||
|         response = await http.get_user(self.session, self.login) | ||||
|         return User(response, self.session) | ||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 sudosnok
						sudosnok