mirror of
https://github.com/status-im/PyGithub.git
synced 2026-08-31 19:01:15 +00:00
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
class PaginatedList:
|
|
def __init__( self, contentClass, requester, headers, data ):
|
|
self.__requester = requester
|
|
self.__contentClass = contentClass
|
|
self.__elements = []
|
|
self.__appendData( headers, data )
|
|
|
|
def __iter__( self ):
|
|
for element in self.__elements:
|
|
yield element
|
|
while self.__nextUrl is not None:
|
|
newElements = self.__fetchNextPage()
|
|
for element in newElements:
|
|
yield element
|
|
|
|
class __Slice:
|
|
def __init__( self, theList, theSlice ):
|
|
self.__list = theList
|
|
self.__start = theSlice.start or 0
|
|
self.__stop = theSlice.stop
|
|
self.__step = theSlice.step or 1
|
|
|
|
def __iter__( self ):
|
|
index = self.__start
|
|
while not self.__finished( index ) :
|
|
if self.__list._isBiggerThan( index ):
|
|
yield self.__list[ index ]
|
|
index += self.__step
|
|
else:
|
|
return
|
|
|
|
def __finished( self, index ):
|
|
return self.__stop is not None and index >= self.__stop
|
|
|
|
def __getitem__( self, index ):
|
|
assert isinstance( index, ( int, slice ) )
|
|
if isinstance( index, int ):
|
|
self.__fetchToIndex( index )
|
|
return self.__elements[ index ]
|
|
else:
|
|
return self.__Slice( self, index )
|
|
|
|
def _isBiggerThan( self, index ):
|
|
return len( self.__elements ) > index or self.__nextUrl is not None
|
|
|
|
def __fetchToIndex( self, index ):
|
|
while len( self.__elements ) <= index and self.__nextUrl is not None:
|
|
self.__fetchNextPage()
|
|
|
|
def __fetchNextPage( self ):
|
|
status, headers, data = self.__requester.request( "GET", self.__nextUrl, None, None )
|
|
return self.__appendData( headers, data )
|
|
|
|
def __appendData( self, headers, data ):
|
|
links = self.__parseLinkHeader( headers )
|
|
if len( data ) > 0 and "next" in links:
|
|
self.__nextUrl = links[ "next" ]
|
|
else:
|
|
self.__nextUrl = None
|
|
|
|
newElements = [
|
|
self.__contentClass( self.__requester, element, completed = False )
|
|
for element in data
|
|
]
|
|
self.__elements += newElements
|
|
|
|
return newElements
|
|
|
|
def __parseLinkHeader( self, headers ):
|
|
links = {}
|
|
if "link" in headers:
|
|
linkHeaders = headers[ "link" ].split( "," )
|
|
for linkHeader in linkHeaders:
|
|
( url, rel ) = linkHeader.split( "; " )
|
|
url = url[ 1 : -1 ]
|
|
rel = rel[ 5 : -1 ]
|
|
links[ rel ] = url
|
|
return links
|