intrgrating suggestions after betta 1

transportmanager/server/routes/subscription.py

@@ -11,6 +11,10 @@ subscription_bp = Blueprint("subscription", __name__, url_prefix="/subscription"
 @jwt_required()
 def get_subscription():
     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']
     subscription_model = Subscription()
     subscriptions = subscription_model.get_by_user_id(user_id)
     return jsonify(subscriptions), 200