Skip to content

Commit edf9e87

Browse files
committed
update init token setup
1 parent 812a3c7 commit edf9e87

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

src/datasmith/__init__.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
import os
2+
import sys
23

3-
with open("tokens.env", encoding="utf-8") as f:
4-
lines = f.readlines()
5-
tokens = {line.split("=")[0].strip(): line.split("=")[1].strip() for line in lines if "=" in line}
6-
os.environ.update(tokens)
4+
if os.path.exists("tokens.env"):
5+
with open("tokens.env", encoding="utf-8") as f:
6+
lines = f.readlines()
7+
tokens = {line.split("=")[0].strip(): line.split("=")[1].strip() for line in lines if "=" in line}
8+
os.environ.update(tokens)
9+
else:
10+
sys.stderr.write("No tokens.env file found. Skipping environment variable setup.\n")

0 commit comments

Comments
 (0)