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+'"{
'
            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("
")
    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('20140803'):
        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')