51Testing软件测试论坛

标题: 如何在Python项目中读取写入配置？ [打印本页]
作者: lsekfe    时间: 2022-9-8 13:46
标题: 如何在Python项目中读取写入配置？
　1、将配置写在Python文件中
　　这种方法非常简单，但它存在严重的安全问题，我们都知道不应该把配置写在代码中，如果有人把我们的源代码上传到了github中，那么数据库的配置就相当于向全世界公开了，当然，当配置文件不包含敏感信息时，也可以才有这种简单的方法。
　　2、利用外部配置文件
　　让配置文件和代码独立开来，通常使用 json、yaml或者ini的文件格式来存储配置。
　　结合环境变量和 python 库读取外部文件，首先开发通常不会接触到生成环境，因此生成环境的配置文件由运维来写，运维将应用所需要的配置写好后，放到生产服务器的指定位置，代码从指定位置读取配置。
　　为了方便程序的统一调试，可以提前约定好一个系统环境变量（XXX_CONFIG_PATH）来指定配置文件的存放路径。
  1. 　例如：export XXX_CONFIG_PATH =
  2. 　　/home/test/configs/config.ini 此为设置临时环境变量
复制代码
linux、ubuntu 环境变量
查看环境变量：
　　 env
　　 设置永久环境变量
　　 1.在/etc/profile 的文件下编辑，所改变的环境变量是面向所有用户的
　　 export CLASSPATH = /../...  该路径为绝对路径
　　 2.在当前用户目录下./barsh_profile文件中修改 进行修改的话，仅对当前的用户生效
　　 vim /home/wens/.barshc
　　 export CLASSPATH = /../...  该路径为绝对路径
　　 最后使用source命令 可以直接使环境变量生效
　　 source  /home/wens/.barshc //直接跟环境变量的文件

windows 环境变量
查看环境变量：
　　 set
　　 查看某个环境变量：
　　 set path
　　 修改环境变量
　　 输入 “set 变量名=变量内容”即可。比如将path设置为“d:\nmake.exe”，只要输入set path="d:\nmake.exe"
　　 注意：所有的在cmd命令行下对环境变量的修改只对当前窗口有效，不是永久性的修改。也就是说当关闭此cmd命令行窗口后，将不再起作用。
　　 永久性修改环境变量的方法有两种：
　　   一种是直接修改注册表
　　   另一种是通过我的电脑-〉属性-〉高级，来设置系统的环境变量（查看详细）   
　　 设置了环境变量后，需要重启 pycharm 生效

3、直接使用系统环境变量读取配置
　　此方法不使用文件来存储配置信息，将所有的配置信息存储到环境变量中，在实践中比较常见，运维通过 ansible 部署脚本，在程序运行前将需要配置的信息导入到环境变量中。
　　不利用文件存储，在一定程度上加强了对密码等配置信息的保护，但也增加了运维的工作量，尤其是当需要修改配置时。
　　4、微服务架构
　　在一些微服务架构中，会专门开发配置中心，程序直接从线上读取配置，配置的管理也会开发一套GUI，方便开发和运维。
　　5、一般项目中建议的配置方式
-app
　　  -__init.py
　　  -app.py
　　  -settings
　　   -__init__.py
　　   -base.py
　　   -dev.py
　　   -prod.py

其中 __init__.py 中，添加判断逻辑，判断当前环境使用 开发环境还是生产环境，从而加载不同的配置参数。
  1. 　# settings/__init__.py
  2. 　　 import os
  3. 　　 # os.environ.get() 用于获取系统中的环境变量，因为在生产环境中，一般都会把一些关键性的参数写到系统的环境中。
  4. 　　 # 所以PROFILE的值其实就是我们配置的环境变量的值。如果没有配这个值，默认走dev的配置。
  5. 　　 # PYTHON_PRO_PROFILE = os.environ.get("PYTHON_PRO_PROFILE", "dev")
  6. 　　 PYTHON_PRO_PROFILE = os.environ.get("PYTHON_PRO_PROFILE")
  7. 　　 print("是开发环境还是生产环境: ", PYTHON_PRO_PROFILE)
  8. 　　 if PYTHON_PRO_PROFILE == "dev":
  9. 　　     from .dev import *
  10. 　　 elif PYTHON_PRO_PROFILE == "prod":
  11. 　　     from .prod import *
  12. 　　 else:
  13. 　　     raise Exception("Not supported runtime profile {}".format(PYTHON_PRO_PROFILE))
复制代码
其中 base.py 存储一些公用的配置，然后在开发环境 dev.py 与生产环境 prod.py 中导入 base.py 的变量。
  1. # settings/base.py
  2. 　　 import os
  3. 　　 import time
  4. 　　 # os.path.abspath: 获取完整路径(包含文件名)
  5. 　　 current_exec_abspath = os.path.abspath(__file__)
  6. 　　 current_exec_dir_name, _ = os.path.split(current_exec_abspath)
  7. 　　 current_up_dir, _ = os.path.split(current_exec_dir_name)
  8. 　　 current_up2_dir, _ = os.path.split(current_up_dir)
  9. 　　 print('------log dir=------', current_up2_dir)
  10. 　　 # 日志文件路径设置
  11. 　　 log_path = f"{current_up2_dir}/logs"
  12. 　　 if not os.path.exists(log_path):
  13. 　　     os.makedirs(log_path)
  14. 　　 t = time.strftime("%Y_%m_%d")
  15. 　　 log_path_file = f"{log_path}/interface_log_{t}.log"
复制代码
其中 dev.py：
  1. # 导入了base下所有参数
  2. 　　 from .base import *
  3. 　　 database = {
  4. 　　     "protocol": "mysql+mysqlconnector",
  5. 　　     "username": "xxxxxx",
  6. 　　     "password": "hash string",
  7. 　　     "port": 3306,
  8. 　　     "database": "repo"
  9. 　　 }
复制代码
其中 prod.py：
  1. # 导入了base下所有参数
  2. 　　 from .base import *
  3. 　　 database = {
  4. 　　     "protocol": "xxxxxxxxxxx",
  5. 　　     "username": "xxxxxxxxxxx",
  6. 　　     "password": "xxxxxxxxxxx",
  7. 　　     "port": 3344,
  8. 　　     "database": "xxxx"
  9. 　　 }
  10. 　　 对于一些敏感信息可在环境变量里设置，通过如下方法获取，例如：
  11. 　　 MAIL_SERVER = os.environ.get('MAIL_SERVER', 'smtp.163.com')
  12. 　　 MAIL_USERNAME = os.environ.get('MAIL_USERNAME') or 'test'
  13. 　　 MAIL_PASSWORD = os.environ.get('MAIL_PASSWORD') or '12345678'
复制代码








欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2