extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-26 00:49:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix #1333 - listen_port isn't always present in the config file

Browse Source
This commit is contained in:
cmdr2 2023-06-07 15:45:21 +05:30
parent 9a0031c47b
commit e23f66a697
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ui/easydiffusion/server.py
View File

@ -338,21 +338,22 @@ def get_image_internal(task_id: int, img_id: int):
class CloudflareTunnel:
def __init__(self):
config = app.getConfig()
self.Urls = None
self.port = config["net"]["listen_port"]
self.urls = None
self.port = config.get("net", {}).get("listen_port")
def start(self):
self.Urls = try_cloudflare(self.port)
if self.port:
self.urls = try_cloudflare(self.port)
def stop(self):
if self.Urls != None:
if self.urls:
try_cloudflare.terminate(self.port)
self.Urls = None
self.urls = None
@property
def address(self):
if self.Urls != None:
return self.Urls.tunnel
if self.urls:
return self.urls.tunnel
else:
return None