-
Notifications
You must be signed in to change notification settings - Fork 35
Expand file tree
/
Copy pathinit_env.py
More file actions
34 lines (27 loc) · 1.12 KB
/
init_env.py
File metadata and controls
34 lines (27 loc) · 1.12 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import os
import json
import configparser
from hdbcli import dbapi
import variables
ROOT_PATH_DIR = os.path.dirname(os.getcwd())
AICORE_CONFIG_FILENAME = '.aicore-config.json'
USER_CONFIG_FILENAME = '.user.ini'
def set_environment_variables() -> None:
with open(os.path.join(ROOT_PATH_DIR, AICORE_CONFIG_FILENAME), 'r') as config_file:
config_data = json.load(config_file)
os.environ["AICORE_AUTH_URL"]=config_data["url"]+"/oauth/token"
os.environ["AICORE_CLIENT_ID"]=config_data["clientid"]
os.environ["AICORE_CLIENT_SECRET"]=config_data["clientsecret"]
os.environ["AICORE_BASE_URL"]=config_data["serviceurls"]["AI_API_URL"]
os.environ["AICORE_RESOURCE_GROUP"]=variables.RESOURCE_GROUP
def connect_to_hana_db() -> dbapi.Connection:
config = configparser.ConfigParser()
config.read(os.path.join(ROOT_PATH_DIR, USER_CONFIG_FILENAME))
return dbapi.connect(
address=config.get('hana', 'url'),
port=config.get('hana', 'port'),
user=config.get('hana', 'user'),
password=config.get('hana', 'passwd'),
autocommit=True,
sslValidateCertificate=False
)