def _run_execute()

in evalbench/databases/sqlserver.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

        def _run_execute(query: str, eval_query: Optional[str] = None, rollback=False):
            result: List = []
            eval_result: List = []
            error = None
            try:
                with self.engine.connect() as connection:
                    with connection.begin() as transaction:
                        result = self._execute_queries(connection, query)

                        if eval_query:
                            eval_result = self._execute_queries(connection, eval_query)

                        if rollback:
                            transaction.rollback()
            except Exception as e:
                error = str(e)
                if "57P03" in error:
                    raise ResourceExhaustedError("DB Exhausted") from e
            return result, eval_result, error