From b21a4c00298a4e52e79a9a55b9fd44bf47de3ffb Mon Sep 17 00:00:00 2001
From: "oluyemi.a.simbrellang.com" <oluyemi.a@simbrellang.com>
Date: Tue, 15 Apr 2025 14:44:37 +0100
Subject: [PATCH] improve responses

---
 app/__init__.py                      |  4 +++-
 app/errors/__init__.py               |  1 -
 app/response/__init__.py             |  2 ++
 app/{errors => response}/handlers.py | 12 ++++++++++++
 4 files changed, 17 insertions(+), 2 deletions(-)
 delete mode 100644 app/errors/__init__.py
 create mode 100644 app/response/__init__.py
 rename app/{errors => response}/handlers.py (70%)

diff --git a/app/__init__.py b/app/__init__.py
index 52cca2e..a42cecb 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -2,7 +2,7 @@ from flask import Flask
 from flask_cors import CORS
 from app.config import Config
 from app.routes import auth_bp, autocall_bp
-from app.errors import method_not_allowed, unsupported_media_type
+from app.response import (method_not_allowed, unsupported_media_type, not_found, bad_request)
 from app.extensions import db
 
 
@@ -23,6 +23,8 @@ def create_app():
     # Error Handlers
     app.register_error_handler(405, method_not_allowed)
     app.register_error_handler(415, unsupported_media_type)
+    app.register_error_handler(404, not_found)
+    app.register_error_handler(400, bad_request)
 
     # Database
     db.init_app(app)
diff --git a/app/errors/__init__.py b/app/errors/__init__.py
deleted file mode 100644
index 15c380c..0000000
--- a/app/errors/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-from .handlers import method_not_allowed, unsupported_media_type
\ No newline at end of file
diff --git a/app/response/__init__.py b/app/response/__init__.py
new file mode 100644
index 0000000..7c0277e
--- /dev/null
+++ b/app/response/__init__.py
@@ -0,0 +1,2 @@
+from .handlers import (method_not_allowed, unsupported_media_type,
+not_found, bad_request, success, created, updated)
\ No newline at end of file
diff --git a/app/errors/handlers.py b/app/response/handlers.py
similarity index 70%
rename from app/errors/handlers.py
rename to app/response/handlers.py
index 30257b3..2ce566a 100644
--- a/app/errors/handlers.py
+++ b/app/response/handlers.py
@@ -16,3 +16,15 @@ def bad_request(error):
 
 def unsupported_media_type(error):
     return ResponseHelper.error(message="Unsupported Media Type", status_code=415)
+
+
+def success(data):
+    return ResponseHelper.success(data=data)
+
+
+def created(data):
+    return ResponseHelper.created(data=data)
+
+
+def updated(data):
+    return ResponseHelper.updated(data=data)