extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-12 08:30:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Prevent block_nsfw from getting edited via the HTTP api

Browse Source
This commit is contained in:
cmdr2 2023-08-30 16:22:05 +05:30
parent 9a9f6e3559
commit 80e47be5a5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ui/easydiffusion/server.py
View File

@ -36,6 +36,7 @@ NOCACHE_HEADERS = {
"Pragma": "no-cache", "Pragma": "no-cache",
"Expires": "0", "Expires": "0",
} }
PROTECTED_CONFIG_KEYS = ("block_nsfw",) # can't change these via the HTTP API
class NoCacheStaticFiles(StaticFiles): class NoCacheStaticFiles(StaticFiles):
@ -175,7 +176,7 @@ def set_app_config_internal(req: SetAppConfigRequest):
config["test_diffusers"] = req.test_diffusers config["test_diffusers"] = req.test_diffusers
for property, property_value in req.dict().items(): for property, property_value in req.dict().items():
if property_value is not None and property not in req.__fields__: if property_value is not None and property not in req.__fields__ and property not in PROTECTED_CONFIG_KEYS:
config[property] = property_value config[property] = property_value
try: try:
@ -456,6 +457,7 @@ def modify_package_internal(package_name: str, req: dict):
log.error(traceback.format_exc()) log.error(traceback.format_exc())
return HTTPException(status_code=500, detail=str(e)) return HTTPException(status_code=500, detail=str(e))
def get_sha256_internal(obj_path): def get_sha256_internal(obj_path):
import hashlib import hashlib
from easydiffusion.utils import sha256sum from easydiffusion.utils import sha256sum
@ -477,4 +479,3 @@ def get_sha256_internal(obj_path):
log.error(str(e)) log.error(str(e))
log.error(traceback.format_exc()) log.error(traceback.format_exc())
return HTTPException(status_code=500, detail=str(e)) return HTTPException(status_code=500, detail=str(e))