python脚本中如何设置系统的环境变量

最新推荐文章于 2024-11-13 17:21:27 发布

double12gzh

最新推荐文章于 2024-11-13 17:21:27 发布

阅读量8.4k

点赞数

分类专栏： Python 文章标签： py python 脚本

个人主页: http://gzh.readthedocs.io; 微信公众号: double12gzh

本文链接：https://blog.youkuaiyun.com/gzhouc/article/details/52915822

版权

Python 专栏收录该内容

31 篇文章

订阅专栏

本文介绍如何在Python脚本中操作系统的环境变量，详细示例代码可在GitHub的PythonLearning仓库查阅。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

具体查看github https://github.com/double12gzh/PythonLearning/blob/master/setup_env_in_script.py

#!/usr/bin/python

####################################################################
# Author:      Jeffrey Guan
# Date:        2016-10-24
# Description: set the environment parameters in python script.
#              For example, if we want to run "nova list" in python 
#              script before "source openrc" in CLI, we can safely 
#              use this script. 
#
#              A part of "/root/openrc" is:
#               #!/bin/sh
#               export LC_ALL=C
#               export OS_NO_CACHE='true'
#               export OS_TENANT_NAME='admin'
#               export OS_PROJECT_NAME='admin'
#               export OS_USERNAME='admin'
#               export OS_PASSWORD='admin'
#####################################################################

import os
import re

# Open the file.
file_obj = open('/root/openrc')

try:
  # Search the str started with "export" and contains "=".
  patt_save_str = re.compile(r'^export.*=.*')
  # Search "=".
  patt_rm_str = re.compile(r'=')

  # Read file content by lines.
  lines = file_obj.readlines()

  for line in lines:
    match = patt_save_str.search(line)
    if match:
      # Remove the "export" and "" from match.group(0).
      temp_str = match.group(0).strip("export").strip()

      # Split the str into a list by "=".
      environ_value_dic = patt_rm_str.split(temp_str)

      # Set the value for each env parameters.
      os.environ[environ_value_dic[0]] = environ_value_dic[1].strip("'")

  # Print the env and values. Or we can run "env" on the CLI to check 
  # all env and values.
  #
  #print os.getenv('OS_PROJECT_NAME')

  # Test nova command.
  os.system('nova list')

finally:
  file_obj.close()