sij/sijapi
1
1
Fork 0
Code Issues 3 Pull requests Projects Releases Packages Wiki Activity Actions

auto-update

Browse source
This commit is contained in:
sanj 2024-11-09 18:12:05 -08:00
parent 56a203ea74
commit 0d9924e8f7
1 changed files with 62 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

88
sijapi/helpers/start.py
View file

@ -74,18 +74,44 @@ def start_local_server(server, pull=False, push=False):
logging.error(f"Failed to start sijapi session on local machine. Error: {e}") logging.error(f"Failed to start sijapi session on local machine. Error: {e}")
logging.error(f"Error output: {e.stderr}") logging.error(f"Error output: {e.stderr}")
def kill_local_server():
try:
if is_local_tmux_session_running('sijapi'):
subprocess.run(['tmux', 'kill-session', '-t', 'sijapi'], check=True)
logging.info("Killed local sijapi tmux session.")
else:
logging.info("No local sijapi tmux session to kill.")
except subprocess.CalledProcessError as e:
logging.error(f"Failed to kill local sijapi tmux session. Error: {e}")
def start_remote_server(server, pull=False, push=False): def start_remote_server(server, pull=False, push=False):
ssh = paramiko.SSHClient() ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try: try:
ssh.connect( # Determine authentication method based on config
server['ts_ip'], if 'ssh_key' in server:
port=server['ssh_port'], # Use SSH key authentication
username=server['ssh_user'], private_key = paramiko.RSAKey.from_private_key_file(server['ssh_key'])
password=server['ssh_pass'], ssh.connect(
timeout=10 server['ts_ip'],
) port=server['ssh_port'],
username=server['ssh_user'],
pkey=private_key,
timeout=10
)
elif 'ssh_pass' in server:
# Use password authentication
ssh.connect(
server['ts_ip'],
port=server['ssh_port'],
username=server['ssh_user'],
password=server['ssh_pass'],
timeout=10
)
else:
logging.error(f"No authentication method specified for {server['ts_id']}")
return
status, output, error = execute_ssh_command(ssh, f"{server['tmux']} has-session -t sijapi 2>/dev/null && echo 'exists' || echo 'not exists'") status, output, error = execute_ssh_command(ssh, f"{server['tmux']} has-session -t sijapi 2>/dev/null && echo 'exists' || echo 'not exists'")
if output == 'exists': if output == 'exists':
@ -108,31 +134,39 @@ def start_remote_server(server, pull=False, push=False):
except paramiko.SSHException as e: except paramiko.SSHException as e:
logging.error(f"Failed to connect to {server['ts_id']}: {str(e)}") logging.error(f"Failed to connect to {server['ts_id']}: {str(e)}")
except Exception as e:
logging.error(f"Error connecting to {server['ts_id']}: {str(e)}")
finally: finally:
ssh.close() ssh.close()
def kill_local_server():
try:
if is_local_tmux_session_running('sijapi'):
subprocess.run(['tmux', 'kill-session', '-t', 'sijapi'], check=True)
logging.info("Killed local sijapi tmux session.")
else:
logging.info("No local sijapi tmux session to kill.")
except subprocess.CalledProcessError as e:
logging.error(f"Failed to kill local sijapi tmux session. Error: {e}")
def kill_remote_server(server): def kill_remote_server(server):
ssh = paramiko.SSHClient() ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try: try:
ssh.connect( # Determine authentication method based on config
server['ts_ip'], if 'ssh_key' in server:
port=server['ssh_port'], # Use SSH key authentication
username=server['ssh_user'], private_key = paramiko.RSAKey.from_private_key_file(server['ssh_key'])
password=server['ssh_pass'], ssh.connect(
timeout=10 server['ts_ip'],
) port=server['ssh_port'],
username=server['ssh_user'],
pkey=private_key,
timeout=10
)
elif 'ssh_pass' in server:
# Use password authentication
ssh.connect(
server['ts_ip'],
port=server['ssh_port'],
username=server['ssh_user'],
password=server['ssh_pass'],
timeout=10
)
else:
logging.error(f"No authentication method specified for {server['ts_id']}")
return
command = f"{server['tmux']} kill-session -t sijapi" command = f"{server['tmux']} kill-session -t sijapi"
status, output, error = execute_ssh_command(ssh, command) status, output, error = execute_ssh_command(ssh, command)
@ -144,9 +178,12 @@ def kill_remote_server(server):
except paramiko.SSHException as e: except paramiko.SSHException as e:
logging.error(f"Failed to connect to {server['ts_id']}: {str(e)}") logging.error(f"Failed to connect to {server['ts_id']}: {str(e)}")
except Exception as e:
logging.error(f"Error connecting to {server['ts_id']}: {str(e)}")
finally: finally:
ssh.close() ssh.close()
def main(): def main():
load_env() load_env()
db_config = load_config('sys') db_config = load_config('sys')
@ -207,5 +244,4 @@ def main():
time.sleep(1) time.sleep(1)
if __name__ == "__main__": if __name__ == "__main__":
main() main()