mirror of
https://github.com/status-im/PyGithub.git
synced 2026-08-31 10:51:14 +00:00
Refactor
This commit is contained in:
Vincent Jacques
+1
-89
@@ -2,6 +2,7 @@ import itertools
|
||||
|
||||
import ArgumentsChecker
|
||||
from ObjectCapacities.Basic import AttributeFromCallable, MethodFromCallable
|
||||
from ObjectCapacities.List import ListOfObjects, ListOfReferences, ListGetable, Creatable
|
||||
|
||||
class BadGithubObjectException( Exception ):
|
||||
pass
|
||||
@@ -62,95 +63,6 @@ class Identity( AttributeFromCallable ):
|
||||
def __init__( self, identity ):
|
||||
AttributeFromCallable.__init__( self, "_identity", identity )
|
||||
|
||||
class ListOfReferences:
|
||||
def __init__( self, attributeName, type, addable = False, removable = False, hasable = False, getParameters = [] ):
|
||||
self.__attributeName = attributeName
|
||||
self.__type = type
|
||||
self.__getName = "get_" + attributeName
|
||||
self.__getParameters = getParameters
|
||||
if addable:
|
||||
self.__addName = "add_to_" + attributeName
|
||||
else:
|
||||
self.__addName = None
|
||||
if removable:
|
||||
self.__removeName = "remove_from_" + attributeName
|
||||
else:
|
||||
self.__removeName = None
|
||||
if hasable:
|
||||
self.__hasName = "has_in_" + attributeName
|
||||
else:
|
||||
self.__hasName = None
|
||||
|
||||
def apply( self, cls ):
|
||||
cls._addMethod( self.__getName, self.__executeGet )
|
||||
if self.__addName is not None:
|
||||
cls._addMethod( self.__addName, self.__executeAdd )
|
||||
if self.__removeName is not None:
|
||||
cls._addMethod( self.__removeName, self.__executeRemove )
|
||||
if self.__hasName is not None:
|
||||
cls._addMethod( self.__hasName, self.__executeHas )
|
||||
|
||||
def __executeGet( self, obj, *args, **kwds ):
|
||||
### @todo ArgumentsChecker?
|
||||
for arg, argumentName in itertools.izip( args, self.__getParameters ):
|
||||
kwds[ argumentName ] = arg
|
||||
return [
|
||||
self.__type( obj._github, attributes, lazy = True )
|
||||
for attributes in obj._github._dataRequest( "GET", obj._baseUrl + "/" + self.__attributeName, kwds, None )
|
||||
]
|
||||
|
||||
def __executeAdd( self, obj, toBeAdded ):
|
||||
assert isinstance( toBeAdded, self.__type )
|
||||
obj._github._statusRequest( "PUT", obj._baseUrl + "/" + self.__attributeName + "/" + toBeAdded._identity, None, None )
|
||||
|
||||
def __executeRemove( self, obj, toBeDeleted ):
|
||||
assert isinstance( toBeDeleted, self.__type )
|
||||
obj._github._statusRequest( "DELETE", obj._baseUrl + "/" + self.__attributeName + "/" + toBeDeleted._identity, None, None )
|
||||
|
||||
def __executeHas( self, obj, toBeQueried ):
|
||||
assert isinstance( toBeQueried, self.__type )
|
||||
return obj._github._statusRequest( "GET", obj._baseUrl + "/" + self.__attributeName + "/" + toBeQueried._identity, None, None ) == 204
|
||||
|
||||
class Creatable:
|
||||
def __init__( self, singularName, mandatoryParameters, optionalParameters ):
|
||||
self.__argumentsChecker = ArgumentsChecker.ArgumentsChecker( mandatoryParameters, optionalParameters )
|
||||
self.__createName = "create_" + singularName
|
||||
|
||||
def apply( self, list, cls ):
|
||||
self.__type = list.type
|
||||
self.__attributeName = list.attributeName
|
||||
cls._addMethod( self.__createName, self.__execute )
|
||||
|
||||
def __execute( self, obj, *args, **kwds ):
|
||||
data = self.__argumentsChecker.check( args, kwds )
|
||||
return self.__type( obj._github, obj._github._dataRequest( "POST", obj._baseUrl + "/" + self.__attributeName, None, data ), lazy = True )
|
||||
|
||||
class ListGetable:
|
||||
def apply( self, list, cls ):
|
||||
self.__type = list.type
|
||||
self.__attributeName = list.attributeName
|
||||
cls._addMethod( "get_" + list.attributeName, self.__execute )
|
||||
|
||||
def __execute( self, obj ):
|
||||
return [
|
||||
self.__type( obj._github, attributes, lazy = True )
|
||||
for attributes in obj._github._dataRequest( "GET", obj._baseUrl + "/" + self.__attributeName, None, None )
|
||||
]
|
||||
|
||||
### @todo Merge ObjectGetter in ListOfObjects, with a SingleGettable similar to Creatable
|
||||
### @todo Add a ListGetable that couls be False for non-getable lists (repo/git/commits for example)
|
||||
class ListOfObjects:
|
||||
def __init__( self, attributeName, type, *capacities ):
|
||||
self.attributeName = attributeName
|
||||
self.type = type
|
||||
self.__getName = "get_" + attributeName
|
||||
self.__capacities = list( capacities )
|
||||
self.__capacities.append( ListGetable() )
|
||||
|
||||
def apply( self, cls ):
|
||||
for capacity in self.__capacities:
|
||||
capacity.apply( self, cls )
|
||||
|
||||
class Editable( MethodFromCallable ):
|
||||
def __init__( self, mandatoryParameters, optionalParameters ):
|
||||
MethodFromCallable.__init__( self, "edit", self.__execute )
|
||||
|
||||
@@ -0,0 +1,92 @@
|
||||
import itertools
|
||||
|
||||
import ArgumentsChecker
|
||||
|
||||
class ListOfReferences:
|
||||
def __init__( self, attributeName, type, addable = False, removable = False, hasable = False, getParameters = [] ):
|
||||
self.__attributeName = attributeName
|
||||
self.__type = type
|
||||
self.__getName = "get_" + attributeName
|
||||
self.__getParameters = getParameters
|
||||
if addable:
|
||||
self.__addName = "add_to_" + attributeName
|
||||
else:
|
||||
self.__addName = None
|
||||
if removable:
|
||||
self.__removeName = "remove_from_" + attributeName
|
||||
else:
|
||||
self.__removeName = None
|
||||
if hasable:
|
||||
self.__hasName = "has_in_" + attributeName
|
||||
else:
|
||||
self.__hasName = None
|
||||
|
||||
def apply( self, cls ):
|
||||
cls._addMethod( self.__getName, self.__executeGet )
|
||||
if self.__addName is not None:
|
||||
cls._addMethod( self.__addName, self.__executeAdd )
|
||||
if self.__removeName is not None:
|
||||
cls._addMethod( self.__removeName, self.__executeRemove )
|
||||
if self.__hasName is not None:
|
||||
cls._addMethod( self.__hasName, self.__executeHas )
|
||||
|
||||
def __executeGet( self, obj, *args, **kwds ):
|
||||
### @todo ArgumentsChecker?
|
||||
for arg, argumentName in itertools.izip( args, self.__getParameters ):
|
||||
kwds[ argumentName ] = arg
|
||||
return [
|
||||
self.__type( obj._github, attributes, lazy = True )
|
||||
for attributes in obj._github._dataRequest( "GET", obj._baseUrl + "/" + self.__attributeName, kwds, None )
|
||||
]
|
||||
|
||||
def __executeAdd( self, obj, toBeAdded ):
|
||||
assert isinstance( toBeAdded, self.__type )
|
||||
obj._github._statusRequest( "PUT", obj._baseUrl + "/" + self.__attributeName + "/" + toBeAdded._identity, None, None )
|
||||
|
||||
def __executeRemove( self, obj, toBeDeleted ):
|
||||
assert isinstance( toBeDeleted, self.__type )
|
||||
obj._github._statusRequest( "DELETE", obj._baseUrl + "/" + self.__attributeName + "/" + toBeDeleted._identity, None, None )
|
||||
|
||||
def __executeHas( self, obj, toBeQueried ):
|
||||
assert isinstance( toBeQueried, self.__type )
|
||||
return obj._github._statusRequest( "GET", obj._baseUrl + "/" + self.__attributeName + "/" + toBeQueried._identity, None, None ) == 204
|
||||
|
||||
class Creatable:
|
||||
def __init__( self, singularName, mandatoryParameters, optionalParameters ):
|
||||
self.__argumentsChecker = ArgumentsChecker.ArgumentsChecker( mandatoryParameters, optionalParameters )
|
||||
self.__createName = "create_" + singularName
|
||||
|
||||
def apply( self, list, cls ):
|
||||
self.__type = list.type
|
||||
self.__attributeName = list.attributeName
|
||||
cls._addMethod( self.__createName, self.__execute )
|
||||
|
||||
def __execute( self, obj, *args, **kwds ):
|
||||
data = self.__argumentsChecker.check( args, kwds )
|
||||
return self.__type( obj._github, obj._github._dataRequest( "POST", obj._baseUrl + "/" + self.__attributeName, None, data ), lazy = True )
|
||||
|
||||
class ListGetable:
|
||||
def apply( self, list, cls ):
|
||||
self.__type = list.type
|
||||
self.__attributeName = list.attributeName
|
||||
cls._addMethod( "get_" + list.attributeName, self.__execute )
|
||||
|
||||
def __execute( self, obj ):
|
||||
return [
|
||||
self.__type( obj._github, attributes, lazy = True )
|
||||
for attributes in obj._github._dataRequest( "GET", obj._baseUrl + "/" + self.__attributeName, None, None )
|
||||
]
|
||||
|
||||
### @todo Merge ObjectGetter in ListOfObjects, with a SingleGettable similar to Creatable
|
||||
### @todo Add a ListGetable that couls be False for non-getable lists (repo/git/commits for example)
|
||||
class ListOfObjects:
|
||||
def __init__( self, attributeName, type, *capacities ):
|
||||
self.attributeName = attributeName
|
||||
self.type = type
|
||||
self.__getName = "get_" + attributeName
|
||||
self.__capacities = list( capacities )
|
||||
self.__capacities.append( ListGetable() )
|
||||
|
||||
def apply( self, cls ):
|
||||
for capacity in self.__capacities:
|
||||
capacity.apply( self, cls )
|
||||
Reference in New Issue
Block a user