mirror of
https://github.com/RGBCube/GitHubWrapper
synced 2025-05-17 22:45:08 +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