1. 问题

在自动化应用的时候 ,有时候环境变量与运行需要不一致。这时候有两种选择:

  1. 改变节点环境变量,使得其和运行需求保持一致;
  2. 在自动化脚本中设置环境变量,其范围只在脚本运行环境中有效。

显然,当需要运行多个自动化脚本,每个的需求不一致的时候,选择2是更好的设计。

2. 解决方案和案例

假设我们有两个java的运行版本,一个是本节点上的java,版本7;另一个是云盘上共享的版本8。我们需要运行elasticsearch 6.2.3,它需要java8。而我们不希望更新节点上的java:

  • java 8路径:/shared/java8/bin
  • elasticsearch 6.2.3路径:/opt/elasticsearch6.2.3/bin
  •  elasticsearch用户:elasticsearch
import subprocess, re, os, socket, resource, pwd, crypt, logging, sys, platform

def demote(user_uid, user_gid):
def result():
os.setgid(user_gid)
os.setuid(user_uid)
return result class setuputil:
@staticmethod
def run_command(args, username=None, usercwd=None):
try:
if username:
pw_record = pwd.getpwnam(username)
env = os.environ.copy()
env['HOME'] = pw_record.pw_dir
env['LOGNAME'] = username
env['USER'] = username
if usercwd:
env['PWD'] = usercwd
p = subprocess.Popen(args, preexec_fn=demote(pw_record.pw_uid, pw_record.pw_gid),
cwd=usercwd, env=env, close_fds=True)
else:
p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE, close_fds = True) outputs = p.communicate()
for out in outputs:
if out:
return (p.returncode, out)
except OSError as os_e:
return (os_e.errno, os_e.strerror)
except ValueError as val_e:
return (val_e.errno, val_e.strerror)
else:
return (p.returncode, outputs) @staticmethod
def java_version():
java_ver_pattern = '\"(\d+\.\d+).*\"'
java_cmd = "java"
if 'JAVA_HOME' in os.environ:
java_cmd = os.environ['JAVA_HOME'] + "/bin/java"
version = subprocess.check_output([java_cmd, '-version'], stderr=subprocess.STDOUT)
if version:
return re.search(java_ver_pattern, version).groups()[0]
else:
return None @staticmethod
def start_app(es_exec, pid_file, es_user):
rtcode, msg = setuputil.run_command([es_exec, '-d', '-p', pid_file], es_user)
if rtcode:
return (rtcode, msg)
else:
return (0, "OK") MIN_JAVA_VERSION = 1.8

ELASTIC_USER = 'elastic'
ELASTIC_PASSWORD = 'elastic'
ELASTIC_PID = '/tmp/elasticpid' def main():
# set up java home
os.environ['JAVA_HOME'] = '/shared/java8' j_ver = setuputil.java_version()
if not j_ver or float(j_ver) < MIN_JAVA_VERSION:
logging.error("Java isn't installed or java version is not up to date: %s", j_ver if j_ver else "None")
return (1, "Java isn't installed or java version is not up to date: %s"%(j_ver if j_ver else "None")) rtcode, msg = setuputil.start_app('/opt/elasticsearch6.2.3/bin/elasticsearch', ELASTIC_PID, ELASTIC_USER)
if rtcode:
logging.error("Failed to start elasticsearch: %d, %s", rtcode, msg)
return (rtcode, msg)
else:
logging.debug("Start elasticsearch: %s", msg)
return (0, None) if __name__ == '__main__':
main()

3. 注意的问题

有几个需要注意的问题:

  • 环境变量的适用范围(scope)

    • 一般只在运行的当前shell有效
    • 就上例而言,JAVA_HOME只在运行脚本的进程里是有效
  • 在哪里设置环境变量?
    • 需要在运行命令之前设置  
    • 就上例而言,JAVA_HOME需要在运行java版本检查,和运行elasticsearch之前设置

  

