From c43fccfefed0806cca52bea48d232ddf72f842cd Mon Sep 17 00:00:00 2001 From: Victor Uriarte Date: Sun, 5 Jun 2016 13:10:04 -0700 Subject: Organize engine module --- sqlparse/engine/__init__.py | 44 ++++++++------------------------------------ 1 file changed, 8 insertions(+), 36 deletions(-) (limited to 'sqlparse/engine/__init__.py') diff --git a/sqlparse/engine/__init__.py b/sqlparse/engine/__init__.py index 7f00c57..2c3599e 100644 --- a/sqlparse/engine/__init__.py +++ b/sqlparse/engine/__init__.py @@ -5,40 +5,12 @@ # This module is part of python-sqlparse and is released under # the BSD License: http://www.opensource.org/licenses/bsd-license.php -"""filter""" - -from sqlparse import lexer from sqlparse.engine import grouping -from sqlparse.engine.filter import StatementFilter - - -class FilterStack(object): - def __init__(self): - self.preprocess = [] - self.stmtprocess = [] - self.postprocess = [] - self._grouping = False - - def enable_grouping(self): - self._grouping = True - - def run(self, sql, encoding=None): - stream = lexer.tokenize(sql, encoding) - # Process token stream - for filter_ in self.preprocess: - stream = filter_.process(stream) - - stream = StatementFilter().process(stream) - - # Output: Stream processed Statements - for stmt in stream: - if self._grouping: - stmt = grouping.group(stmt) - - for filter_ in self.stmtprocess: - filter_.process(stmt) - - for filter_ in self.postprocess: - stmt = filter_.process(stmt) - - yield stmt +from sqlparse.engine.filter_stack import FilterStack +from sqlparse.engine.statement_splitter import StatementSplitter + +__all__ = [ + 'grouping', + 'FilterStack', + 'StatementSplitter', +] -- cgit v1.2.1