Created
October 19, 2018 15:09
-
-
Save SpotlightKid/d7f455dcd9ffc20ce1c83ebb8afda37e to your computer and use it in GitHub Desktop.
Download a single file from a GitHub repository
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
# -*- coding: utf-8 -*- | |
# | |
# gh-get.py | |
# | |
"""Download a single file from a GitHub repository.""" | |
import shutil | |
from os.path import expanduser | |
from posixpath import basename | |
import requests | |
import yaml | |
GITHUB_CONFIG = '~/.config/hub' | |
ENDPOINT = 'https://api.github.com/repos/{user}/{repo}/contents/{path}' | |
API_TYPE = 'application/vnd.github.v4.raw' | |
def main(args=None): | |
try: | |
user, repo, path = args.pop(0).split('/', 2) | |
except (IndexError, ValueError): | |
return "Usage: gh-get.py <gh-user>/<repo>/<path/to/file> [<dest>]" | |
else: | |
destfn = args.pop(0) if args else basename(path) | |
try: | |
with open(expanduser(GITHUB_CONFIG)) as fp: | |
hub_config = yaml.load(fp.read()) | |
access_token = hub_config['github.com'][0]['oauth_token'] | |
except OSError as exc: | |
return "Could not read configuration file '{}': {}".format( | |
GITHUB_CONFIG, exc) | |
except (IndexError, KeyError): | |
return "Access token not found in configuration file." | |
headers = dict(Authorization='token {}'.format(access_token), | |
Accept=API_TYPE) | |
url = ENDPOINT.format(path=path, repo=repo, user=user) | |
try: | |
with requests.get(url, headers=headers, stream=True) as req: | |
req.raise_for_status() | |
with open(destfn, 'wb') as fp: | |
shutil.copyfileobj(req.raw, fp) | |
except (requests.HTTPError, OSError) as exc: | |
return "Error downloading file: {}".format(exc) | |
if __name__ == '__main__': | |
import sys | |
sys.exit(main(sys.argv[1:]) or 0) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment