Skip to content

Commit

Permalink
perf: log display
Browse files Browse the repository at this point in the history
  • Loading branch information
O-Jiangweidong committed Aug 1, 2024
1 parent 812948c commit a2114c3
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
8 changes: 4 additions & 4 deletions apps/behemoth/libs/pools/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,12 +159,12 @@ def mark_task_status(self, execution_id: str, status: str):
cache.set(self.execution_status_key.format(execution_id), status)

def __pre_run(self, execution: Execution) -> None:
print(p.info('%s【%s】' % (_('Looking for effective worker'), _('Start'))))
print(p.info('【%s】%s' % (_('Start'), _('Looking for effective worker'))))
self.select_org(execution.org_id)
self.refresh_all_workers()
execution.worker_id = self.__get_valid_worker(execution)
execution.save(update_fields=['worker_id'])
print(p.info('%s【%s】' % (_('Looking for effective worker'), _('End'))))
print(p.info('【%s】%s' % (_('Start'), _('Looking for effective worker'))))

@staticmethod
def __generate_command_file(commands: list, execution: Execution) -> (str, str):
Expand Down Expand Up @@ -229,7 +229,7 @@ def __build_params(self, execution: Execution) -> dict | None:
if not execution.asset or not execution.account:
raise JMSException(_('Task has no asset or account'))

print(p.info(f'%s【%s】' % (_("Building the params required for command execution"), _('Start'))))
print(p.info(f'【%s】%s' % (_('Start'), _("Building the params required for command execution"))))
remote_command_dir = f'/tmp/behemoth/commands/{execution.id}'
command_filepath: str = f'{execution.id}.bs'
local_cmds_file, cmd_file = self.__generate_command_file(commands, execution)
Expand Down Expand Up @@ -260,7 +260,7 @@ def __build_params(self, execution: Execution) -> dict | None:
})
else:
raise JMSException(f'{_("Unsupported asset type")}: {execution.asset.type}')
print(p.info(f'%s【%s】' % (_("Building the params required for command execution"), _('End'))))
print(p.info(f'【%s】%s' % (_('Start'), _("Building the params required for command execution"))))
return {
'host': settings.SITE_URL, 'cmd_type': cmd_type, 'script': script,
'auth': auth, 'token': str(token), 'task_id': str(execution.id),
Expand Down
16 changes: 8 additions & 8 deletions apps/behemoth/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ def test_connectivity(self) -> bool:

def _scp(self, local_path: str, remote_path: str, mode=0o544) -> None:
filename = os.path.basename(remote_path)
print(p.info(f'%s: %s【%s】' % (_("Upload file"), filename, _('Start'))))
print(p.info(f'【%s】 %s: %s' % (_('Start'), _("Upload file"), filename)))
sftp = self.ssh_client.open_sftp()
try:
sftp.remove(remote_path)
Expand All @@ -138,10 +138,10 @@ def _scp(self, local_path: str, remote_path: str, mode=0o544) -> None:
sftp.put(local_path, remote_path)
sftp.chmod(remote_path, mode)
sftp.close()
print(p.info(f'%s: %s【%s】' % (_("Upload file"), filename, _('End'))))
print(p.info(f'【%s】 %s: %s' % (_('End'), _("Upload file"), filename)))

def __ensure_script_exist(self) -> None:
print(p.info(f'%s【%s】' % (_("Processing script file"), _('Start'))))
print(p.info(f'【%s】%s' % (_('Start'), _("Processing script file"))))
platform_named = {
'mac': ('jms_cli_darwin', '/tmp/behemoth', 'md5', -1),
'linux': ('jms_cli_linux', '/tmp/behemoth', 'md5sum', 0),
Expand All @@ -167,13 +167,13 @@ def __ensure_script_exist(self) -> None:
self.ssh_client.exec_command(f'mkdir -p {os.path.dirname(self._remote_script_path)}')
self._scp(str(local_path), str(self._remote_script_path))

print(p.info(f'%s【%s】' % (_("Processing script file"), _('End'))))
print(p.info(f'【%s】%s' % (_('End'), _("Processing script file"))))

def __process_commands_file(
self, remote_commands_file: str, local_commands_file: str,
token: str, **kwargs: dict
) -> None:
print(p.info(f'%s【%s】' % (_("Processing command files"), _('Start'))))
print(p.info(f'【%s】%s' % (_('Start'), _("Processing command files"))))
encrypted_data = kwargs.get('encrypted_data', False)
if encrypted_data:
local_commands_file = encrypt_json_file(local_commands_file, token[:32])
Expand All @@ -184,7 +184,7 @@ def __process_commands_file(
if cmd_file := kwargs.pop('cmd_file_real', ''):
cmd_filename = os.path.basename(cmd_file)
self._scp(cmd_file, os.path.join(remote_command_dir, cmd_filename), mode=0o400)
print(p.info(f'%s【%s】' % (_("Processing command files"), _('End'))))
print(p.info(f'【%s】%s' % (_('End'), _("Processing command files"))))

def __process_file(self, **kwargs: dict) -> None:
self.__ensure_script_exist()
Expand All @@ -200,7 +200,7 @@ def __clear(self, remote_commands_file: str, local_commands_file: str, **kwargs:
os.remove(local_commands_file)

def __execute_cmd(self, **kwargs: dict) -> None:
print(p.info(f'%s【%s】' % (_('Execute commands'), _('Start'))))
print(p.info(f'【%s】%s' % (_('Start'), _('Execute commands'))))
revert_key = {'remote_commands_file': 'cmd_set_filepath'}
exclude_params = ['local_commands_file', 'cmd_file_real']
params = {revert_key.get(k, k): v for k, v in kwargs.items() if k not in exclude_params}
Expand All @@ -215,7 +215,7 @@ def __execute_cmd(self, **kwargs: dict) -> None:
raise JMSException(error)
except SSHException as e:
raise JMSException(str(e))
print(p.info(f'%s【%s】' % (_('Execute commands'), _('End'))))
print(p.info(f'【%s】%s' % (_('End'), _('Execute commands'))))

def __execute(self, **kwargs: dict) -> None:
self.__process_file(**kwargs)
Expand Down

0 comments on commit a2114c3

Please sign in to comment.