您的位置:首页 > 编程语言 > Python开发

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

2016-10-24 21:22 706 查看
具体查看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()
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: