thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increase safe limits of token requests to pick it

Browse Source
This commit is contained in:
Mikhail f. Shiryaev 2024-06-04 12:39:07 +02:00
parent 938b5d267b
commit 0cd4e0ecaa
No known key found for this signature in database
GPG Key ID: 4B02ED204C7D93F4
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/ci/get_robot_token.py
View File

@ -1,7 +1,7 @@
#!/usr/bin/env python3
import logging
from dataclasses import dataclass
import random
from dataclasses import dataclass
from typing import Any, Dict, List, Optional, Union
import boto3 # type: ignore
@ -18,6 +18,9 @@ class Token:
rest: int
SAFE_REQUESTS_LIMIT = 1000
def get_parameter_from_ssm(
name: str, decrypt: bool = True, client: Optional[Any] = None
) -> str:
@ -94,7 +97,7 @@ def get_best_robot_token(tokens_path: str = "/github-tokens") -> str:
best_token = Token(user, value, rest)
elif best_token.rest < rest:
best_token = Token(user, value, rest)
if best_token.rest > 300:
if best_token.rest > SAFE_REQUESTS_LIMIT:
break
assert best_token
ROBOT_TOKEN = best_token