Skip to content

Commit 39f8c72

Browse files
authored
Update rotator.py
1 parent 8e80bcb commit 39f8c72

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

rotator.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from datetime import datetime as dt
2+
from dateutil.relativedelta import relativedelta
23
import enum
34
import gitlab
45
import requests
@@ -56,10 +57,11 @@ def check_and_rotate_tokens(gitlab_url: str, gitlab_token : str, token_type: Tok
5657
for gl_rest_object_access_token in gl_rest_objects_access_tokens:
5758
parsed_expired_at_date = dt.strptime(gl_rest_object_access_token.expires_at, "%Y-%m-%d")
5859
diff_date = abs((parsed_expired_at_date - parsed_today_date).days)
59-
message = f"{first_message_word} access token {gl_rest_object_access_token.name} from {first_message_word.lower()} {gl_rest_object_name} expires at {gl_rest_object_access_token.expires_at}"
60+
message = f"{first_message_word} access token {gl_rest_object_access_token.name} from {first_message_word.lower()} {gl_rest_object_name} expires at {gl_rest_object_access_token.expires_at}. Rotating..."
6061
print(message)
61-
if parsed_expired_at_date < parsed_today_date or diff_date <= 7:
62-
gl_rest_object_access_token.rotate()
62+
if parsed_expired_at_date < parsed_today_date or diff_date <= 30:
63+
new_token_expires_at = (parsed_today_date + relativedelta(years=1)).strftime('%Y-%m-%d')
64+
gl_rest_object_access_token.rotate(expires_at=new_token_expires_at)
6365
debug_message += message
6466
debug_message += "\n"
6567

0 commit comments

Comments
 (0)