Skip to content

Commit 2b88080

Browse files
authored
api_client: create from_env factory (#16)
1 parent 54a1cc0 commit 2b88080

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

flareio/api_client.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import requests
23

34
from datetime import datetime
@@ -31,6 +32,17 @@ def __init__(
3132
self._api_token_exp: t.Optional[datetime] = None
3233
self._session = session or self._create_session()
3334

35+
@classmethod
36+
def from_env(cls) -> "FlareApiClient":
37+
api_key: t.Optional[str] = os.environ.get("FLARE_API_KEY")
38+
if not api_key:
39+
raise Exception("Please set the FLARE_API_KEY environment variable")
40+
tenant_id: t.Optional[str] = os.environ.get("FLARE_TENANT_ID")
41+
return cls(
42+
api_key=api_key,
43+
tenant_id=int(tenant_id) if tenant_id is not None else None,
44+
)
45+
3446
@staticmethod
3547
def _create_session() -> requests.Session:
3648
session = requests.Session()

0 commit comments

Comments
 (0)