Remove govn add products (#3615)

Author: Jxio

auth-api/src/auth_api/exceptions/errors.py

@@ -126,7 +126,6 @@ class Error(Enum):
     )
     BCEID_USERS_CANT_BE_OWNERS = "BCEID Users cant be owners", HTTPStatus.BAD_REQUEST
     PAYMENT_ACCOUNT_UPSERT_FAILED = "Account upsert failed in Pay", HTTPStatus.INTERNAL_SERVER_ERROR
-    ACCOUNT_FEES_FETCH_FAILED = "Failed to fetch account fees from Pay API", HTTPStatus.INTERNAL_SERVER_ERROR
     GOVM_ACCOUNT_DATA_MISSING = (
         "GOVM account creation needs payment info , gl code and mailing address",
         HTTPStatus.BAD_REQUEST,

auth-api/src/auth_api/models/dataclass.py

@@ -183,7 +183,6 @@ class TaskSearch:  # pylint: disable=too-many-instance-attributes
     submitted_sort_order: str = "asc"
     page: int = 1
     limit: int = 10
-    action: str = ""
 
 
 @dataclass
@@ -202,7 +201,6 @@ class ProductReviewTask:
 
     org_id: str
     org_name: str
-    org_access_type: str
     product_code: str
     product_description: str
     product_subscription_id: int

auth-api/src/auth_api/models/task.py

@@ -66,8 +66,6 @@ def fetch_tasks(cls, task_search: TaskSearch):
             query = query.filter(Task.type == task_search.type)
         if task_search.status:
             query = query.filter(Task.status.in_(task_search.status))
-        if task_search.action:
-            query = query.filter(Task.action.in_(task_search.action))
         start_date = None
         end_date = None
         if task_search.start_date:

auth-api/src/auth_api/resources/v1/task.py

@@ -46,7 +46,6 @@ def get_tasks():
             relationship_status=request.args.get("relationshipStatus", None),
             type=request.args.get("type", None),
             status=request.args.getlist("status", None),
-            action=request.args.getlist("action", None),
             modified_by=request.args.get("modifiedBy", None),
             submitted_sort_order=request.args.get("submittedSortOrder", None),
             page=int(request.args.get("page", 1)),

auth-api/src/auth_api/services/org.py

@@ -141,16 +141,9 @@ def create_org(org_info: dict, user_id):
         # create the membership record for this user if its not created by staff and access_type is anonymous
         Org.create_membership(access_type, org, user_id)
 
-        # Send an email to staff to remind review the pending account
-        is_staff_review_needed = access_type == AccessType.GOVN.value or (
-            access_type in (AccessType.EXTRA_PROVINCIAL.value, AccessType.REGULAR_BCEID.value)
-            and not AffidavitModel.find_approved_by_user_id(user_id=user_id)
-            and current_app.config.get("SKIP_STAFF_APPROVAL_BCEID") is False
-        )
-
         if product_subscriptions is not None:
             ProductService.create_product_subscription(
-                org.id, subscription_data={"subscriptions": product_subscriptions}, skip_auth=True, staff_review_for_create_org=is_staff_review_needed
+                org.id, subscription_data={"subscriptions": product_subscriptions}, skip_auth=True
             )
 
         ProductService.create_subscription_from_bcol_profile(org.id, bcol_profile_flags)
@@ -160,6 +153,13 @@ def create_org(org_info: dict, user_id):
         if payment_account_status == PaymentAccountStatus.FAILED and error is not None:
             current_app.logger.warning(f"Account update payment Error: {error}")
 
+        # Send an email to staff to remind review the pending account
+        is_staff_review_needed = access_type == AccessType.GOVN.value or (
+            access_type in (AccessType.EXTRA_PROVINCIAL.value, AccessType.REGULAR_BCEID.value)
+            and not AffidavitModel.find_approved_by_user_id(user_id=user_id)
+            and current_app.config.get("SKIP_STAFF_APPROVAL_BCEID") is False
+        )
+
         if is_staff_review_needed:
             Org._create_staff_review_task(org, UserModel.find_by_jwt_token())
 

auth-api/src/auth_api/services/products.py

@@ -56,7 +56,7 @@
     get_product_notification_data,
     get_product_notification_type,
 )
