python 从windows获取环境变量:os.environ

paopao00 2020-06-06

问题场景:
写脚本涉及到数据库的配置信息,然后上面要求这种关键信息不能体现到脚本中,哦,那怎么办,那我去哪里读取数据库信息呢?

解决方法:
从运行脚本的机器的环境变量里读取数据库信息,这样脚本中全是代码,就没有明文配置信息了。

具体就是:
1、先在自己的windows机器上配置好环境变量(命令行设置方法之前已讲过)。
2、在脚本中引入 value = os.environ.get(key)获取环境变量值。

代码演示:

# 从环境变量中获取数据库配置
user = os.environ.get(USERNAME)
pwd = os.environ.get(PWD)
host = os.environ.get(HOST)
port = os.environ.get(PORT)
dbName = os.environ.get(DBNAME)

# 获取环境变量的所有key
keys = os.environ.keys()  # 
values = os.environ.values()
print(type(keys))  # <class ‘collections.abc.KeysView‘>
pritn(list(keys))
print(list(values))

相关推荐