Skip to content

Commit a8a951c

Browse files
gflarityGeoff Flarity
andauthored
[CLI] fix credential file format (#82)
* standardize on firebase's credential format * support for .env files for development * lint fix --------- Co-authored-by: Geoff Flarity <[email protected]>
1 parent 46e2af2 commit a8a951c

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

centml/sdk/auth.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def store_centml_cred(token_file):
2828
try:
2929
with open(token_file, 'r') as f:
3030
os.makedirs(settings.CENTML_CONFIG_PATH, exist_ok=True)
31-
refresh_token = json.load(f)["refreshToken"]
31+
refresh_token = json.load(f)["refresh_token"]
3232

3333
refresh_centml_token(refresh_token)
3434
except Exception:
@@ -51,12 +51,12 @@ def get_centml_token():
5151
if not cred:
5252
sys.exit("CentML credentials not found. Please login...")
5353

54-
exp_time = int(jwt.decode(cred["idToken"], options={"verify_signature": False})["exp"])
54+
exp_time = int(jwt.decode(cred["id_token"], options={"verify_signature": False})["exp"])
5555

5656
if time.time() >= exp_time - 100:
57-
cred = refresh_centml_token(cred["refreshToken"])
57+
cred = refresh_centml_token(cred["refresh_token"])
5858

59-
return cred["idToken"]
59+
return cred["id_token"]
6060

6161

6262
def remove_centml_cred():

centml/sdk/config.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,13 @@
11
import os
2-
from pydantic_settings import BaseSettings
2+
from pathlib import Path
3+
from pydantic_settings import BaseSettings, SettingsConfigDict
34

45

56
class Config(BaseSettings):
7+
8+
# It is possible to override the default values by setting the environment variables
9+
model_config = SettingsConfigDict(env_file=Path('.env'))
10+
611
CENTML_WEB_URL: str = "https://app.centml.com/"
712
CENTML_CONFIG_PATH: str = os.path.expanduser("~/.centml")
813
CENTML_CRED_FILE: str = "credentials.json"

0 commit comments

Comments
 (0)