1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-16 14:15:00 +00:00

Adding user object

Editing Exceptions and changing _state to session
This commit is contained in:
VarMonke 2022-03-27 11:03:47 +05:30
parent 25710629f2
commit 77de27ea5a
3 changed files with 40 additions and 11 deletions

View file

@ -11,15 +11,23 @@ __all__ = (
'User',
)
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',
'_state'
'session'
)
def __init__(self, response: dict[str, str | int], session: aiohttp.ClientSession) -> None:
self._response = response
self._state = session
self.session = session
def __repr__(self) -> str:
@ -52,7 +60,28 @@ class User(_BaseUser):
)
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 == 'login':
self.login = value
return
if '_at' in key and value is not None:
setattr(self, key, dt_formatter(value))
return
setattr(self, key, value)
def __repr__(self):
return f'<User; login: {self.login}, id: {self.id}, created_at: {repr_dt(self.created_at)}>'
@classmethod
async def get_user(self, 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)