michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add username and password in ngrok.

Browse source
This commit is contained in:
JingShing 2022-11-10 20:39:20 +08:00 committed by GitHub
parent ac08562854
commit 2505f39e28
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/ngrok.py
View file

@ -1,13 +1,21 @@
from pyngrok import ngrok, conf, exception from pyngrok import ngrok, conf, exception
def connect(token, port, region): def connect(token, port, region):
if token == None: if token == None:
token = 'None' token = 'None'
else:
if ':' in token:
# token = authtoken:username:password
account = token.split(':')[1] + ':' + token.split(':')[-1]
token = token.split(':')[0]
config = conf.PyngrokConfig( config = conf.PyngrokConfig(
auth_token=token, region=region auth_token=token, region=region
) )
try: try:
if account:
public_url = ngrok.connect(port, pyngrok_config=config, auth=account).public_url
else:
public_url = ngrok.connect(port, pyngrok_config=config).public_url public_url = ngrok.connect(port, pyngrok_config=config).public_url
except exception.PyngrokNgrokError: except exception.PyngrokNgrokError:
print(f'Invalid ngrok authtoken, ngrok connection aborted.\n' print(f'Invalid ngrok authtoken, ngrok connection aborted.\n'