在python脚本中设置环境变量,并运行相关应用的更多相关文章

  1. python 程序中设置环境变量

    python 中调用系统命令有三种方法: 1.os.system('command') ,这个方法是直接调用标准C的system() 函数,仅仅在一个子终端运行系统命令,而不能获取命令执行后的返回信息 ...

  2. ant脚本中设置环境变量

    http://blog.csdn.net/quqi99/article/details/5329841

  3. 怎样在Ubuntu中设置环境变量

    首先启动终端. 单击屏幕左上角的Ubuntu图标,在弹出的窗口中点击搜索栏,输入"terminal", 稍等片刻,终端就会赫然在目!二话不说,直接点击!     然后打开环境设置文 ...

  4. Python IDLE arcpy设置环境变量

    在IDLE中 import arcpy help(arcpy) 得到的路径为: 但是在arcmap中,路径为: 说明IDLE的环境变量设置有问题: 在windows的环境变量中设置环境变量PYTHON ...

  5. Ubuntu中设置环境变量详解

    1, 为单一用户:.bashrc: 为每一个运行bash shell的用户执行此文件.当bash shell被打开时,该文件被读取.打开用户主目录下的.bashrc,在这个文件中加入export PA ...

  6. dockerfile中设置环境变量

    设置环境变量要用 ENV ENV JAVA_HOME /usr/local/jdk1.8.0_171 ENV PATH $JAVA_HOME/bin:$PATH 用 echo "JAVA_H ...

  7. Mac笔记本中是用Idea开发工具在Java项目中调用python脚本遇到的环境变量问题解决

    问题描述: mac笔记本本身会自带几个python版本,比如python2.7版本,我没有改动mac默认的python版本,只是安装了python3.7版本. 使用Pycharm开发Python项目没 ...

  8. 如何在pycharm中设置环境变量

    今天运行tensorflow的时候,发现在pycharm下,程序无法找到CUDA的libcupti.so文件.而在添加完环境变量: export LD_LIBRARY_PATH=$LD_LIBRARY ...

  9. MySQL在Windows中设置环境变量

    在桌面选择“计算机”的图标(或者是我的电脑),右键-->属性-->点击“高级系统设置” 然后选择 高级  选项点击环境变量 然后点击新建-> 变量名为MYSQL_HOME 变量值为m ...

随机推荐

  1. 七:Java之封装、抽象、多态和继承

    本文章介绍了关于Java中的面向对象封装.抽象.继承.多态特点 Java面向对象主要有四大特性:封装.抽象.继承和多态. 一.封装 封装就是将抽象得到的数据和行为(或功能)相结合,形成一个有机的总体, ...

  2. playbook-拓展

    一.Handles 1. Handlers 在发生改变时执行的操作 2. handlers也是一些task的列表,通过名字来引用,它们和一般的task并没有什么区别. 3. handlers是由not ...

  3. 20155314 2016-2017-2 《Java程序设计》第4周学习总结

    20155314 2016-2017-2 <Java程序设计>第4周学习总结 教材学习内容总结 理解封装.继承.多态的关系 理解抽象类与接口的区别 掌握S.O.L.I.D原则 了解模式和设 ...

  4. Sequelize-nodejs-11-Raw queries

    Raw queries原始查询 就是使用了原始的查询语句,如UPDATE users SET y = 42 WHERE x = 12 As there are often use cases in w ...

  5. Srv数据格式

    1.简介 类似msg文件, srv文件是用来描述服务( service数据类型的, service通信的数据格式定义在*.srv中. 它声明了一个服务, 包括请求(request)和响应( reply ...

  6. CAN--UART的协议转换器

    CAN--UART的协议转换器 //------------------------------------------------------//  CAN <==> UART的协议转换 ...

  7. Kafka集群配置

    kafka_2.11-0.9.0.1.tgz 1.进入项目前的目录 cd /home/dongshanxia mkdir kafka #创建项目目录 cd kafka #进入项目目录 mkdir ka ...

  8. 一、Delphi中Cxgrid表格滚动条粗细设置

    1.Delphi VCL新版本的Cxgrid滚动条默认是触屏模式(如下图),很细的滚动条,在电脑版显示非常不方便. 2.如果需要改成传统的滚动条模式,需要设置一下LookAndFeel里面的Scrol ...

  9. Linux下onvif客户端获取h265 IPC摄像头的RTSP地址

    1. 设备搜索,去获取webserver 的地址 ,目的是在获取能力提供服务地址,demo:https://www.cnblogs.com/croxd/p/10683429.html 2. GetCa ...

  10. 尝试用docker上的jenkins

    比起dockerfile的编写,我更习惯使用简单的docker-compose.yml.但是官方的docker-compose.yml并不是独立的.于是经过一番研究,暂时打造了自己的文件: docke ...