def create()

in fbpcs/infra/cloud_bridge/deployment_helper/aws/aws_deployment_helper_tool.py [0:0]

• 41 lines of code
• 9 McCabe index (conditional complexity)

    def create(self):
        if self.cli_args.add_iam_user:
            if self.cli_args.user_name is None:
                raise Exception(
                    "Need username to add user. Please add username using"
                    " --user_name argument in cli.py"
                )
            self.aws_deployment_helper_obj.create_user_workflow(
                user_name=self.cli_args.user_name
            )

        if self.cli_args.add_iam_policy:
            if self.cli_args.policy_name is None or self.cli_args.region is None:
                raise Exception(
                    "Need policy name to add IAM policy. Please add username using"
                    " --policy_name argument in cli.py"
                )
            policy_params = PolicyParams(
                firehose_stream_name=self.cli_args.firehose_stream_name,
                data_bucket_name=self.cli_args.data_bucket_name,
                config_bucket_name=self.cli_args.config_bucket_name,
                database_name=self.cli_args.database_name,
                data_ingestion_kms_key=self.cli_args.data_ingestion_kms_key,
                cluster_name=self.cli_args.cluster_name,
                ecs_task_execution_role_name=self.cli_args.ecs_task_execution_role_name,
            )
            self.aws_deployment_helper_obj.create_policy(
                policy_name=self.cli_args.policy_name, policy_params=policy_params
            )

        if self.cli_args.attach_iam_policy:
            if (
                self.cli_args.iam_policy_name is None
                or self.cli_args.iam_user_name is None
            ):
                raise Exception(
                    "Need username and policy_name to attach policy to user. Please use"
                    " --user_name and --policy_name arguments in cli.py"
                )
            self.aws_deployment_helper_obj.attach_user_policy(
                policy_name=self.cli_args.iam_policy_name,
                user_name=self.cli_args.iam_user_name,
            )