moto/moto/rdsdata/responses.py

import json

from moto.core.responses import BaseResponse

from .models import RDSDataServiceBackend, rdsdata_backends


class RDSDataServiceResponse(BaseResponse):
    def __init__(self) -> None:
        super().__init__(service_name="rds-data")

    @property
    def rdsdata_backend(self) -> RDSDataServiceBackend:
        """Return backend instance specific for this region."""
        return rdsdata_backends[self.current_account][self.region]

    def execute_statement(self) -> str:
        resource_arn = self._get_param("resourceArn")
        sql = self._get_param("sql")
        query_result = self.rdsdata_backend.execute_statement(
            resource_arn=resource_arn, sql=sql
        )
        return json.dumps(query_result.to_json())