python 使用自定义环境值启动TouchDesigner

Posted 2021-05-08

import os

def Paths(current_dir, toe_file):
	start_file = os.path.join(current_dir, toe_file)
	return start_file

def Launch(**kwargs):
	envVar = kwargs.get('envVar')
	envVal = kwargs.get('envVal')
	start_file = kwargs.get('start_file')

	os.environ[envVar] = envVal
	os.startfile(start_file)

	msg = '''
	STARTING -------------------
	File 			| {file}
	Env Variable	| {var}
	Env Value		| {val}
	'''.format(file = start_file, var = envVar, val = envVal)

	return msg

def Start_up(**kwargs):
	current_dir =	kwargs.get('current_dir')
	toe_file = kwargs.get('toe_file')
	launch_args = kwargs.get('launch_args')

	start_file = Paths(current_dir, toe_file)

	for each_set in launch_args:
		envVar = each_set.get('envVar')
		envVal = each_set.get('envVal')
		
		launch_event = Launch(envVar=envVar, envVal=envVal, start_file=start_file)
		print(launch_event)

import Launcher
import os

current_dir = os.path.dirname(os.path.abspath(__file__))
toe_file = 'startup.toe'

launch_args =[
	{
		"envVar" : 'STARTUP',
		"envVal" : 'controller_1'
	},
	{
		"envVar" : 'STARTUP',
		"envVal" : 'node_1'
	}
]

Launcher.Start_up(current_dir=current_dir, toe_file=toe_file, launch_args=launch_args)

'''
Kill process in case
'''

import psutil

#for proc in psutil.process_iter():
#    if(proc.name() == 'TouchDesigner099.exe'):
#        proc.kill()
#    print(proc.name())
    
def killList(kill_list):
    '''Process kill function'''    
    for proc in psutil.process_iter():
        # check whether the process name matches
        # print(proc.name())
        if any(procstr in proc.name() for procstr in kill_list):
            print('Killing {}'.format(proc.name()))
            proc.kill()
            
def kill_by_process_name(name):
    for proc in psutil.process_iter():
        if proc.name() == name:
            print("Killing process: " + name)
            if(check_process_exist_by_name(name)):
                print("Killing process: " + name + " sucess")
            else:
                print("Killing process: " + name + " failed")
            return
 
    print("Not found process: " + name)
 
def check_process_exist_by_name(name):
    for proc in psutil.process_iter():
        if proc.name() == name:
            return True
    return False
            
killList(['TouchDesigner099.exe', 'Photoshop.exe'])