Skip to content

Commit 2df97c2

Browse files
authored
fix enum constants naming
1 parent 458ea4a commit 2df97c2

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

rotator.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,15 @@ def main():
2222
telegram_message += f"Host {args.gitlab_url} has issues with tokens:\n"
2323
telegram_message += "\n"
2424

25-
telegram_message += check_and_rotate_tokens(args.gitlab_url, args.gitlab_token, TokenType.GroupAccessToken)
26-
telegram_message += check_and_rotate_tokens(args.gitlab_url, args.gitlab_token, TokenType.ProjectAccessToken)
25+
telegram_message += check_and_rotate_tokens(args.gitlab_url, args.gitlab_token, TokenType.GROUP_ACCESS_TOKEN)
26+
telegram_message += check_and_rotate_tokens(args.gitlab_url, args.gitlab_token, TokenType.PROJECT_ACCESS_TOKEN)
2727
send_to_telegram(args.telegram_bot_token, args.telegram_chat_id, telegram_message)
2828
except Exception as err:
2929
print(f"Unexpected {err=}, {type(err)=}")
3030

3131
class TokenType(enum.Enum):
32-
GroupAccessToken = 0,
33-
ProjectAccessToken = 1
32+
GROUP_ACCESS_TOKEN = 0,
33+
PROJECT_ACCESS_TOKEN = 1
3434

3535
def check_and_rotate_tokens(gitlab_url: str, gitlab_token : str, token_type: TokenType):
3636

@@ -44,10 +44,10 @@ def check_and_rotate_tokens(gitlab_url: str, gitlab_token : str, token_type: Tok
4444
first_message_word=''
4545

4646
match token_type:
47-
case TokenType.GroupAccessToken:
47+
case TokenType.GROUP_ACCESS_TOKEN:
4848
gl_rest_objects = gl.groups.list(all=True)
4949
first_message_word='Group'
50-
case TokenType.ProjectAccessToken:
50+
case TokenType.PROJECT_ACCESS_TOKEN:
5151
gl_rest_objects = gl.projects.list(all=True)
5252
first_message_word='Project'
5353

0 commit comments

Comments
 (0)