def main():
os.umask(DEFAULT_UMASK)
# Need to call setup_console_logging like this as this module is always called from another.
setup_console_logging("__main__")
with tempfile.NamedTemporaryFile(delete=True) as submission_tmp_file:
from firexapp.submit.submit import SubmitBaseApp
submit_app = SubmitBaseApp(
submission_tmp_file=submission_tmp_file.name)
app = FireXBaseApp(submit_app=submit_app)
ExitSignalHandler(app)
app.run()
import os
from psutil import Process, TimeoutExpired
import subprocess
from firexapp.submit.tracking_service import TrackingService
from firexapp.common import qualify_firex_bin
from firexapp.submit.console import setup_console_logging
from firex_keeper.keeper_helper import get_keeper_dir
from firexapp.discovery import PkgVersionInfo
logger = setup_console_logging(__name__)
class FireXKeeperLauncher(TrackingService):
def __init__(self):
self.broker_recv_ready_file = None
def start(self, args, uid=None, **kwargs) -> {}:
keeper_debug_dir = get_keeper_dir(uid.logs_dir)
os.makedirs(keeper_debug_dir, exist_ok=True)
self.broker_recv_ready_file = os.path.join(
keeper_debug_dir, 'keeper_celery_recvr_ready')
stdout_file = os.path.join(keeper_debug_dir, 'keeper.stdout.txt')
cmd = [
qualify_firex_bin("firex_keeper"),
"--uid",
str(uid),
"--logs_dir",
uid.logs_dir,