diff options
author | Jordan Petridis <jordan@centricular.com> | 2018-10-23 22:10:16 +0300 |
---|---|---|
committer | Jordan Petridis <jordan@centricular.com> | 2018-11-05 11:24:12 +0200 |
commit | 61c5d7b6cf943a708bab3400c8947925f63fae46 (patch) | |
tree | a282b85b6e1605cb0a093627cd6b15e8fc3d44f8 | |
parent | 6772fa76aaa6920ac76b983050efed125280216e (diff) |
build_manifest.py: split the request function in two
Make it so it does not depend on env variables so it will be easier
to test.
-rwxr-xr-x | gitlab/build_manifest.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/gitlab/build_manifest.py b/gitlab/build_manifest.py index a94ce31..dda8266 100755 --- a/gitlab/build_manifest.py +++ b/gitlab/build_manifest.py @@ -33,13 +33,19 @@ MANIFEST_TEMPLATE: str = """<?xml version="1.0" encoding="UTF-8"?> </manifest>""" -def request(path: str) -> Dict[str, str]: - gitlab_header: Dict[str, str] = {'JOB_TOKEN': os.environ["CI_JOB_TOKEN"]} - base_url: str = get_hostname(os.environ['CI_PROJECT_URL']) +def request_raw(path: str, token: str, project_url: str) -> Dict[str, str]: + gitlab_header: Dict[str, str] = {'JOB_TOKEN': token } + base_url: str = get_hostname(project_url) return requests.get(f"https://{base_url}/api/v4/" + path, headers=gitlab_header).json() +def request(path: str) -> Dict[str, str]: + token = os.environ["CI_JOB_TOKEN"] + project_url = os.environ['CI_PROJECT_URL'] + return request_raw(path, token, project_url) + + def get_hostname(url: str) -> str: return urlparse(url).hostname |