intrgrating suggestions after betta 1

transportmanager/server/routes/transporters.py

@@ -1,6 +1,7 @@
 from flask import Blueprint, request, jsonify
 from flask_jwt_extended import jwt_required, get_jwt_identity
 from models.transporters import Transporters
+from models.user import Users
 
 transporters_bp = Blueprint("transporters", __name__, url_prefix="/transporters")
 
@@ -8,6 +9,10 @@ transporters_bp = Blueprint("transporters", __name__, url_prefix="/transporters"
 @jwt_required()
 def list_transporters():
     user_id = get_jwt_identity()
+    users = Users()
+    user = users.get_user_by_id(user_id)
+    if user['user_role'] == 'company_user':
+        user_id = user['company_id']
     transporters_db = Transporters()
     transporters = transporters_db.get_all_transporters_by_user(user_id)
     return jsonify(transporters), 200
@@ -18,6 +23,10 @@ def create_transporter():
     transporters_db = Transporters()
     data = request.get_json()
     user_id = get_jwt_identity()
+    users = Users()
+    user = users.get_user_by_id(user_id)
+    if user['user_role'] == 'company_user':
+        user_id = user['company_id']
     transporter_id = transporters_db.create_transporter(
         user_id=user_id,
         name=data.get("name"),
@@ -36,6 +45,10 @@ def create_transporter():
 def update_transporter(transporter_id):
     transporters_db = Transporters()
     user_id = get_jwt_identity()
+    users = Users()
+    user = users.get_user_by_id(user_id)
+    if user['user_role'] == 'company_user':
+        user_id = user['company_id']
     data = request.get_json()
     transporter = transporters_db.get_transporter_by_id(transporter_id)
     if not transporter:
@@ -58,6 +71,10 @@ def update_transporter(transporter_id):
 def delete_transporter(transporter_id):
     transporters_db = Transporters()
     user_id = get_jwt_identity()
+    users = Users()
+    user = users.get_user_by_id(user_id)
+    if user['user_role'] == 'company_user':
+        user_id = user['company_id']
     transporter = transporters_db.get_transporter_by_id(transporter_id)
     if not transporter:
         return jsonify({"error": "Transporter not found"}), 404