-from auth_api.utils.roles import CLIENT_ADMIN_ROLES, CLIENT_AUTH_ROLES, GOV_ORG_TYPES, STAFF
+from auth_api.utils.roles import CLIENT_ADMIN_ROLES, CLIENT_AUTH_ROLES, STAFF
 from auth_api.utils.user_context import UserContext, user_context
 
 from .activity_log_publisher import ActivityLogPublisher
@@ -170,7 +170,7 @@ def _is_previously_approved(org_id: int, product_code: str):
         )
         if task is None or (
             task.relationship_status != TaskRelationshipStatus.ACTIVE.value
-            and task.action in (TaskAction.PRODUCT_REVIEW.value, TaskAction.NEW_PRODUCT_FEE_REVIEW.value)
+            and task.action == TaskAction.PRODUCT_REVIEW.value
         ):
             return False, None
 
@@ -183,7 +183,6 @@ def create_product_subscription(
         is_new_transaction: bool = True,
         skip_auth=False,
         auto_approve=False,
-        staff_review_for_create_org=False,
     ):
         """Create product subscription for the user.
 
@@ -199,7 +198,6 @@ def create_product_subscription(
 
         subscriptions_list = subscription_data.get("subscriptions")
         for subscription in subscriptions_list:
-            auto_approve_current = auto_approve
             product_code = subscription.get("productCode")
             if ProductSubscriptionModel.find_by_org_id_product_code(org_id, product_code):
                 raise BusinessException(Error.PRODUCT_SUBSCRIPTION_EXISTS, None)
@@ -210,11 +208,9 @@ def create_product_subscription(
                     check_auth(system_required=True, org_id=org_id)
                 previously_approved, inactive_sub = Product._is_previously_approved(org_id, product_code)
                 if previously_approved:
-                    auto_approve_current = True
+                    auto_approve = True
 
-                subscription_status = Product.find_subscription_status(
-                    org, product_model, auto_approve_current, staff_review_for_create_org
-                )
+                subscription_status = Product.find_subscription_status(org, product_model, auto_approve)
                 product_subscription = Product._subscribe_and_publish_activity(
                     SubscriptionRequest(
                         org_id=org_id,
@@ -249,7 +245,6 @@ def create_product_subscription(
                         ProductReviewTask(
                             org_id=org.id,
                             org_name=org.name,
-                            org_access_type=org.access_type,
                             product_code=product_subscription.product_code,
                             product_description=product_model.description,
                             product_subscription_id=product_subscription.id,
@@ -376,14 +371,11 @@ def _reset_subscription_and_review_task(
     @staticmethod
     def _create_review_task(review_task: ProductReviewTask):
         task_type = review_task.product_description
-
-        required_review_types = {AccessType.GOVM.value, AccessType.GOVN.value}
-        if review_task.product_code in QUALIFIED_SUPPLIER_PRODUCT_CODES:
-            action_type = TaskAction.QUALIFIED_SUPPLIER_REVIEW.value
-        elif review_task.org_access_type in required_review_types:
-            action_type = TaskAction.NEW_PRODUCT_FEE_REVIEW.value
-        else:
-            action_type = TaskAction.PRODUCT_REVIEW.value
+        action_type = (
+            TaskAction.QUALIFIED_SUPPLIER_REVIEW.value
+            if review_task.product_code in QUALIFIED_SUPPLIER_PRODUCT_CODES
+            else TaskAction.PRODUCT_REVIEW.value
+        )
 
         task_info = {
             "name": review_task.org_name,
@@ -401,13 +393,14 @@ def _create_review_task(review_task: ProductReviewTask):
         TaskService.create_task(task_info, False)
 
     @staticmethod
-    def find_subscription_status(org, product_model, auto_approve=False, staff_review_for_create_org=False):
+    def find_subscription_status(org, product_model, auto_approve=False):
         """Return the subscriptions status based on org type."""
-        skip_review = org.access_type in GOV_ORG_TYPES and staff_review_for_create_org # prevent create second task when it's already added a staff review when creating org
-        if (product_model.need_review or org.access_type in GOV_ORG_TYPES) and not auto_approve:
+        # GOVM accounts has default active subscriptions
+        skip_review_types = [AccessType.GOVM.value]
+        if product_model.need_review and auto_approve is False:
             return (
                 ProductSubscriptionStatus.ACTIVE.value
-                if skip_review
+                if (org.access_type in skip_review_types)
                 else ProductSubscriptionStatus.PENDING_STAFF_REVIEW.value
             )
         return ProductSubscriptionStatus.ACTIVE.value
@@ -459,10 +452,7 @@ def get_all_product_subscription(org_id, skip_auth=False, **kwargs):
             check_auth(one_of_roles=(*CLIENT_AUTH_ROLES, STAFF), org_id=org_id)
 
         product_subscriptions: list[ProductSubscriptionModel] = ProductSubscriptionModel.find_by_org_ids([org_id])
-        subscription_by_code = {
-            sub.product_code: sub
-            for sub in product_subscriptions
-        }
+        subscriptions_dict = {x.product_code: x.status_code for x in product_subscriptions}
 
         # Include hidden products only for staff and SBC staff
         include_hidden = (
@@ -474,9 +464,9 @@ def get_all_product_subscription(org_id, skip_auth=False, **kwargs):
 
         products = Product.get_products(include_hidden=include_hidden, staff_check=False)
         for product in products:
-            sub = subscription_by_code.get(product.get("code"))
-            product["subscriptionStatus"] = getattr(sub, "status_code", ProductSubscriptionStatus.NOT_SUBSCRIBED.value)
-            product["id"] = getattr(sub, "id", None)
+            product["subscriptionStatus"] = subscriptions_dict.get(
+                product.get("code"), ProductSubscriptionStatus.NOT_SUBSCRIBED.value
+            )
 
         return products
 
@@ -490,6 +480,7 @@ def update_product_subscription(product_sub_info: ProductSubscriptionInfo, is_ne
         is_hold = product_sub_info.is_hold
         org_id = product_sub_info.org_id
         org_name = product_sub_info.org_name
+
         # Approve/Reject Product subscription
         product_subscription: ProductSubscriptionModel = ProductSubscriptionModel.find_by_id(product_subscription_id)
 
@@ -510,6 +501,7 @@ def update_product_subscription(product_sub_info: ProductSubscriptionInfo, is_ne
         product_model: ProductCodeModel = ProductCodeModel.find_by_code(product_subscription.product_code)
         # Find admin email addresses
         admin_emails = UserService.get_admin_emails_for_org(org_id)
+
         if admin_emails != "" and not is_hold:
             Product.send_product_subscription_notification(
                 ProductNotificationInfo(
@@ -519,7 +511,7 @@ def update_product_subscription(product_sub_info: ProductSubscriptionInfo, is_ne
                     is_reapproved=is_reapproved,
                     remarks=product_sub_info.task_remarks,
                     org_id=org_id,
-                    org_name=org_name,
+                    org_name=org_name
                 )
             )
 

auth-api/src/auth_api/utils/enums.py

@@ -328,7 +328,6 @@ class TaskAction(Enum):
     ACCOUNT_REVIEW = "ACCOUNT_REVIEW"
     PRODUCT_REVIEW = "PRODUCT_REVIEW"
     QUALIFIED_SUPPLIER_REVIEW = "QUALIFIED_SUPPLIER_REVIEW"
-    NEW_PRODUCT_FEE_REVIEW = "NEW_PRODUCT_FEE_REVIEW"
 
 
 class ActivityAction(Enum):

auth-api/src/auth_api/utils/roles.py

@@ -15,7 +15,7 @@
 
 from enum import Enum
 
-from .enums import AccessType, OrgStatus, OrgType, ProductSubscriptionStatus, Status
+from .enums import OrgStatus, OrgType, ProductSubscriptionStatus, Status
 
 
 class Role(Enum):
@@ -88,5 +88,3 @@ class Role(Enum):
 EXCLUDED_FIELDS = ("status_code", "type_code")
 
 PREMIUM_ORG_TYPES = (OrgType.PREMIUM.value, OrgType.SBC_STAFF.value, OrgType.STAFF.value)
-
-GOV_ORG_TYPES = (AccessType.GOVM.value, AccessType.GOVN.value)