Given a statement type, return the proper statement generator.
(statement_type)
| 148 | |
| 149 | |
| 150 | def get_generator(statement_type): |
| 151 | """ |
| 152 | Given a statement type, return the proper statement generator. |
| 153 | """ |
| 154 | STATEMENT_GENERATOR_MAP = { |
| 155 | InsertStatement: InsertStatementGenerator, |
| 156 | Query: QueryGenerator, |
| 157 | } |
| 158 | return STATEMENT_GENERATOR_MAP[statement_type] |