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

具体查看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()