From 0602b61eb852ddda5fa1dbf664109dc12c2148f8 Mon Sep 17 00:00:00 2001
From: Kamil Malinowski <kmalinowski@soldevelo.com>
Date: Tue, 22 Oct 2024 17:32:47 +0200
Subject: [PATCH] OM-349 Fixed query errors

---
 worker_voucher/schema.py | 45 ++++++++++++++++------------------------
 1 file changed, 18 insertions(+), 27 deletions(-)

diff --git a/worker_voucher/schema.py b/worker_voucher/schema.py
index eae8bf8..7db8d56 100644
--- a/worker_voucher/schema.py
+++ b/worker_voucher/schema.py
@@ -135,7 +135,7 @@ def resolve_previous_workers(self, info, economic_unit_code=None, date_range=Non
 
         eu = PolicyHolder.objects.filter(economic_unit_user_filter(info.context.user), code=economic_unit_code).first()
         if not eu:
-            return [{"message": _("workers.validation.economic_unit_not_exist")}]
+            raise AttributeError(_("workers.validation.economic_unit_not_exist"))
 
         query = Insuree.get_queryset(None, info.context.user).distinct('id').filter(
             worker_user_filter(info.context.user, economic_unit_code=economic_unit_code),
@@ -208,35 +208,26 @@ def resolve_assign_vouchers_validation(self, info, economic_unit_code=None, work
         return AcquireVouchersValidationSummaryGQLType(**validation_summary)
 
     def resolve_online_worker_data(self, info, national_id=None, economic_unit_code=None, **kwargs):
-        try:
-            logger.info("Online worker data: step 1" )
-            Query._check_permissions(info.context.user, InsureeConfig.gql_query_insuree_perms)
-
-            logger.info("Online worker data: step 2")
-            eu = PolicyHolder.objects.filter(economic_unit_user_filter(info.context.user), code=economic_unit_code).first()
-            if not eu:
-                return [{"message": _("workers.validation.economic_unit_not_exist")}]
-
-            logger.info("Online worker data: step 3")
-            if InsureeConfig.get_insuree_number_validator():
-                errors = custom_insuree_number_validation(national_id)
-                if errors:
-                    return errors
-
-            logger.info("Online worker data: step 4")
-            online_result = MConnectWorkerService().fetch_worker_data(national_id, info.context.user, eu)
-            if not online_result.get("success", False):
-                logger.error("Online worker data failed% %s", online_result)
-                raise AttributeError(online_result.get("error", _("Unknown Error")))
-
-            logger.info("Online worker data: %s", online_result)
-            return OnlineWorkerDataGQLType(
+        Query._check_permissions(info.context.user, InsureeConfig.gql_query_insuree_perms)
+
+        eu = PolicyHolder.objects.filter(economic_unit_user_filter(info.context.user), code=economic_unit_code).first()
+        if not eu:
+            raise AttributeError(_("workers.validation.economic_unit_not_exist"))
+
+        if InsureeConfig.get_insuree_number_validator():
+            errors = custom_insuree_number_validation(national_id)
+            if errors:
+                raise AttributeError(_("Insuree number not valid"))
+
+        online_result = MConnectWorkerService().fetch_worker_data(national_id, info.context.user, eu)
+        if not online_result.get("success", False):
+            raise AttributeError(online_result.get("error", _("Unknown Error")))
+
+        return OnlineWorkerDataGQLType(
             other_names=online_result["data"]["GivenName"],
             last_name=online_result["data"]["FamilyName"],
             photo=online_result["data"]["Photo"]
-            )
-        except Exception as e:
-            logger.error("Error", exc_info=e)
+        )
 
     def resolve_group_of_worker(self, info, economic_unit_code=None, **kwargs):
         if not info.context.user.has_perms(WorkerVoucherConfig.gql_group_of_worker_search_perms):