【Python实例】用脚本自动拿一个或多个仿真环境

注1:之前使用的是perl,现在尝试切换到python;

注2:该脚本用于实现自动拿仿真环境,里面应该还有很多不足之处,后续逐渐完善;

注3:假设脚本名字为get_env.py,直接执行脚本,会有两次输入,分别为ll和hl,如果ll=hl,比如均为1,那么最终生成env_1环境;如果hl>ll,比如ll=1,hl=3,则最终生成env_1, env_2, env_3;

资料参考:

(1) https://www.cnblogs.com/heibaimao123/p/16294111.html (可以使用python命令行参数对脚本进行升级)

注1:其实不是很麻烦,在实例脚本中,import sys模块,并且添加ll=int(sys.argv[1])与hl=int(sys.argv[2]);

(2) https://blog.csdn.net/lx_ros/article/details/122811361 (推荐阅读)

实例

注1:11行下面漏掉两行内容,分别为"ll=int("pls enter low limit:")"与"hl=int("pls enter high limit:")";

#/usr/bin/python3

import os

#设置仿真环境放置的目录位置,该处在脚本中写死,也可以通过微调脚本在仿真命令行输入目录位置;
main_work_dir="***"
print(main_work_dir)

#打印信息,说明接下来将会创建env,env的名字是env_*,*是ll到hl之间的数字;
#ll为low limit, hl为high limit,后续会需要user在命令行输入;
print("env will be created with name-env_*, * is the number from ll to hl!")

#为了实现打印彩色输出信息而添加的内容;
class bcolors:
    OK             ='\033[92m'
    WARNING  ='\033[93m'
    ERROR       ='\033[91m'
    END           ='\033[0m'
    BOLD         ='\033[1m'

#函数实现跳转到输入参数指定的路径,并打印当前路径;
def chdir_print_dir(work_dir):
    os.chdir(work_dir)
    work_dir_tmp=os.getcwd()
    print(work_dir_tmp)

#函数实现根据输入参数info_type的不同进行不同色彩的打印
def colored_print(user_str, info_type):
    if(info_type=="OK"):
        start_str=bcolors.OK+bcolors.BOLD
        end_str  =bcolors.END
     elif(info_type=="WARNING"):
        start_str=bcolors.WARNING+bcolors.BOLD
        end_str  =bcolors.END
     elif(info_type=="ERROR"):
        start_str=bcolors.ERROR+bcolors.BOLD
        end_str  =bcolors.END
     else:
         raise Exception("info_type %s is not supported"%info_type)  

for num in range(ll, hl+1):
    chdir_print_dir(main_work_dir)
    #下行的***需要用实际的git repo代替
    os.system("git clone ***"+"env_"+str(num))
    colored_print("step1: git clone done!!","OK")
    chdir_print_dir(main_work_dir+"env_"+str(num))
    #下行根据实际仿真环境的获取而有所不同;
    os.system("git submodule update --recursive --int")
    colored_print("step2: git submodule udpate done!!", "OK")
    chdir_print_dir(main_work_dir)
    colored_print("step3: env-env_%d is created done!"%num,"OK")