Source code for pystratum_mysql.wrapper.MySqlRowsWithKeyWrapper

from typing import Dict, Any

from pystratum_common.wrapper.RowsWithKeyWrapper import RowsWithKeyWrapper
from pystratum_mysql.wrapper.MySqlWrapper import MySqlWrapper


class MySqlRowsWithKeyWrapper(RowsWithKeyWrapper, MySqlWrapper):
    """
    Wrapper method generator for stored procedures whose result set must be returned using tree structure using a
    combination of unique columns.
    """

    # ------------------------------------------------------------------------------------------------------------------
    def _write_execute_rows(self, routine: Dict[str, Any]) -> None:
        self._write_line('rows = self.execute_sp_rows({0!s})'.format(self._generate_command(routine)))

# ----------------------------------------------------------------------------------------------------------------------