diff options
Diffstat (limited to 'sqlparse')
| -rw-r--r-- | sqlparse/filters.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/sqlparse/filters.py b/sqlparse/filters.py index 8f4b3dd..3906bad 100644 --- a/sqlparse/filters.py +++ b/sqlparse/filters.py @@ -503,11 +503,16 @@ def Tokens2Unicode(stream): return result -class OutputFilter(Filter): +class OutputFilter(TokenFilter): + varname_prefix = '' + def __init__(self, varname='sql'): self.varname = self.varname_prefix + varname self.count = 0 + def _process(self, stream, varname, has_nl): + raise NotImplementedError + def process(self, stack, stmt): self.count += 1 if self.count > 1: @@ -521,8 +526,6 @@ class OutputFilter(Filter): class OutputPythonFilter(OutputFilter): - varname_prefix = '' - def _process(self, stream, varname, has_nl): if self.count > 1: yield sql.Token(T.Whitespace, '\n') |