UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-28 06:53:14 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update config.py

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-03 21:46:40 +02:00
parent 6bbb755997
commit 9bcbf5e9b3
1 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
backend/config.py
View File

@ -47,7 +47,7 @@ def run_migrations():
from alembic import command from alembic import command
from alembic.config import Config from alembic.config import Config
alembic_cfg = Config(BACKEND_DIR / "alembic.ini") alembic_cfg = Config("alembic.ini")
command.upgrade(alembic_cfg, "head") command.upgrade(alembic_cfg, "head")
except Exception as e: except Exception as e:
print(f"Error: {e}") print(f"Error: {e}")
@ -90,15 +90,6 @@ if os.path.exists(f"{DATA_DIR}/config.json"):
save_to_db(data) save_to_db(data)
os.rename(f"{DATA_DIR}/config.json", f"{DATA_DIR}/old_config.json") os.rename(f"{DATA_DIR}/config.json", f"{DATA_DIR}/old_config.json")
def save_config():
try:
with open(f"{DATA_DIR}/config.json", "w") as f:
json.dump(CONFIG_DATA, f, indent="\t")
except Exception as e:
log.exception(e)
DEFAULT_CONFIG = { DEFAULT_CONFIG = {
"version": 0, "version": 0,
"ui": { "ui": {
@ -172,6 +163,25 @@ def get_config_value(config_path: str):
return cur_config return cur_config
PERSISTENT_CONFIG_REGISTRY = []
def save_config(config):
global CONFIG_DATA
global PERSISTENT_CONFIG_REGISTRY
try:
save_to_db(config)
CONFIG_DATA = config
# Trigger updates on all registered PersistentConfig entries
for config_item in PERSISTENT_CONFIG_REGISTRY:
config_item.update()
except Exception as e:
log.exception(e)
return False
return True
T = TypeVar("T") T = TypeVar("T")
@ -187,6 +197,8 @@ class PersistentConfig(Generic[T]):
else: else:
self.value = env_value self.value = env_value
PERSISTENT_CONFIG_REGISTRY.append(self)
def __str__(self): def __str__(self):
return str(self.value) return str(self.value)
@ -203,6 +215,12 @@ class PersistentConfig(Generic[T]):
) )
return super().__getattribute__(item) return super().__getattribute__(item)
def update(self):
new_value = get_config_value(self.config_path)
if new_value is not None:
self.value = new_value
log.info(f"Updated {self.env_name} to new value {self.value}")
def save(self): def save(self):
log.info(f"Saving '{self.env_name}' to the database") log.info(f"Saving '{self.env_name}' to the database")
path_parts = self.config_path.split(".") path_parts = self.config_path.split(".")