thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-27 01:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #53833 from ClickHouse/reduce-get-parameter-calls

Browse Source 
Get rid of describe_parameters for the best robot token
This commit is contained in:
Mikhail f. Shiryaev 2023-08-25 13:36:34 +02:00 committed by GitHub
commit b158a83e89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/ci/get_robot_token.py
View File

@ -54,18 +54,18 @@ def get_parameters_from_ssm(
ROBOT_TOKEN = None # type: Optional[Token]
def get_best_robot_token(token_prefix_env_name="github_robot_token_"):
def get_best_robot_token(tokens_path: str = "/github-tokens") -> str:
global ROBOT_TOKEN
if ROBOT_TOKEN is not None:
return ROBOT_TOKEN.value
client = boto3.client("ssm", region_name="us-east-1")
parameters_list = client.describe_parameters(
ParameterFilters=[
{"Key": "Name", "Option": "BeginsWith", "Values": [token_prefix_env_name]}
tokens = {
p["Name"]: p["Value"]
for p in client.get_parameters_by_path(Path=tokens_path, WithDecryption=True)[
"Parameters"
]
)["Parameters"]
assert parameters_list
tokens = get_parameters_from_ssm([p["Name"] for p in parameters_list], True, client)
}
assert tokens
for value in tokens.values():
gh = Github(value, per_page=100)