github_lib.py

import requests
import re
import sys

API_BASE = 'https://api.github.com'


class Github:
    def __init__(self, token):
        self.TOKEN = token
        return

    def __get_user_repos(self, username):
        """
        obtain repo info from Github given a username
        :param username: String. Github handle of the user requested
        :return: JSON containing repo info through pagination and response status
        """
        # initial call to obtain the first 100 public repos
        all_content = []
        content = requests.get('{0}/users/{1}/repos?per_page=100'.format(API_BASE, username), headers={'Content-Type': 'application/json'})

        # return error directly before attempting to paginate
        if content.status_code == 404:
            return 404

        all_content += content.json()

        # check if there's more pages to go and perform pagination
        if 'Link' in content.headers.keys() and 'rel="next"' in content.headers['Link']:
            next_link = re.search(r"<([a-z:/.?_=&0-9]+)>; rel=\"next\"", content.headers['Link']).group(1)
            content = requests.get(next_link, headers={'Authorization': 'TOKEN {0}'.format(self.TOKEN)})
            all_content += content.json()

        return all_content

    def __find_top5(self, content_list, simple=False):
        """
        get the largest 5 public repos
        :param content_list: List of dict's. Repo information in Python list/ dict formats, converted from JSON
        :param simple: Boolean. [default = False] Whether to send full response or high level information about each repo
        :return: list of dict's of the largest 5 public repos
        """
        print(content_list)
        top5 = sorted(content_list, key=lambda x: x['size'], reverse=True)[0:5]

        # if only top level info is required
        if simple:
            top5 = map(lambda x:
                       {'name': x['name'],
                        'size': x['size'],
                        'html_url' : x['html_url']
                       }, top5)

        return top5

    def get_top5(self, username, simple=False):
        """
        public interface for API use
        :param username: String. Github handle of the user requested
        :param simple: Boolean. [default = False] Whether to send full response or high level information about each repo
        :return:
        """
        content = self.__get_user_repos(username)

        # continue to find top 5 if we did not encounter an error
        if type(content) == int and content == 404:
            return content
        else:
            return self.__find_top5(content, simple)


# for testing purposes
#if __name__ == '__main__':
    #g = Github("130e60bfb7a9c2f2c4c532a62e7c16b2c79a16e2")
    #print g.get_top5('facebook', simple=True)