summaryrefslogtreecommitdiff
path: root/sqlparse
diff options
context:
space:
mode:
authorVictor Uriarte <victor.m.uriarte@intel.com>2016-06-05 13:10:04 -0700
committerVictor Uriarte <victor.m.uriarte@intel.com>2016-06-06 07:47:42 -0700
commitc43fccfefed0806cca52bea48d232ddf72f842cd (patch)
tree6431a781cba6a7cf2915781e8c887743f4178355 /sqlparse
parent0b3aa55ebdde97676ebca6b7fb185ca9739603be (diff)
downloadsqlparse-c43fccfefed0806cca52bea48d232ddf72f842cd.tar.gz
Organize engine module
Diffstat (limited to 'sqlparse')
-rw-r--r--sqlparse/engine/__init__.py44
-rw-r--r--sqlparse/engine/filter_stack.py44
-rw-r--r--sqlparse/engine/statement_splitter.py (renamed from sqlparse/engine/filter.py)2
3 files changed, 53 insertions, 37 deletions
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',
+]
diff --git a/sqlparse/engine/filter_stack.py b/sqlparse/engine/filter_stack.py
new file mode 100644
index 0000000..e51c6f2
--- /dev/null
+++ b/sqlparse/engine/filter_stack.py
@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2016 Andi Albrecht, albrecht.andi@gmail.com
+#
+# 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.statement_splitter import StatementSplitter
+
+
+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 = StatementSplitter().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
diff --git a/sqlparse/engine/filter.py b/sqlparse/engine/statement_splitter.py
index ea2033a..1d1d842 100644
--- a/sqlparse/engine/filter.py
+++ b/sqlparse/engine/statement_splitter.py
@@ -8,7 +8,7 @@
from sqlparse import sql, tokens as T
-class StatementFilter(object):
+class StatementSplitter(object):
"""Filter that split stream at individual statements"""
def __init__(self):