mirror of
				https://github.com/RGBCube/GitHubWrapper
				synced 2025-10-31 14:02:46 +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
				
			
		|  | @ -9,7 +9,6 @@ import re | ||||||
| 
 | 
 | ||||||
| from .exceptions import * | from .exceptions import * | ||||||
| from .objects import * | from .objects import * | ||||||
| from .objects import APIOBJECT |  | ||||||
| from .urls import * | from .urls import * | ||||||
| 
 | 
 | ||||||
| LINK_PARSING_RE = re.compile(r"<(\S+(\S))>; rel=\"(\S+)\"") | LINK_PARSING_RE = re.compile(r"<(\S+(\S))>; rel=\"(\S+)\"") | ||||||
|  |  | ||||||
|  | @ -9,7 +9,7 @@ import aiohttp | ||||||
| 
 | 
 | ||||||
| from . import http | from . import http | ||||||
| from .exceptions import AlreadyStarted, NotStarted | from .exceptions import AlreadyStarted, NotStarted | ||||||
| from .objects import User | from .objects import User, Repository | ||||||
| 
 | 
 | ||||||
| class Github: | class Github: | ||||||
|     _auth = None |     _auth = None | ||||||
|  | @ -61,7 +61,7 @@ class Github: | ||||||
|         """Fetch a Github user from their username.""" |         """Fetch a Github user from their username.""" | ||||||
|         return User(await http.get_user(self.session, username), self.session) |         return User(await http.get_user(self.session, username), self.session) | ||||||
| 
 | 
 | ||||||
|     async def get_repo(self, repo_name: str) -> 'Repo': |     async def get_repo(self, repo_name: str) -> Repository: | ||||||
|         """Fetch a Github repository from it's name.""" |         """Fetch a Github repository from it's name.""" | ||||||
|         pass |         pass | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
							
								
								
									
										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}, >' | ||||||
|  | @ -1,37 +1,15 @@ | ||||||
| #== objects.py ==# | #== user.py ==# | ||||||
| 
 |  | ||||||
| from datetime import datetime |  | ||||||
| 
 | 
 | ||||||
| import aiohttp | import aiohttp | ||||||
| 
 | 
 | ||||||
| from . import http | from .objects import APIOBJECT, dt_formatter | ||||||
|  | from .. import http | ||||||
| 
 | 
 | ||||||
| __all__ = ( | __all__ = ( | ||||||
|     'User' |     'User', | ||||||
|  |     'PartialUser' | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| 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__}>' |  | ||||||
| 
 |  | ||||||
| class _BaseUser(APIOBJECT): | class _BaseUser(APIOBJECT): | ||||||
|     __slots__ = ( |     __slots__ = ( | ||||||
|         'login', |         'login', | ||||||
|  | @ -72,7 +50,7 @@ class User(_BaseUser): | ||||||
|             setattr(self, key, value) |             setattr(self, key, value) | ||||||
| 
 | 
 | ||||||
|     def __repr__(self): |     def __repr__(self): | ||||||
|         return f'<User; login: {self.login}, id: {self.id}, created_at: {repr_dt(self.created_at)}>' |         return f'<User; login: {self.login}, id: {self.id}, created_at: {dt_formatter(self.created_at)}>' | ||||||
| 
 | 
 | ||||||
|     @classmethod |     @classmethod | ||||||
|     async def get_user(cls, session: aiohttp.ClientSession, username: str) -> 'User': |     async def get_user(cls, session: aiohttp.ClientSession, username: str) -> 'User': | ||||||
|  | @ -95,49 +73,9 @@ class PartialUser(_BaseUser): | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
|     def __repr__(self): |     def __repr__(self): | ||||||
|         return f'<PartialUser; login: {self.login}, id: {self.id}, site_admin: {self.site_admin}, html_url: {self.html_url}, created_at: {repr_dt(self.created_at)}>' |         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): |     async def _get_user(self): | ||||||
|         """Upgrades the PartialUser to a User object."""  |         """Upgrades the PartialUser to a User object."""  | ||||||
|         response = await http.get_user(self.session, self.login) |         response = await http.get_user(self.session, self.login) | ||||||
|         return User(response, self.session) |         return User(response, self.session) | ||||||
| 
 |  | ||||||
| 
 |  | ||||||
| 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__ |  | ||||||
|         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: {repr_dt(self.created_at)}, default_branch: {self.default_branch}, license: {self.license}, >' |  | ||||||
| 
 |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 sudosnok
						sudosnok