fix: catching database exceptions

Author: Cinnamoon-dev

src/services/userService.py

@@ -40,12 +40,15 @@ def all(self, query_params: QueryParams) -> JSONResponse:
     def view(self, user_id: int) -> JSONResponse:
         user = None
 
-        with PgDatabase() as db:
-            db.cursor.execute(f"SELECT {self.all_columns} FROM {self.table} WHERE id = %s", (user_id,))
-            row = db.cursor.fetchone()
+        try:
+            with PgDatabase() as db:
+                db.cursor.execute(f"SELECT {self.all_columns} FROM {self.table} WHERE id = %s", (user_id,))
+                row = db.cursor.fetchone()
 
-            if row is None:
-                return JSONResponse(status_code=404, content={"error": True, "message": "Usuário não encontrado"})
+                if row is None:
+                    return JSONResponse(status_code=404, content={"error": True, "message": "Usuário não encontrado"})
+        except Exception:
+            return JSONResponse(status_code=500, content={"error": True, "message": "Database error"})
 
         user = line_to_dict(row, self.columns)
         return JSONResponse(status_code=200, content={"error": False, "data": user})

src/services/userTypeService.py

@@ -35,12 +35,15 @@ def all(self, query_params: QueryParams) -> JSONResponse:
     def view(self, user_type_id: int) -> JSONResponse:
         user_type = None
 
-        with PgDatabase() as db:
-            db.cursor.execute(f"SELECT id, nome FROM {self.table} WHERE id = %s", (user_type_id,))
-            row = db.cursor.fetchone()
+        try:
+            with PgDatabase() as db:
+                db.cursor.execute(f"SELECT id, nome FROM {self.table} WHERE id = %s", (user_type_id,))
+                row = db.cursor.fetchone()
 
-            if row is None:
-                return JSONResponse(status_code=404, content={"error": True, "message": "Tipo de usuário não encontrado"})
+                if row is None:
+                    return JSONResponse(status_code=404, content={"error": True, "message": "Tipo de usuário não encontrado"})
+        except Exception:
+            return JSONResponse(status_code=500, content={"error": True, "message": "Database error"})
 
         user_type = line_to_dict(row, self.columns)
         return JSONResponse(status_code=200, content={"error": False, "data": user_type})