PyGithub/github/Legacy.py

# Copyright 2012 Vincent Jacques
# vincent@vincent-jacques.net

# This file is part of PyGithub. http://vincent-jacques.net/PyGithub

# PyGithub is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License
# as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

# PyGithub is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public License along with PyGithub.  If not, see <http://www.gnu.org/licenses/>.

import urlparse

from PaginatedList import PaginatedListBase

class PaginatedList( PaginatedListBase ):
    def __init__( self, url, args, requester, key, convert, contentClass ):
        PaginatedListBase.__init__( self )
        self.__url = url
        self.__args = args
        self.__requester = requester
        self.__key = key
        self.__convert = convert
        self.__contentClass = contentClass
        self.__nextPage = 0
        self.__continue = True

    def _couldGrow( self ):
        return self.__continue

    def _fetchNextPage( self ):
        page = self.__nextPage
        self.__nextPage += 1
        return self.get_page( page )

    def get_page( self, page ):
        assert isinstance( page, int ), page
        args = dict( self.__args )
        if page != 0:
            args[ "start_page" ] = page + 1
        headers, data = self.__requester.requestAndCheck(
            "GET",
            self.__url,
            args,
            None
        )
        self.__continue = len( data[ self.__key ] ) > 0
        return [
            self.__contentClass( self.__requester, self.__convert( element ), completed = False )
            for element in data[ self.__key ]
        ]

def convertUser( attributes ):
    convertedAttributes = {
        "login": attributes[ "login" ],
        "url": "/users/" + attributes[ "login" ],
    }
    if "gravatar_id" in attributes: convertedAttributes[ "gravatar_id" ] = attributes[ "gravatar_id" ]
    if "followers" in attributes: convertedAttributes[ "followers" ] = attributes[ "followers" ]
    if "repos" in attributes: convertedAttributes[ "public_repos" ] = attributes[ "repos" ]
    if "name" in attributes: convertedAttributes[ "name" ] = attributes[ "name" ]
    if "created_at" in attributes: convertedAttributes[ "created_at" ] = attributes[ "created_at" ]
    if "location" in attributes: convertedAttributes[ "location" ] = attributes[ "location" ]
    return convertedAttributes

def convertRepo( attributes ):
    convertedAttributes = {
        "owner": { "login": attributes[ "owner" ], "url": "/users/" + attributes[ "owner" ] },
        "url": "/repos/" + attributes[ "owner" ] + "/" + attributes[ "name" ],
    }
    if "pushed_at" in attributes: convertedAttributes[ "pushed_at" ] = attributes[ "pushed_at" ]
    if "homepage" in attributes: convertedAttributes[ "homepage" ] = attributes[ "homepage" ]
    if "created_at" in attributes: convertedAttributes[ "created_at" ] = attributes[ "created_at" ]
    if "watchers" in attributes: convertedAttributes[ "watchers" ] = attributes[ "watchers" ]
    if "has_downloads" in attributes: convertedAttributes[ "has_downloads" ] = attributes[ "has_downloads" ]
    if "fork" in attributes: convertedAttributes[ "fork" ] = attributes[ "fork" ]
    if "has_issues" in attributes: convertedAttributes[ "has_issues" ] = attributes[ "has_issues" ]
    if "has_wiki" in attributes: convertedAttributes[ "has_wiki" ] = attributes[ "has_wiki" ]
    if "forks" in attributes: convertedAttributes[ "forks" ] = attributes[ "forks" ]
    if "size" in attributes: convertedAttributes[ "size" ] = attributes[ "size" ]
    if "private" in attributes: convertedAttributes[ "private" ] = attributes[ "private" ]
    if "open_issues" in attributes: convertedAttributes[ "open_issues" ] = attributes[ "open_issues" ]
    if "description" in attributes: convertedAttributes[ "description" ] = attributes[ "description" ]
    if "language" in attributes: convertedAttributes[ "language" ] = attributes[ "language" ]
    if "name" in attributes: convertedAttributes[ "name" ] = attributes[ "name" ]
    return convertedAttributes

def convertIssue( attributes ):
    convertedAttributes = {
        "number": attributes[ "number" ],
        "url": "/repos" + urlparse.urlparse( attributes[ "html_url" ] ).path,
        "user": { "login": attributes[ "user" ], "url": "/users/" + attributes[ "user" ] },
    }
    if "labels" in attributes: convertedAttributes[ "labels" ] = [ { "name": label } for label in attributes[ "labels" ] ]
    if "title" in attributes: convertedAttributes[ "title" ] = attributes[ "title" ]
    if "created_at" in attributes: convertedAttributes[ "created_at" ] = attributes[ "created_at" ]
    if "comments" in attributes: convertedAttributes[ "comments" ] = attributes[ "comments" ]
    if "body" in attributes: convertedAttributes[ "body" ] = attributes[ "body" ]
    if "updated_at" in attributes: convertedAttributes[ "updated_at" ] = attributes[ "updated_at" ]
    if "state" in attributes: convertedAttributes[ "state" ] = attributes[ "state" ]
    return convertedAttributes