mirror of
				https://github.com/RGBCube/GitHubWrapper
				synced 2025-10-30 21:42:45 +00:00 
			
		
		
		
	Massive shift to avoid circular imports, functionally the same, though objects are forced to be in one file to avoid the circular nature
This commit is contained in:
		
							parent
							
								
									e0623fe367
								
							
						
					
					
						commit
						8b91010b12
					
				
					 10 changed files with 326 additions and 335 deletions
				
			
		|  | @ -1,5 +1,7 @@ | ||||||
| #== cache.py ==# | #== cache.py ==# | ||||||
| 
 | 
 | ||||||
|  | from __future__ import annotations | ||||||
|  | 
 | ||||||
| __all__ = ( | __all__ = ( | ||||||
|     'UserCache', |     'UserCache', | ||||||
|     'RepoCache', |     'RepoCache', | ||||||
|  | @ -7,7 +9,8 @@ __all__ = ( | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| from collections import deque | from collections import deque | ||||||
| from .objects import APIOBJECT, User, Repository, Organization | 
 | ||||||
|  | from .objects import APIObject, User, Repository, Organization | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| class _BaseCache(dict): | class _BaseCache(dict): | ||||||
|  | @ -19,12 +22,12 @@ class _BaseCache(dict): | ||||||
|         self._lru_keys = deque(maxlen=self._max_size) |         self._lru_keys = deque(maxlen=self._max_size) | ||||||
|         super().__init__(args) |         super().__init__(args) | ||||||
|      |      | ||||||
|     def __getitem__(self, __k: str) -> APIOBJECT: |     def __getitem__(self, __k: str) -> APIObject: | ||||||
|         target = self._lru_keys.pop(self._lru_keys.index(__k)) |         target = self._lru_keys.pop(self._lru_keys.index(__k)) | ||||||
|         self._lru_keys.appendleft(target) |         self._lru_keys.appendleft(target) | ||||||
|         return super().__getitem__(__k) |         return super().__getitem__(__k) | ||||||
| 
 | 
 | ||||||
|     def __setitem__(self, __k: str, __v: APIOBJECT) -> None: |     def __setitem__(self, __k: str, __v: APIObject) -> None: | ||||||
|         if len(self) == self._max_size: |         if len(self) == self._max_size: | ||||||
|             to_pop = self._lru_keys.pop(-1) |             to_pop = self._lru_keys.pop(-1) | ||||||
|             del self[to_pop] |             del self[to_pop] | ||||||
|  | @ -37,12 +40,12 @@ class _BaseCache(dict): | ||||||
| 
 | 
 | ||||||
| class UserCache(_BaseCache): | class UserCache(_BaseCache): | ||||||
|     """This adjusts the typehints to reflect User objects""" |     """This adjusts the typehints to reflect User objects""" | ||||||
|     def __getitem__(self, __k: str) -> User: |     def __getitem__(self, __k: str) -> 'User': | ||||||
|         target = self._lru_keys.pop(self._lru_keys.index(__k)) |         target = self._lru_keys.pop(self._lru_keys.index(__k)) | ||||||
|         self._lru_keys.appendleft(target) |         self._lru_keys.appendleft(target) | ||||||
|         return super().__getitem__(__k) |         return super().__getitem__(__k) | ||||||
| 
 | 
 | ||||||
|     def __setitem__(self, __k: str, __v: User) -> None: |     def __setitem__(self, __k: str, __v: 'User') -> None: | ||||||
|         if len(self) == self._max_size: |         if len(self) == self._max_size: | ||||||
|             to_pop = self._lru_keys.pop(-1) |             to_pop = self._lru_keys.pop(-1) | ||||||
|             del self[to_pop] |             del self[to_pop] | ||||||
|  | @ -55,12 +58,12 @@ class UserCache(_BaseCache): | ||||||
| 
 | 
 | ||||||
| class RepoCache(_BaseCache): | class RepoCache(_BaseCache): | ||||||
|     """This adjusts the typehints to reflect Repo objects""" |     """This adjusts the typehints to reflect Repo objects""" | ||||||
|     def __getitem__(self, __k: str) -> Repository: |     def __getitem__(self, __k: str) -> 'Repository': | ||||||
|         target = self._lru_keys.pop(self._lru_keys.index(__k)) |         target = self._lru_keys.pop(self._lru_keys.index(__k)) | ||||||
|         self._lru_keys.appendleft(target) |         self._lru_keys.appendleft(target) | ||||||
|         return super().__getitem__(__k) |         return super().__getitem__(__k) | ||||||
| 
 | 
 | ||||||
|     def __setitem__(self, __k: str, __v: Repository) -> None: |     def __setitem__(self, __k: str, __v: 'Repository') -> None: | ||||||
|         if len(self) == self._max_size: |         if len(self) == self._max_size: | ||||||
|             to_pop = self._lru_keys.pop(-1) |             to_pop = self._lru_keys.pop(-1) | ||||||
|             del self[to_pop] |             del self[to_pop] | ||||||
|  | @ -72,12 +75,12 @@ class RepoCache(_BaseCache): | ||||||
|             self[key] = value |             self[key] = value | ||||||
| 
 | 
 | ||||||
| class OrgCache(_BaseCache): | class OrgCache(_BaseCache): | ||||||
|     def __getitem__(self, __k: str) -> Organization: |     def __getitem__(self, __k: str) -> 'Organization': | ||||||
|         target = self._lru_keys.pop(self._lru_keys.index(__k)) |         target = self._lru_keys.pop(self._lru_keys.index(__k)) | ||||||
|         self._lru_keys.appendleft(target) |         self._lru_keys.appendleft(target) | ||||||
|         return super().__getitem__(__k) |         return super().__getitem__(__k) | ||||||
| 
 | 
 | ||||||
|     def __setitem__(self, __k: str, __v: Organization) -> None: |     def __setitem__(self, __k: str, __v: 'Organization') -> None: | ||||||
|         if len(self) == self._max_size: |         if len(self) == self._max_size: | ||||||
|             to_pop = self._lru_keys.pop(-1) |             to_pop = self._lru_keys.pop(-1) | ||||||
|             del self[to_pop] |             del self[to_pop] | ||||||
|  |  | ||||||
|  | @ -1,15 +1,17 @@ | ||||||
| #== http.py ==# | #== http.py ==# | ||||||
| 
 | 
 | ||||||
| import aiohttp | 
 | ||||||
|  | import json | ||||||
|  | import re | ||||||
| from collections import namedtuple | from collections import namedtuple | ||||||
| from datetime import datetime | from datetime import datetime | ||||||
| from types import SimpleNamespace | from types import SimpleNamespace | ||||||
| import re | 
 | ||||||
| import json | import aiohttp | ||||||
| 
 | 
 | ||||||
| from .exceptions import * | from .exceptions import * | ||||||
| from .exceptions import RepositoryAlreadyExists, GistNotFound | from .exceptions import GistNotFound, RepositoryAlreadyExists | ||||||
| from .objects import * | from .objects import APIObject, User | ||||||
| from .urls import * | from .urls import * | ||||||
| 
 | 
 | ||||||
| __all__ = ( | __all__ = ( | ||||||
|  | @ -71,7 +73,7 @@ class Paginator: | ||||||
|         self.session = session |         self.session = session | ||||||
|         self.response = response |         self.response = response | ||||||
|         self.should_paginate = bool(self.response.headers.get('Link', False)) |         self.should_paginate = bool(self.response.headers.get('Link', False)) | ||||||
|         types: dict[str, APIOBJECT] = { |         types: dict[str, APIObject] = { | ||||||
|             'user': User, |             'user': User, | ||||||
|         } |         } | ||||||
|         self.target_type = types[target_type] |         self.target_type = types[target_type] | ||||||
|  | @ -85,11 +87,11 @@ class Paginator: | ||||||
|         """Fetches a specific page and returns the JSON.""" |         """Fetches a specific page and returns the JSON.""" | ||||||
|         return await (await self.session.get(link)).json() |         return await (await self.session.get(link)).json() | ||||||
| 
 | 
 | ||||||
|     async def early_return(self) -> list[APIOBJECT]: |     async def early_return(self) -> list[APIObject]: | ||||||
|         # I don't rightly remember what this does differently, may have a good ol redesign later |         # I don't rightly remember what this does differently, may have a good ol redesign later | ||||||
|         return [self.target_type(data, self.session) for data in await self.response.json()] |         return [self.target_type(data, self.session) for data in await self.response.json()] | ||||||
| 
 | 
 | ||||||
|     async def exhaust(self) -> list[APIOBJECT]: |     async def exhaust(self) -> list[APIObject]: | ||||||
|         """Iterates through all of the pages for the relevant object and creates them.""" |         """Iterates through all of the pages for the relevant object and creates them.""" | ||||||
|         if self.should_paginate: |         if self.should_paginate: | ||||||
|             return await self.early_return() |             return await self.early_return() | ||||||
|  | @ -162,6 +164,27 @@ class http: | ||||||
|             return await result.json() |             return await result.json() | ||||||
|         raise UserNotFound |         raise UserNotFound | ||||||
| 
 | 
 | ||||||
|  |     async def get_user_repos(self, _user: User) -> list[GithubRepoData]: | ||||||
|  |         result = await self.session.get(USER_REPOS_URL.format(_user.login)) | ||||||
|  |         if 200 <= result.status <= 299: | ||||||
|  |             return await result.json() | ||||||
|  |         else: | ||||||
|  |             print('This shouldn\'t be reachable') | ||||||
|  | 
 | ||||||
|  |     async def get_user_gists(self, _user: User) -> list[GithubGistData]: | ||||||
|  |         result = await self.session.get(USER_GISTS_URL.format(_user.login)) | ||||||
|  |         if 200 <= result.status <= 299: | ||||||
|  |             return await result.json() | ||||||
|  |         else: | ||||||
|  |             print('This shouldn\'t be reachable') | ||||||
|  | 
 | ||||||
|  |     async def get_user_orgs(self, _user: User) -> list[GithubOrgData]: | ||||||
|  |         result = await self.session.get(USER_ORGS_URL.format(_user.login)) | ||||||
|  |         if 200 <= result.status <= 299: | ||||||
|  |             return await result.json() | ||||||
|  |         else: | ||||||
|  |             print('This shouldn\'t be reachable') | ||||||
|  | 
 | ||||||
|     async def get_repo(self, owner: str, repo_name: str) -> GithubRepoData: |     async def get_repo(self, owner: str, repo_name: str) -> GithubRepoData: | ||||||
|         """Returns a Repo's raw JSON from the given owner and repo name.""" |         """Returns a Repo's raw JSON from the given owner and repo name.""" | ||||||
|         result = await self.session.get(REPO_URL.format(owner, repo_name)) |         result = await self.session.get(REPO_URL.format(owner, repo_name)) | ||||||
|  |  | ||||||
|  | @ -1,18 +1,20 @@ | ||||||
| #== main.py ==# | #== main.py ==# | ||||||
|  | from __future__ import annotations | ||||||
| 
 | 
 | ||||||
| __all__ = ( | __all__ = ( | ||||||
|     'GHClient', |     'GHClient', | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| import aiohttp |  | ||||||
| import asyncio | import asyncio | ||||||
| import functools | import functools | ||||||
| from getpass import getpass |  | ||||||
| 
 | 
 | ||||||
| from .http import http | import aiohttp | ||||||
|  | 
 | ||||||
| from . import exceptions | from . import exceptions | ||||||
| from .objects import User, PartialUser, Repository, Organization, Issue, Gist | from .cache import RepoCache, UserCache | ||||||
| from .cache import UserCache, RepoCache | from .http import http | ||||||
|  | from .objects import Gist, Issue, Organization, Repository, User | ||||||
|  | 
 | ||||||
| 
 | 
 | ||||||
| class GHClient: | class GHClient: | ||||||
|     _auth = None |     _auth = None | ||||||
|  |  | ||||||
							
								
								
									
										276
									
								
								Github/objects.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										276
									
								
								Github/objects.py
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,276 @@ | ||||||
|  | #== objects.py ==# | ||||||
|  | from __future__ import annotations | ||||||
|  | 
 | ||||||
|  | from datetime import datetime | ||||||
|  | 
 | ||||||
|  | __all__ = ( | ||||||
|  |     'APIObject', | ||||||
|  |     'dt_formatter', | ||||||
|  |     'repr_dt', | ||||||
|  |     'PartialUser', | ||||||
|  |     'User', | ||||||
|  |     'Repository', | ||||||
|  |     'Issue', | ||||||
|  |     'Gist', | ||||||
|  |     'Organization', | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | def dt_formatter(time_str: str) -> datetime: | ||||||
|  |     if time_str is not None: | ||||||
|  |         return datetime.strptime(time_str, r"%Y-%m-%dT%H:%M:%SZ") | ||||||
|  |     return None | ||||||
|  | 
 | ||||||
|  | def repr_dt(_datetime: datetime) -> str: | ||||||
|  |     return _datetime.strftime(r'%d-%m-%Y, %H:%M:%S') | ||||||
|  | 
 | ||||||
|  | class APIObject: | ||||||
|  |     __slots__ = ( | ||||||
|  |         '_response', | ||||||
|  |         '_http' | ||||||
|  |     ) | ||||||
|  | 
 | ||||||
|  |     def __init__(self, response: dict[str, str | int | dict[str, str | int]], _http: 'http') -> None: | ||||||
|  |         self._http = _http | ||||||
|  |         self._response = response | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<{self.__class__.__name__}>' | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | #=== User stuff ===# | ||||||
|  | 
 | ||||||
|  | class _BaseUser(APIObject): | ||||||
|  |     __slots__ = ( | ||||||
|  |         'login', | ||||||
|  |         'id', | ||||||
|  |         ) | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         self._http = _http | ||||||
|  |         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!r}>' | ||||||
|  | 
 | ||||||
|  |     async def repos(self) -> list[Repository]: | ||||||
|  |         results = await self._http.get_user_repos(self) | ||||||
|  |         return [Repository(data, self._http) for data in results] | ||||||
|  | 
 | ||||||
|  |     async def gists(self) -> list[Gist]: | ||||||
|  |         results = await self._http.get_user_gists(self) | ||||||
|  |         return [Gist(data, self._http) for data in results] | ||||||
|  | 
 | ||||||
|  |     async def orgs(self) -> list[Organization]: | ||||||
|  |         results = await self._http.get_user_orgs(self) | ||||||
|  |         return [Organization(data, self._http) for data in results] | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | class User(_BaseUser):  | ||||||
|  |     __slots__ = ( | ||||||
|  |         'login', | ||||||
|  |         'id', | ||||||
|  |         'avatar_url', | ||||||
|  |         'html_url', | ||||||
|  |         'public_repos', | ||||||
|  |         'public_gists', | ||||||
|  |         'followers', | ||||||
|  |         'following', | ||||||
|  |         'created_at', | ||||||
|  |         ) | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         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 | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<User; login: {self.login!r}, id: {self.id}, created_at: {self.created_at}>' | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | class PartialUser(_BaseUser): | ||||||
|  |     __slots__ = ( | ||||||
|  |         'site_admin', | ||||||
|  |         'html_url', | ||||||
|  |         'avatar_url', | ||||||
|  |         ) + _BaseUser.__slots__ | ||||||
|  | 
 | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         self.site_admin = response.get('site_admin') | ||||||
|  |         self.html_url = response.get('html_url') | ||||||
|  |         self.avatar_url = response.get('avatar_url') | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<PartialUser; login: {self.login!r}, id: {self.id}, site_admin: {self.site_admin}, html_url: {self.html_url}>' | ||||||
|  | 
 | ||||||
|  |     async def _get_user(self) -> User: | ||||||
|  |         """Upgrades the PartialUser to a User object."""  | ||||||
|  |         response = await self._http.get_user(self.login) | ||||||
|  |         return User(response, self._http) | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | #=== Repository stuff ===# | ||||||
|  | 
 | ||||||
|  | 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, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         tmp = self.__slots__ + APIObject.__slots__ | ||||||
|  |         keys = {key: value for key,value in self._response.items() if key in tmp} | ||||||
|  |         for key, value in keys.items(): | ||||||
|  |             if key == 'owner': | ||||||
|  |                 setattr(self, key, PartialUser(value, self._http)) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if key == 'name': | ||||||
|  |                 setattr(self, key, value) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if '_at' in key and value is not None: | ||||||
|  |                 setattr(self, key, dt_formatter(value)) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if 'license' in key and value is None: | ||||||
|  |                 setattr(self, key, None) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if 'license' in key and value is not None: | ||||||
|  |                 setattr(self, key, value['name']) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             else: | ||||||
|  |                 setattr(self, key, value) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<Repository; id: {self.id}, name: {self.name!r}, owner: {self.owner}, updated_at: {self.updated_at}, default_branch: {self.default_branch!r}, license: {self.license!r}>' | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | class Issue(APIObject): | ||||||
|  |     __slots__ = ( | ||||||
|  |         'id', | ||||||
|  |         'title', | ||||||
|  |         'user', | ||||||
|  |         'labels', | ||||||
|  |         'state', | ||||||
|  |         'created_at', | ||||||
|  |         'closed_by', | ||||||
|  |     ) | ||||||
|  | 
 | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         tmp = self.__slots__ + APIObject.__slots__ | ||||||
|  |         keys = {key: value for key,value in self._response.items() if key in tmp} | ||||||
|  |         for key, value in keys.items(): | ||||||
|  |             if key == 'user': | ||||||
|  |                 setattr(self, key, PartialUser(value, self._http)) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if key == 'labels': | ||||||
|  |                 setattr(self, key, [label['name'] for label in value]) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             if key == 'closed_by': | ||||||
|  |                 setattr(self, key, User(value, self._http)) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             else: | ||||||
|  |                 setattr(self, key, value) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<Issue; id: {self.id}, title: {self.title}, user: {self.user}, created_at: {self.created_at}, state: {self.state}>' | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | #=== Gist stuff ===# | ||||||
|  | 
 | ||||||
|  | class Gist(APIObject): | ||||||
|  |     __slots__ = ( | ||||||
|  |         'id', | ||||||
|  |         'description', | ||||||
|  |         'html_url', | ||||||
|  |         'node_id', | ||||||
|  |         'files', | ||||||
|  |         'public', | ||||||
|  |         'owner', | ||||||
|  |         'created_at', | ||||||
|  |         'comments', | ||||||
|  |         'truncated', | ||||||
|  |         ) | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         tmp = self.__slots__ + APIObject.__slots__ | ||||||
|  |         keys = {key: value for key,value in self._response.items() if key in tmp} | ||||||
|  |         for key, value in keys.items(): | ||||||
|  |             if key == 'owner': | ||||||
|  |                 setattr(self, key, PartialUser(value, self._http)) | ||||||
|  |                 continue | ||||||
|  |             if key == 'created_at': | ||||||
|  |                 setattr(self, key, dt_formatter(value)) | ||||||
|  |                 continue | ||||||
|  |             else: | ||||||
|  |                 setattr(self, key, value) | ||||||
|  | 
 | ||||||
|  |     def __repr__(self) -> str: | ||||||
|  |         return f'<Gist; id: {self.id}, owner: {self.owner}, created_at: {self.created_at}>' | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | #=== Organization stuff ===# | ||||||
|  | 
 | ||||||
|  | class Organization(APIObject): | ||||||
|  |     __slots__ = ( | ||||||
|  |     'login', | ||||||
|  |     'id', | ||||||
|  |     'html_url', | ||||||
|  |     'is_verified', | ||||||
|  |     'public_repos', | ||||||
|  |     'public_gists', | ||||||
|  |     'followers', | ||||||
|  |     'following', | ||||||
|  |     'created_at', | ||||||
|  |     'avatar_url', | ||||||
|  |     ) | ||||||
|  | 
 | ||||||
|  |     def __init__(self, response: dict, _http: 'http') -> None: | ||||||
|  |         super().__init__(response, _http) | ||||||
|  |         tmp = self.__slots__ + APIObject.__slots__ | ||||||
|  |         keys = {key: value for key,value in self._response.items() if key in tmp} | ||||||
|  |         for key, value in keys.items(): | ||||||
|  |             if key == 'login': | ||||||
|  |                 setattr(self, key, value) | ||||||
|  |                 continue | ||||||
|  |             if '_at' in key and value is not None: | ||||||
|  |                 setattr(self, key, dt_formatter(value)) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |             else: | ||||||
|  |                 setattr(self, key, value) | ||||||
|  |                 continue | ||||||
|  | 
 | ||||||
|  |     def __repr__(self): | ||||||
|  |         return f'<Organization; login: {self.login!r}, id: {self.id}, html_url: {self.html_url}, is_verified: {self.is_verified}, public_repos: {self.public_repos}, public_gists: {self.public_gists}, created_at: {self.created_at}>' | ||||||
|  | @ -1,7 +0,0 @@ | ||||||
| #== __init__.py ==# |  | ||||||
| 
 |  | ||||||
| from .objects import * |  | ||||||
| from .user import * |  | ||||||
| from .repo import * |  | ||||||
| from .org import * |  | ||||||
| from .gists import * |  | ||||||
|  | @ -1,41 +0,0 @@ | ||||||
| #== gists.py ==# |  | ||||||
| 
 |  | ||||||
| import aiohttp |  | ||||||
| 
 |  | ||||||
| from .objects import APIOBJECT, dt_formatter |  | ||||||
| from . import PartialUser, User |  | ||||||
| from .. import http |  | ||||||
| 
 |  | ||||||
| __all__ = ( |  | ||||||
|     'Gist', |  | ||||||
|     ) |  | ||||||
| 
 |  | ||||||
| class Gist(APIOBJECT): |  | ||||||
|     __slots__ = ( |  | ||||||
|         'id', |  | ||||||
|         'description', |  | ||||||
|         'html_url', |  | ||||||
|         'node_id', |  | ||||||
|         'files', |  | ||||||
|         'public', |  | ||||||
|         'owner', |  | ||||||
|         'created_at', |  | ||||||
|         'comments', |  | ||||||
|         'truncated', |  | ||||||
|         ) |  | ||||||
|     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 keys.items(): |  | ||||||
|             if key == 'owner': |  | ||||||
|                 setattr(self, key, PartialUser(value, session)) |  | ||||||
|                 continue |  | ||||||
|             if key == 'created_at': |  | ||||||
|                 setattr(self, key, dt_formatter(value)) |  | ||||||
|                 continue |  | ||||||
|             else: |  | ||||||
|                 setattr(self, key, value) |  | ||||||
| 
 |  | ||||||
|     def __repr__(self) -> str: |  | ||||||
|         return f'<Gist; id: {self.id}, owner: {self.owner}, created_at: {self.created_at}>' |  | ||||||
|  | @ -1,29 +0,0 @@ | ||||||
| #== 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 | dict[str, str | int]], session: aiohttp.ClientSession) -> None: |  | ||||||
|         self._response = response |  | ||||||
|         self.session = session |  | ||||||
| 
 |  | ||||||
|     def __repr__(self) -> str: |  | ||||||
|         return f'<{self.__class__.__name__}>' |  | ||||||
|  | @ -1,52 +0,0 @@ | ||||||
| #== org.py ==# |  | ||||||
| 
 |  | ||||||
| import aiohttp |  | ||||||
| 
 |  | ||||||
| from .objects import APIOBJECT, dt_formatter |  | ||||||
| from . import PartialUser |  | ||||||
| from .. import http |  | ||||||
| 
 |  | ||||||
| __all__ = ( |  | ||||||
|     'Organization', |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| class Organization(APIOBJECT): |  | ||||||
|     __slots__ = ( |  | ||||||
|     'login', |  | ||||||
|     'id', |  | ||||||
|     'html_url', |  | ||||||
|     'is_verified', |  | ||||||
|     'public_repos', |  | ||||||
|     'public_gists', |  | ||||||
|     'followers', |  | ||||||
|     'following', |  | ||||||
|     'created_at', |  | ||||||
|     'avatar_url', |  | ||||||
|     ) |  | ||||||
| 
 |  | ||||||
|     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 keys.items(): |  | ||||||
|             if key == 'login': |  | ||||||
|                 setattr(self, key, value) |  | ||||||
|                 continue |  | ||||||
|             if '_at' in key and value is not None: |  | ||||||
|                 setattr(self, key, dt_formatter(value)) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             else: |  | ||||||
|                 setattr(self, key, value) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|     def __repr__(self): |  | ||||||
|         return f'<Organization; login: {self.login!r}, id: {self.id}, html_url: {self.html_url}, is_verified: {self.is_verified}, public_repos: {self.public_repos}, public_gists: {self.public_gists}, created_at: {self.created_at}>' |  | ||||||
| 
 |  | ||||||
|     @classmethod |  | ||||||
|     async def from_name(cls, session: aiohttp.ClientSession, org: str) -> 'Organization': |  | ||||||
|         """Fetch a repository from its name.""" |  | ||||||
|         response = await http.get_repo_from_name(session, org) |  | ||||||
|         return Organization(response, session) |  | ||||||
| 
 |  | ||||||
| 
 |  | ||||||
|  | @ -1,106 +0,0 @@ | ||||||
| #== repo.py ==# |  | ||||||
| 
 |  | ||||||
| import aiohttp |  | ||||||
| 
 |  | ||||||
| from .objects import APIOBJECT, dt_formatter |  | ||||||
| from . import PartialUser, User |  | ||||||
| from .. import http |  | ||||||
| 
 |  | ||||||
| __all__ = ( |  | ||||||
|     'Repository', |  | ||||||
|     'Issue' |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| 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 keys.items(): |  | ||||||
|             if key == 'owner': |  | ||||||
|                 setattr(self, key, PartialUser(value, session)) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if key == 'name': |  | ||||||
|                 setattr(self, key, value) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if '_at' in key and value is not None: |  | ||||||
|                 setattr(self, key, dt_formatter(value)) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if 'license' in key and value is None: |  | ||||||
|                 setattr(self, key, None) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if 'license' in key and value is not None: |  | ||||||
|                 setattr(self, key, value['name']) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             else: |  | ||||||
|                 setattr(self, key, value) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|     def __repr__(self) -> str: |  | ||||||
|         return f'<Repository; id: {self.id}, name: {self.name!r}, owner: {self.owner}, updated_at: {self.updated_at}, default_branch: {self.default_branch!r}, license: {self.license!r}>' |  | ||||||
| 
 |  | ||||||
|     @classmethod |  | ||||||
|     async def from_name(cls, session: aiohttp.ClientSession,owner: str, repo_name: str) -> 'Repository': |  | ||||||
|         """Fetch a repository from its name.""" |  | ||||||
|         response = await http.get_repo_from_name(session, owner, repo_name) |  | ||||||
|         return Repository(response, session) |  | ||||||
| 
 |  | ||||||
| 
 |  | ||||||
| class Issue(APIOBJECT): |  | ||||||
|     __slots__ = ( |  | ||||||
|         'id', |  | ||||||
|         'title', |  | ||||||
|         'user', |  | ||||||
|         'labels', |  | ||||||
|         'state', |  | ||||||
|         'created_at', |  | ||||||
|         'closed_by', |  | ||||||
|     ) |  | ||||||
| 
 |  | ||||||
|     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 keys.items(): |  | ||||||
|             if key == 'user': |  | ||||||
|                 setattr(self, key, PartialUser(value, session)) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if key == 'labels': |  | ||||||
|                 setattr(self, key, [label['name'] for label in value]) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             if key == 'closed_by': |  | ||||||
|                 setattr(self, key, User(value, session)) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|             else: |  | ||||||
|                 setattr(self, key, value) |  | ||||||
|                 continue |  | ||||||
| 
 |  | ||||||
|     def __repr__(self) -> str: |  | ||||||
|         return f'<Issue; id: {self.id}, title: {self.title}, user: {self.user}, created_at: {self.created_at}, state: {self.state}>' |  | ||||||
|  | @ -1,78 +0,0 @@ | ||||||
| #== 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!r}>' |  | ||||||
| 
 |  | ||||||
| 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 |  | ||||||
| 
 |  | ||||||
|     def __repr__(self) -> str: |  | ||||||
|         return f'<User; login: {self.login!r}, id: {self.id}, created_at: {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', |  | ||||||
|         'avatar_url', |  | ||||||
|         ) + _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) -> str: |  | ||||||
|         return f'<PartialUser; login: {self.login!r}, id: {self.id}, site_admin: {self.site_admin}, html_url: {self.html_url}>' |  | ||||||
| 
 |  | ||||||
|     async def _get_user(self) -> User: |  | ||||||
|         """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