stable-diffusion-webui/modules/ngrok.py

27 lines
1020 B
Python
Raw Normal View History

2022-10-11 09:40:27 +00:00
from pyngrok import ngrok, conf, exception
2022-10-16 00:24:01 +00:00
def connect(token, port, region):
2022-11-10 12:42:41 +00:00
account = None
2022-10-11 09:40:27 +00:00
if token == None:
token = 'None'
2022-11-10 12:39:20 +00:00
else:
if ':' in token:
# token = authtoken:username:password
account = token.split(':')[1] + ':' + token.split(':')[-1]
token = token.split(':')[0]
2022-10-16 00:24:01 +00:00
config = conf.PyngrokConfig(
auth_token=token, region=region
)
2022-10-11 09:40:27 +00:00
try:
2022-11-10 12:42:41 +00:00
if account == None:
2022-11-10 12:39:20 +00:00
public_url = ngrok.connect(port, pyngrok_config=config).public_url
2022-11-10 12:42:41 +00:00
else:
public_url = ngrok.connect(port, pyngrok_config=config, auth=account).public_url
2022-10-11 09:40:27 +00:00
except exception.PyngrokNgrokError:
print(f'Invalid ngrok authtoken, ngrok connection aborted.\n'
f'Your token: {token}, get the right one on https://dashboard.ngrok.com/get-started/your-authtoken')
else:
print(f'ngrok connected to localhost:{port}! URL: {public_url}\n'
2022-10-11 09:48:27 +00:00
'You can use this link after the launch is complete.')