Elgato_dark/bunkerweb
1
0
Fork 0
mirror of https://github.com/bunkerity/bunkerweb synced 2026-05-24 09:28:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor main.py to handle Response objects in run_action and custom_plugin functions

Browse source
This commit is contained in:
Théophile Diot 2024-04-06 09:51:43 +01:00
parent b7555141c0
commit 7cf0f55f99
No known key found for this signature in database
GPG key ID: 248FEA4BAE400D06
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/ui/main.py
View file

@ -340,6 +340,9 @@ def run_action(plugin: str, function_name: str = ""):
if message or not isinstance(res, dict) and not res:
return {"status": "ko", "code": 500, "message": message or "The plugin did not return a valid response"}
if isinstance(res, Response):
return res
return {"status": "ok", "code": 200, "data": res}
@ -1625,11 +1628,19 @@ def custom_plugin(plugin: str):
)
action_result = run_action(plugin)
if isinstance(action_result, Response):
app.logger.info(f"Plugin {plugin} action executed successfully")
return action_result
# case error
if action_result["status"] == "ko":
return error_message(action_result["message"]), action_result["code"]
app.logger.info(f"Plugin {plugin} action executed successfully")
if request.content_type == "application/x-www-form-urlencoded":
return redirect(f"{url_for('plugins')}/{plugin}", code=303)
return jsonify({"message": "ok", "data": action_result["data"]}), 200