From d0f57fafd012defd1447675aa459de7774f66525 Mon Sep 17 00:00:00 2001 From: vusallyv Date: Tue, 31 Oct 2023 12:37:50 +0400 Subject: [PATCH] OutputFormats written as enum --- sqlparse/constants.py | 6 ++++++ sqlparse/formatter.py | 5 +++-- 2 files changed, 9 insertions(+), 2 deletions(-) create mode 100644 sqlparse/constants.py diff --git a/sqlparse/constants.py b/sqlparse/constants.py new file mode 100644 index 00000000..d8840445 --- /dev/null +++ b/sqlparse/constants.py @@ -0,0 +1,6 @@ +from enum import Enum + + +class OutputFormats(Enum): + PYTHON = "python" + PHP = "php" diff --git a/sqlparse/formatter.py b/sqlparse/formatter.py index 1d1871cf..349ea000 100644 --- a/sqlparse/formatter.py +++ b/sqlparse/formatter.py @@ -8,6 +8,7 @@ """SQL formatter""" from sqlparse import filters +from sqlparse.constants import OutputFormats from sqlparse.exceptions import SQLParseError @@ -186,9 +187,9 @@ def build_filter_stack(stack, options): # Serializer if options.get('output_format'): frmt = options['output_format'] - if frmt.lower() == 'php': + if frmt.lower() == OutputFormats.PHP.value: fltr = filters.OutputPHPFilter() - elif frmt.lower() == 'python': + elif frmt.lower() == OutputFormats.PYTHON.value: fltr = filters.OutputPythonFilter() else: fltr = None