holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Write the config script files only if necessary

Browse Source
This commit is contained in:
cmdr2 2022-11-16 11:40:51 +05:30
parent ded9cb0358
commit 1f66daf2f3
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
ui/server.py
View File

@ -77,36 +77,37 @@ def setConfig(config):
print(traceback.format_exc()) print(traceback.format_exc())
try: # config.bat try: # config.bat
config_bat = [
f"@set update_branch={config['update_branch']}" if 'update_branch' in config else ""
]
config_bat_path = os.path.join(CONFIG_DIR, 'config.bat') config_bat_path = os.path.join(CONFIG_DIR, 'config.bat')
config_bat = []
if 'update_branch' in config:
config_bat.append(f"@set update_branch={config['update_branch']}")
if os.getenv('SD_UI_BIND_PORT') is not None: if os.getenv('SD_UI_BIND_PORT') is not None:
config_bat.append(f"@set SD_UI_BIND_PORT={os.getenv('SD_UI_BIND_PORT')}") config_bat.append(f"@set SD_UI_BIND_PORT={os.getenv('SD_UI_BIND_PORT')}")
if os.getenv('SD_UI_BIND_IP') is not None: if os.getenv('SD_UI_BIND_IP') is not None:
config_bat.append(f"@set SD_UI_BIND_IP={os.getenv('SD_UI_BIND_IP')}") config_bat.append(f"@set SD_UI_BIND_IP={os.getenv('SD_UI_BIND_IP')}")
with open(config_bat_path, 'w', encoding='utf-8') as f: if len(config_bat) > 0:
f.write('\r\n'.join(config_bat)) with open(config_bat_path, 'w', encoding='utf-8') as f:
except Exception as e: f.write('\r\n'.join(config_bat))
except:
print(traceback.format_exc()) print(traceback.format_exc())
try: # config.sh try: # config.sh
config_sh = [
'#!/bin/bash',
f"export update_branch={config['update_branch']}" if 'update_branch' in config else ""
]
config_sh_path = os.path.join(CONFIG_DIR, 'config.sh') config_sh_path = os.path.join(CONFIG_DIR, 'config.sh')
config_sh = ['#!/bin/bash']
if 'update_branch' in config:
config_sh.append(f"export update_branch={config['update_branch']}")
if os.getenv('SD_UI_BIND_PORT') is not None: if os.getenv('SD_UI_BIND_PORT') is not None:
config_sh.append(f"export SD_UI_BIND_PORT={os.getenv('SD_UI_BIND_PORT')}") config_sh.append(f"export SD_UI_BIND_PORT={os.getenv('SD_UI_BIND_PORT')}")
if os.getenv('SD_UI_BIND_IP') is not None: if os.getenv('SD_UI_BIND_IP') is not None:
config_sh.append(f"export SD_UI_BIND_IP={os.getenv('SD_UI_BIND_IP')}") config_sh.append(f"export SD_UI_BIND_IP={os.getenv('SD_UI_BIND_IP')}")
with open(config_sh_path, 'w', encoding='utf-8') as f: if len(config_sh) > 1:
f.write('\n'.join(config_sh)) with open(config_sh_path, 'w', encoding='utf-8') as f:
except Exception as e: f.write('\n'.join(config_sh))
except:
print(traceback.format_exc()) print(traceback.format_exc())
def resolve_model_to_use(model_name:str, model_type:str, model_dir:str, model_extensions:list, default_models=[]): def resolve_model_to_use(model_name:str, model_type:str, model_dir:str, model_extensions:list, default_models=[]):