GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-26 19:00:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor loading .env file

Browse Source
This commit is contained in:
debugtalk
2018-08-07 15:06:08 +08:00
parent aaac5ba323
commit 167fc9cc02
5 changed files with 34 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
httprunner/loader.py
View File

@@ -138,24 +138,31 @@ def load_folder_files(folder_path, recursive=True):
def load_dot_env_file(path):
""" load .env file and set to os.environ
""" load .env file
"""
if not path:
path = os.path.join(os.getcwd(), ".env")
if not os.path.isfile(path):
logger.log_debug(".env file not exist: {}".format(path))
return
return {}
else:
if not os.path.isfile(path):
raise exceptions.FileNotFound("env file not exist: {}".format(path))
logger.log_info("Loading environment variables from {}".format(path))
env_variables_mapping = {}
with io.open(path, 'r', encoding='utf-8') as fp:
for line in fp:
variable, value = line.split("=")
variable = variable.strip()
os.environ[variable] = value.strip()
logger.log_debug("Loaded variable: {}".format(variable))
if "=" in line:
variable, value = line.split("=")
elif ":" in line:
variable, value = line.split(":")
else:
raise exceptions.FileFormatError(".env format error")
env_variables_mapping[variable.strip()] = value.strip()
return env_variables_mapping
###############################################################################