{devbugfix} batchupload create initial

app/applicationlayer/management/batchupload/views.py

@@ -420,6 +420,11 @@ class BatchUploadViewSet(viewsets.ModelViewSet):
                         excel_code__in=keys['Application'].split(',')
                     )
 
+                    app_excel_code = []
+
+                    for i in app:
+                        app_excel_code.append(i.excel_code)
+
                     current_user.application.set(app)
                     current_user = User.objects.get(id=current_user.id)
 
@@ -442,12 +447,15 @@ class BatchUploadViewSet(viewsets.ModelViewSet):
                 else:
                     current_user = check_user[0]
                 
+                # cms privilege
                 try:
                     privilege_department_code = keys['Change_Request_Privilege_Department_Code']
                     privilege_department_code = Department.objects.get(
                         code=privilege_department_code
                     )
-
+                    # if privilege_department_code:
+                        
+                    #     if 'cms' in app_excel_code:
                     ccr = True if str(keys['Create_CR']).lower() == 'yes' else False
                     crt = True if str(keys['Create_Template']).lower() == 'yes' else False
                     view_all = True if str(keys['View_All_CR']).lower() == 'yes' else False
@@ -464,31 +472,68 @@ class BatchUploadViewSet(viewsets.ModelViewSet):
                         "view_all_change_request_template": view_all_template,
                         "approve_cr": approve_cr
                     }
-
                     AllowedCompany.objects.create(**privilege_object)
+                # else:
+                #     return Response(
+                #         {"message": f"Please indicate cms application at row {data + 2}"},
+                #         status=status.HTTP_400_BAD_REQUEST
+                #     )
 
-                    # Asset Management Privilege
-                    
-                    privilege_department_code = keys[
+                except IntegrityError as e:
+                    # etl2.delete()
+                    # transaction.savepoint_rollback(sp1)
+                    return Response(
+                        {"message": f"Duplicate user privilege at row {data + 2}"},
+                        status=status.HTTP_400_BAD_REQUEST
+                    )
+
+                except ObjectDoesNotExist as e:
+                    # etl2.delete()
+                    transaction.savepoint_rollback(sp1)
+                    return Response(
+                        {"message": f"Department Does not Exist at Privilege row {data + 2}"},
+                        status=status.HTTP_400_BAD_REQUEST
+                    )
+
+                # ams privilege
+                try:
+                    # ams_privilege_department_code = keys['Asset_Management_Privilege_Department_Code']
+                    # if('ams' in app_excel_code and
+                    #         ams_privilege_department_code is None):
+                    #             print("none")
+                    # elif:
+                    #     return Response(
+                    #         {"message": f"Please indicate ams application at row {data + 2}"},
+                    #         status=status.HTTP_400_BAD_REQUEST
+                    #     )
+
+                    ams_privilege_department_code = keys['Asset_Management_Privilege_Department_Code']
+                    ams_privilege_department_code = Department.objects.get(
+                        code=ams_privilege_department_code
+                    )
+
+                    # if ams_privilege_department_code:
+                        
+                    ams_privilege_department_code = keys[
                         'Asset_Management_Privilege_Department_Code']
-                    privilege_department_code = Department.objects.get(
-                        code=privilege_department_code
+                    ams_privilege_department_code = Department.objects.get(
+                        code=ams_privilege_department_code
                     )
                     asset_user_code = keys['Asset_Management_User_Group']
 
                     privilege_object_ams = {
                         "id_number": current_user,
-                        "department": privilege_department_code,
+                        "department": ams_privilege_department_code,
                         "ams_user_type": asset_user_code
                     }
-
+                    
                     AMSPrivilege.objects.create(**privilege_object_ams)
-
+                    
                 except IntegrityError as e:
                     # etl2.delete()
                     # transaction.savepoint_rollback(sp1)
                     return Response(
-                        {"message": f"Duplicate user privilege at row {data + 2}"},
+                        {"message": f"Duplicate ams privilege at row {data + 2}"},
                         status=status.HTTP_400_BAD_REQUEST
                     )
 
@@ -496,13 +541,15 @@ class BatchUploadViewSet(viewsets.ModelViewSet):
                     # etl2.delete()
                     transaction.savepoint_rollback(sp1)
                     return Response(
-                        {"message": f"Department Does not Exist at Privilege row {data + 2}"},
+                        {"message": f"Department Does not Exist at AMS Privilege row {data + 2}"},
                         status=status.HTTP_400_BAD_REQUEST
                     )
+                
+                # end of ams privilege
 
             except IntegrityError as e:
                 # etl2.delete()
-                transaction.savepoint_rollback(sp1)
+                # transaction.savepoint_rollback(sp1)
                 return Response(
                     {"message": f"Record already exist at row {data + 2}"},
                     status=status.HTTP_400_BAD_REQUEST
@@ -520,7 +567,7 @@ class BatchUploadViewSet(viewsets.ModelViewSet):
                 # etl2.delete()
                 transaction.savepoint_rollback(sp1)
                 return Response(
-                    {"message": f"Department Does not Exist at rowaaaa {data + 2}"},
+                    {"message": f"Department Does not Exist at row {data + 2}"},
                     status=status.HTTP_400_BAD_REQUEST
                 )