19 lines
557 B
Python
19 lines
557 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
from gpu_rent.config import load_config
|
|
from gpu_rent.state import load_state
|
|
from gpu_rent.ssh_ops import _connect, put_file_on, run_ssh_on
|
|
|
|
cfg = load_config(require_auth=True)
|
|
state = load_state()
|
|
client = _connect(cfg, state.floating_ip)
|
|
try:
|
|
put_file_on(client, Path("tmp_smoke.sh"), "/tmp/tmp_smoke.sh")
|
|
out = run_ssh_on(
|
|
client, "sed -i 's/\\r$//' /tmp/tmp_smoke.sh && bash /tmp/tmp_smoke.sh", check=False, timeout=120
|
|
)
|
|
sys.stdout.write(out)
|
|
finally:
|
|
client.close()
|