feat: allow enabling after url is provided

This commit is contained in:
Timothy J. Baek 2024-02-21 18:57:53 -08:00
parent d863e7e60b
commit 797ed9ac7f
2 changed files with 13 additions and 4 deletions

View file

@ -43,8 +43,12 @@ async def get_enable_status(request: Request, user=Depends(get_admin_user)):
@app.get("/enabled/toggle", response_model=bool)
async def toggle_enabled(request: Request, user=Depends(get_admin_user)):
app.state.ENABLED = not app.state.ENABLED
return app.state.ENABLED
try:
r = requests.head(app.state.AUTOMATIC1111_BASE_URL)
app.state.ENABLED = not app.state.ENABLED
return app.state.ENABLED
except Exception as e:
raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
class UrlUpdateForm(BaseModel):