-
Notifications
You must be signed in to change notification settings - Fork 8
/
setup_kaggle_ssh.py
63 lines (50 loc) · 1.84 KB
/
setup_kaggle_ssh.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import sys
import random
import string
import subprocess
import time
import json
import os
if len(sys.argv) != 2:
print("Usage: python setup_kaggle_ssh.py <ngrok_authtoken>")
exit(1)
ngrok_auth_token = sys.argv[1]
def generate_random_password(length=16):
characters = (string.ascii_letters + string.digits + "!@#$%^*()-_=+{}[]<>.,?")
return ''.join(random.choices(characters, k=length))
password = generate_random_password()
print(f"Setting password for root user: {password}")
subprocess.run(f"echo 'root:{password}' | sudo chpasswd", shell=True, check=True)
subprocess.run("service ssh start", check=True, shell=True)
subprocess.run("service ssh restart", check=True, shell=True)
subprocess.run(f"ngrok authtoken {ngrok_auth_token}", check=True, shell=True)
ngrok_process = subprocess.Popen(["ngrok", "tcp", "22", "--region", "ap"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
time.sleep(10)
def get_ngrok_tunnel():
try:
output = subprocess.check_output(["curl", "-s", "http://localhost:4040/api/tunnels"]).decode()
tunnels = json.loads(output)
for tunnel in tunnels['tunnels']:
if tunnel['proto'] == 'tcp':
return tunnel['public_url']
except Exception as e:
print(f"Failed to fetch tunnels: {str(e)}")
return None
ngrok_url = get_ngrok_tunnel()
if ngrok_url:
hostname = ngrok_url.split('//')[1].split(':')[0]
port = ngrok_url.split(':')[2]
print(f"ngrok tunnel opened at: {ngrok_url}")
print(f"To connect via SSH, use the following command:")
print(f"ssh root@{hostname} -p {port}")
print(f"Password: {password}")
sys.stdout.flush()
else:
print("Failed to start ngrok tunnel.")
exit(1)
try:
ngrok_process.wait()
except KeyboardInterrupt:
print("Shutting down ngrok tunnel")
ngrok_process.terminate()
ngrok_process.wait()