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

Ran black.

Browse Source
This commit is contained in:
Halvor Holsten Strand
2023-09-29 14:40:16 +02:00
parent 20f5fd7a86
commit a9d35c0fd5
1 changed files with 36 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
github_backup/github_backup.py
View File

@@ -127,13 +127,13 @@ def parse_args(args=None):
"-t",
"--token",
dest="token_classic",
help="personal access, OAuth, or JSON Web token, or path to token (file://...)"
help="personal access, OAuth, or JSON Web token, or path to token (file://...)",
) # noqa
parser.add_argument(
"-f",
'--token-fine',
"--token-fine",
dest="token_fine",
help="fine-grained personal access token (github_pat_....), or path to token (file://...)"
help="fine-grained personal access token (github_pat_....), or path to token (file://...)",
) # noqa
parser.add_argument(
"--as-app",
@@ -440,18 +440,24 @@ def get_auth(args, encode=True, for_git_cli=False):
elif args.token_fine:
_path_specifier = "file://"
if args.token_fine.startswith(_path_specifier):
args.token_fine = open(args.token_fine[len(_path_specifier):],
"rt").readline().strip()
args.token_fine = (
open(args.token_fine[len(_path_specifier) :], "rt").readline().strip()
)
if args.token_fine.startswith("github_pat_"):
auth = args.token_fine
else:
raise Exception("Fine-grained token supplied does not look like a GitHub PAT")
raise Exception(
"Fine-grained token supplied does not look like a GitHub PAT"
)
elif args.token_classic:
_path_specifier = "file://"
if args.token_classic.startswith(_path_specifier):
args.token_classic = open(args.token_classic[len(_path_specifier):],
"rt").readline().strip()
args.token_classic = (
open(args.token_classic[len(_path_specifier) :], "rt")
.readline()
.strip()
)
if not args.as_app:
auth = args.token_classic + ":" + "x-oauth-basic"
@@ -528,7 +534,7 @@ def get_github_repo_url(args, repository):
)
else:
repo_url = "https://{0}@{1}/{2}/{3}.git".format(
"oauth2:"+auth,
"oauth2:" + auth,
get_github_host(args),
repository["owner"]["login"],
repository["name"],
@@ -548,7 +554,13 @@ def retrieve_data_gen(args, template, query_args=None, single_request=False):
while True:
page = page + 1
request = _construct_request(
per_page, page, query_args, template, auth, as_app=args.as_app, fine=True if args.token_fine is not None else False
per_page,
page,
query_args,
template,
auth,
as_app=args.as_app,
fine=True if args.token_fine is not None else False,
) # noqa
r, errors = _get_response(request, auth, template)
@@ -584,7 +596,13 @@ def retrieve_data_gen(args, template, query_args=None, single_request=False):
retries += 1
time.sleep(5)
request = _construct_request(
per_page, page, query_args, template, auth, as_app=args.as_app, fine=True if args.token_fine is not None else False
per_page,
page,
query_args,
template,
auth,
as_app=args.as_app,
fine=True if args.token_fine is not None else False,
) # noqa
r, errors = _get_response(request, auth, template)
@@ -668,10 +686,13 @@ def _get_response(request, auth, template):
return r, errors
def _construct_request(per_page, page, query_args, template, auth, as_app=None, fine=False):
def _construct_request(
per_page, page, query_args, template, auth, as_app=None, fine=False
):
querystring = urlencode(
dict(
list({"per_page": per_page, "page": page}.items()) + list(query_args.items())
list({"per_page": per_page, "page": page}.items())
+ list(query_args.items())
)
)