From 7f081f4a793e9cbf3f104adb9f970a5879da8251 Mon Sep 17 00:00:00 2001 From: Jiachen Yang Date: Wed, 15 Apr 2015 14:11:17 +0900 Subject: [PATCH] add access_token for github api, use NVCHECKER_GITHUB_TOKEN env var --- nvchecker/source/github.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/nvchecker/source/github.py b/nvchecker/source/github.py index 7e2bdda..4172c9b 100644 --- a/nvchecker/source/github.py +++ b/nvchecker/source/github.py @@ -1,7 +1,8 @@ +import os import json from functools import partial -from tornado.httpclient import AsyncHTTPClient +from tornado.httpclient import AsyncHTTPClient, HTTPRequest GITHUB_URL = 'https://api.github.com/repos/%s/commits?sha=%s' @@ -9,7 +10,11 @@ def get_version(name, conf, callback): repo = conf.get('github') br = conf.get('branch', 'master') url = GITHUB_URL % (repo, br) - AsyncHTTPClient().fetch(url, user_agent='lilydjwg/nvchecker', + headers = {} + if 'NVCHECKER_GITHUB_TOKEN' in os.environ: + headers = {'Authorization': 'token %s' % os.environ['NVCHECKER_GITHUB_TOKEN']} + request = HTTPRequest(url, headers=headers) + AsyncHTTPClient().fetch(request, user_agent='lilydjwg/nvchecker', callback=partial(_github_done, name, callback)) def _github_done(name, callback, res):