diff --git a/__init__.py b/__init__.py index cfb0ddb..bb5324d 100644 --- a/__init__.py +++ b/__init__.py @@ -72,8 +72,10 @@ if os.path.isfile(config_path): with open(config_path, 'w') as f: yaml.dump(data, f) directory = f"./web_version/v1" - else: + elif "WEB_VERSION" in data: directory = f"./web_version/{data['WEB_VERSION']}" + else: + directory = f"./web_version/v1" if not os.path.exists(os.path.join(cwd_path, directory)): print(f"Web version {data['WEB_VERSION']} not found, using default") directory = f"./web_version/v1"