class Executor:
__control_dict = {
Sections.EXECUTOR_DATA: {
"cmd": control_str(True),
"repo_executor": control_str(True),
"max_size": control_int(True),
}
}
def __init__(self, name: str, config):
name = name.strip()
self.control_config(name, config)
self.name = name
executor_section = Sections.EXECUTOR_DATA.format(name)
params_section = Sections.EXECUTOR_PARAMS.format(name)
varenvs_section = Sections.EXECUTOR_VARENVS.format(name)
self.repo_name = config[executor_section].get("repo_executor", None)
if self.repo_name:
metadata = executor_metadata(self.repo_name)
repo_path = executor_folder() / self.repo_name
self.cmd = metadata["cmd"].format(EXECUTOR_FILE_PATH=repo_path)
else:
self.cmd = config[executor_section].get("cmd")
self.max_size = int(config[executor_section].get("max_size", 64 * 1024))
self.params = dict(config[params_section]) if params_section in config else {}
self.params = {key: value.lower() in ["t", "true"] for key, value in self.params.items()}
self.varenvs = dict(config[varenvs_section]) if varenvs_section in config else {}
def control_config(self, name, config):
if " " in name:
raise ValueError("Executor names can't contains space character, passed name:" f"{name}")
if Sections.EXECUTOR_DATA.format(name) not in config:
raise ValueError(f"{name} is an executor name but there is no proper section")
for section in self.__control_dict:
for option in self.__control_dict[section]:
value = config.get(section.format(name), option) if option in config[section.format(name)] else None
self.__control_dict[section][option](option, value)
params_section = Sections.EXECUTOR_PARAMS.format(name)
if params_section in config:
for option in config[params_section]:
value = config.get(params_section, option)
control_bool(option, value)
async def check_cmds(self):
if self.repo_name is None:
return True
metadata = executor_metadata(self.repo_name)
if not await check_commands(metadata):
logger.info(
f"{Bcolors.WARNING}Invalid bash dependency for " f"{Bcolors.BOLD}{self.repo_name}{Bcolors.ENDC}"
)
return False
else:
return True
class Executor:
__control_dict = {
"cmd": control_str(True),
"repo_executor": control_str(True),
"max_size": control_int(True),
}
def __init__(self, name: str, config):
name = name.strip()
self.control_config(name, config)
self.name = name
self.repo_executor = config.get("repo_executor")
if self.repo_executor:
self.repo_name = re.search(r"(^[a-zA-Z0-9_-]+)(?:\..*)*$",
self.repo_executor).group(1)
metadata = executor_metadata(self.repo_name)
repo_path = executor_folder() / self.repo_executor
self.cmd = metadata["cmd"].format(EXECUTOR_FILE_PATH=repo_path)
else:
self.cmd = config.get("cmd")
self.max_size = int(config.get("max_size", 64 * 1024))
self.params = dict(config[Sections.EXECUTOR_PARAMS]
) if Sections.EXECUTOR_PARAMS in config else {}
self.varenvs = dict(config[Sections.EXECUTOR_VARENVS]
) if Sections.EXECUTOR_VARENVS in config else {}
def control_config(self, name, config):
if " " in name:
raise ValueError(
"Executor names can't contains space character, passed name:"
f"{name}")
for option in self.__control_dict:
value = config[option] if option in config else None
self.__control_dict[option](option, value)
if Sections.EXECUTOR_PARAMS in config:
value = config.get(Sections.EXECUTOR_PARAMS)
errors = ParamsSchema().validate({"params": value})
if errors:
raise ValueError(errors)
async def check_cmds(self):
if self.repo_executor is None:
return True
repo_name = re.search(r"(^[a-zA-Z0-9_-]+)(?:\..*)*$",
self.repo_executor).group(1)
metadata = executor_metadata(repo_name)
if not await check_commands(metadata):
logger.info(f"{Bcolors.WARNING}Invalid bash dependency for "
f"{Bcolors.BOLD}{self.repo_name}{Bcolors.ENDC}")
return False
else:
return True
class Executor:
__control_dict = {
Sections.EXECUTOR_DATA: {
"cmd": control_str(),
"max_size": control_int(True)
}
}
def __init__(self, name: str, config):
name = name.strip()
self.control_config(name, config)
self.name = name
executor_section = Sections.EXECUTOR_DATA.format(name)
params_section = Sections.EXECUTOR_PARAMS.format(name)
varenvs_section = Sections.EXECUTOR_VARENVS.format(name)
self.cmd = config.get(executor_section, "cmd")
self.max_size = int(config[executor_section].get(
"max_size", 64 * 1024))
self.params = dict(
config[params_section]) if params_section in config else {}
self.params = {
key: value.lower() in ["t", "true"]
for key, value in self.params.items()
}
self.varenvs = dict(
config[varenvs_section]) if varenvs_section in config else {}
def control_config(self, name, config):
if " " in name:
raise ValueError(
f"Executor names can't contains space character, passed name: {name}"
)
if Sections.EXECUTOR_DATA.format(name) not in config:
raise ValueError(
f"{name} is an executor name but there is no proper section")
for section in self.__control_dict:
for option in self.__control_dict[section]:
value = config.get(
section.format(name),
option) if option in config[section.format(name)] else None
self.__control_dict[section][option](option, value)
params_section = Sections.EXECUTOR_PARAMS.format(name)
if params_section in config:
for option in config[params_section]:
value = config.get(params_section, option)
control_bool(option, value)
class Sections:
TOKENS = "tokens"
SERVER = "server"
AGENT = "agent"
EXECUTOR_VARENVS = "{}_varenvs"
EXECUTOR_PARAMS = "{}_params"
EXECUTOR_DATA = "{}"
__control_dict = {
Sections.SERVER: {
"host": control_host,
"ssl": control_bool,
"ssl_cert": control_str(nullable=True),
"api_port": control_int(),
"websocket_port": control_int(),
"workspaces": control_list(can_repeat=False),
},
Sections.TOKENS: {
"registration": control_registration_token,
"agent": control_agent_token,
},
Sections.AGENT: {
"agent_name": control_str(),
"executors": control_list(can_repeat=False),
},
}