""" github3.models ~~~~~~~~~~~~~~ This module provides the Github3 object model. """ from .helpers import to_python, to_api class BaseResource(object): """A BaseResource object.""" _strs = [] _ints = [] _dates = [] _bools = [] _map = {} _writeable = [] _modified = [] def __init__(self): self._bootstrap() super(BaseResource, self).__init__() def __dir__(self): d = self.__dict__.copy() try: del d['_gh'] except KeyError: pass return d.keys() def _bootstrap(self): """Bootstraps the model object based on configured values.""" for attr in (self._strs + self._ints + self._dates + self._bools + self._map.keys()): setattr(self, attr, None) @classmethod def new_from_dict(cls, d, gh=None): return to_python( obj=cls(), in_dict=d, str_keys = cls._strs, int_keys = cls._ints, date_keys = cls._dates, bool_keys = cls._bools, object_map = cls._map, _gh = gh ) def update(self): pass def setattr(self, k, v): # TODO: when writable key changed, pass class Plan(BaseResource): """Github Plan object model.""" _strs = ['name'] _ints = ['space', 'collaborators', 'private_repos'] def __repr__(self): return ''.format(str(self.name)) class User(BaseResource): """Github User object model.""" _strs = [ 'login','gravatar_url', 'url', 'name', 'company', 'blog', 'location', 'email', 'bio', 'html_url'] _ints = ['id', 'public_repos', 'public_gists', 'followers', 'following'] _dates = ['created_at',] _bools = ['hireable', ] # _map = {} # _writeable = [] def __repr__(self): return ''.format(self.login) def repos(self, limit=None): return self._gh._get_resources(('users', self.login, 'repos'), Repo) def orgs(self): return self._gh._get_resources(('users', self.login, 'orgs'), Org) class CurrentUser(User): """Github Current User object model.""" _ints = [ 'id', 'public_repos', 'public_gists', 'followers', 'following', 'total_private_repos', 'owned_private_repos', 'private_gists', 'disk_usage', 'collaborators'] _map = {'plan': Plan} _writeable = ['name', 'email', 'blog', 'company', 'location', 'hireable', 'bio'] def __repr__(self): return ''.format(self.login) def repos(self, limit=None): return self._gh._get_resources(('user', 'repos'), Repo) def orgs(self): return self._gh._get_resources(('user', 'orgs'), Org) class Org(BaseResource): """Github Organization object model.""" _strs = [ 'login', 'url', 'avatar_url', 'name', 'company', 'blog', 'location', 'email' 'html_url', 'type'] _ints = [ 'id', 'public_repos', 'public_gists', 'followers', 'following', 'total_private_repos', 'owned_private_repos', 'private_gists', 'disk_usage', 'collaborators'] _dates = ['created_at'] _map = {'plan': Plan} _writable = ['billing_email', 'blog', 'company', 'email', 'location', 'name'] def __repr__(self): return ''.format(self.login) def repos(self, limit=None): return self._gh._get_resources(('orgs', self.login, 'repos'), Repo) class Repo(BaseResource): _strs = [ 'url', 'html_url', 'clone_url', 'git_url', 'ssh_url', 'svn_url', 'name', 'description', 'homepage', 'language', 'master_branch'] _bools = ['private', 'fork'] _ints = ['forks', 'watchers', 'size',] _dates = ['pushed_at', 'created_at'] _map = {'owner': User} def __repr__(self): return ''.format(self.owner.login, self.name) # owner