python拓展应用：运行do文件及其衍生内容

 #-*-coding:utf8-*-
 import os
 import subprocess
 import shutil
 import psutil
 import time

 #查找指定文件名称和查找范围的文件路径
 def search(name, path):
     name = name.lower()
     for root,dirs,files in os.walk(path):
         for file in files+dirs:
             if name in file.lower():
                 return os.path.join(root,file)
     return -1

 #修改do文件内容
 def do(in_file):
     w = ''
     for line in open('sort.do','r').readlines():
         if 'foreach' in line:
             ceil = line.split(" ")
             ceil[len(ceil)-1] = '"'+in_file+'"{\n'
             w = ""
             for i in ceil:
                 w = w + i + ' '
         else:
             w += line

     with open('sort.do','wb') as writer:
         writer.write(w)

 #运行指定的其他类型的程序，如do文件
 def run_do(do_file,in_file):

     do(in_file)

     # 查找windows所有盘符名称
     information = os.popen("wmic LOGICALDISK get name").read().split("\r\n")
     code = []
     for inform in information:
         if ':' in inform:
             code.append(inform.strip())
     print code

     # 遍历所有盘符，直到找到StataMP-64.exe文件
     for path in code:
         stata_path = search('StataMP-64.exe', path + '\\')
         if stata_path != -1:
             break
     print stata_path

     if os.path.exists(''):
         pass
         print '已存在'
     else:
         do_file = os.path.abspath(do_file + '.do')
         cmd = [stata_path,'do',do_file]
         child = subprocess.Popen(cmd)
         #找出本机逻辑核个数
         cpu_count = psutil.cpu_count()
         while True:
             time.sleep(5)
             # 传入进程PID,实现监控功能
             cpu_percent = psutil.Process(child.pid).cpu_percent(interval=2)
             percent = cpu_percent/cpu_count
             print cpu_count,cpu_percent,percent
             if percent == 0:
                 child.kill()
                 print 'Success'
                 break

 if __name__ == '__main__':

     run_do('sort')