1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-17 22:45:08 +00:00

Moved objects into their own place in a folder for readability

This commit is contained in:
sudosnok 2022-03-27 13:04:00 +01:00
parent 9a2889702b
commit d5ba5bc58d
6 changed files with 93 additions and 73 deletions

View file

@ -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+)\"")

View file

@ -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

View file

@ -0,0 +1,5 @@
#== __init__.py ==#
from .objects import *
from .user import *
from .repo import *

29
Github/objects/objects.py Normal file
View 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
View 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}, >'

View file

@ -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}, >'