Wavyzz/python-github-backup
2
0
Fork 0
mirror of https://github.com/josegonzalez/python-github-backup.git synced 2025-12-05 16:18:02 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #146 from timm3/upstream-123

Browse Source 
Authenticate as Github App
This commit is contained in:
Jose Diaz-Gonzalez
2020-02-25 14:39:27 -05:00
committed by GitHub
parent 3f65eadee1 523c811cc6
commit c8171b692a
2 changed files with 30 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bin/github-backup
View File

@@ -26,9 +26,12 @@ def main():
if args.lfs_clone: if args.lfs_clone:
check_git_lfs_install() check_git_lfs_install()
log_info('Backing up user {0} to {1}'.format(args.user, output_directory)) if not args.as_app:
log_info('Backing up user {0} to {1}'.format(args.user, output_directory))
authenticated_user = get_authenticated_user(args)
else:
authenticated_user = {'login': None}
authenticated_user = get_authenticated_user(args)
repositories = retrieve_repositories(args, authenticated_user) repositories = retrieve_repositories(args, authenticated_user)
repositories = filter_repositories(args, repositories) repositories = filter_repositories(args, repositories)
backup_repositories(args, output_directory, repositories) backup_repositories(args, output_directory, repositories)

34
github_backup/github_backup.py
View File

@@ -168,7 +168,11 @@ def parse_args():
parser.add_argument('-t', parser.add_argument('-t',
'--token', '--token',
dest='token', dest='token',
help='personal access or OAuth token, or path to token (file://...)') # noqa help='personal access, OAuth, or JSON Web token, or path to token (file://...)') # noqa
parser.add_argument('--as-app',
action='store_true',
dest='as_app',
help='authenticate as github app instead of as a user.')
parser.add_argument('-o', parser.add_argument('-o',
'--output-directory', '--output-directory',
default='.', default='.',
@@ -325,7 +329,7 @@ def parse_args():
return parser.parse_args() return parser.parse_args()
def get_auth(args, encode=True): def get_auth(args, encode=True, for_git_cli=False):
auth = None auth = None
if args.osx_keychain_item_name: if args.osx_keychain_item_name:
@@ -353,7 +357,13 @@ def get_auth(args, encode=True):
if args.token.startswith(_path_specifier): if args.token.startswith(_path_specifier):
args.token = open(args.token[len(_path_specifier):], args.token = open(args.token[len(_path_specifier):],
'rt').readline().strip() 'rt').readline().strip()
auth = args.token + ':' + 'x-oauth-basic' if not args.as_app:
auth = args.token + ':' + 'x-oauth-basic'
else:
if not for_git_cli:
auth = args.token
else:
auth = 'x-access-token:' + args.token
elif args.username: elif args.username:
if not args.password: if not args.password:
args.password = getpass.getpass() args.password = getpass.getpass()
@@ -399,7 +409,7 @@ def get_github_repo_url(args, repository):
if args.prefer_ssh: if args.prefer_ssh:
return repository['ssh_url'] return repository['ssh_url']
auth = get_auth(args, False) auth = get_auth(args, encode=False, for_git_cli=True)
if auth and repository['private'] == True: if auth and repository['private'] == True:
repo_url = 'https://{0}@{1}/{2}/{3}.git'.format( repo_url = 'https://{0}@{1}/{2}/{3}.git'.format(
auth, auth,
@@ -413,14 +423,14 @@ def get_github_repo_url(args, repository):
def retrieve_data_gen(args, template, query_args=None, single_request=False): def retrieve_data_gen(args, template, query_args=None, single_request=False):
auth = get_auth(args) auth = get_auth(args, encode=not args.as_app)
query_args = get_query_args(query_args) query_args = get_query_args(query_args)
per_page = 100 per_page = 100
page = 0 page = 0
while True: while True:
page = page + 1 page = page + 1
request = _construct_request(per_page, page, query_args, template, auth) # noqa request = _construct_request(per_page, page, query_args, template, auth, as_app=args.as_app) # noqa
r, errors = _get_response(request, auth, template) r, errors = _get_response(request, auth, template)
status_code = int(r.getcode()) status_code = int(r.getcode())
@@ -430,7 +440,7 @@ def retrieve_data_gen(args, template, query_args=None, single_request=False):
print('API request returned HTTP 502: Bad Gateway. Retrying in 5 seconds') print('API request returned HTTP 502: Bad Gateway. Retrying in 5 seconds')
retries += 1 retries += 1
time.sleep(5) time.sleep(5)
request = _construct_request(per_page, page, query_args, template, auth) # noqa request = _construct_request(per_page, page, query_args, template, auth, as_app=args.as_app) # noqa
r, errors = _get_response(request, auth, template) r, errors = _get_response(request, auth, template)
status_code = int(r.getcode()) status_code = int(r.getcode())
@@ -495,7 +505,7 @@ def _get_response(request, auth, template):
return r, errors return r, errors
def _construct_request(per_page, page, query_args, template, auth): def _construct_request(per_page, page, query_args, template, auth, as_app=None):
querystring = urlencode(dict(list({ querystring = urlencode(dict(list({
'per_page': per_page, 'per_page': per_page,
'page': page 'page': page
@@ -503,7 +513,13 @@ def _construct_request(per_page, page, query_args, template, auth):
request = Request(template + '?' + querystring) request = Request(template + '?' + querystring)
if auth is not None: if auth is not None:
request.add_header('Authorization', 'Basic '.encode('ascii') + auth) if not as_app:
request.add_header('Authorization', 'Basic '.encode('ascii') + auth)
else:
if not PY2:
auth = auth.encode('ascii')
request.add_header('Authorization', 'token '.encode('ascii') + auth)
request.add_header('Accept', 'application/vnd.github.machine-man-preview+json')
log_info('Requesting {}?{}'.format(template, querystring)) log_info('Requesting {}?{}'.format(template, querystring))
return request return request