mirror of https://github.com/bmaltais/kohya_ss
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import subprocess
|
|
import psutil
|
|
from library.custom_logging import setup_logging
|
|
|
|
# Set up logging
|
|
log = setup_logging()
|
|
|
|
|
|
class CommandExecutor:
|
|
def __init__(self):
|
|
self.process = None
|
|
|
|
def execute_command(self, run_cmd):
|
|
if self.process and self.process.poll() is None:
|
|
log.info(
|
|
'The command is already running. Please wait for it to finish.'
|
|
)
|
|
else:
|
|
self.process = subprocess.Popen(run_cmd, shell=True)
|
|
|
|
def kill_command(self):
|
|
if self.process and self.process.poll() is None:
|
|
try:
|
|
parent = psutil.Process(self.process.pid)
|
|
for child in parent.children(recursive=True):
|
|
child.kill()
|
|
parent.kill()
|
|
log.info('The running process has been terminated.')
|
|
except psutil.NoSuchProcess:
|
|
log.info('The process does not exist.')
|
|
except Exception as e:
|
|
log.info(f'Error when terminating process: {e}')
|
|
else:
|
|
log.info('There is no running process to kill.')
|