FilesExpand file tree

 master

/

functions.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

44 lines (30 loc) · 991 Bytes

 master

/

functions.py

Copy path

Top

File metadata and controls

• Code
• Blame

44 lines (30 loc) · 991 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

'''

Created on 17/05/2012

@author: piranna

Several utility functions to extract info from the SQL sentences

'''

from sqlparse.filters import ColumnsSelect, Limit

from sqlparse.pipeline import Pipeline

from sqlparse.tokens import Keyword, Whitespace

def getlimit(stream):

"""Function that return the LIMIT of a input SQL """

pipe = Pipeline()

pipe.append(Limit())

result = pipe(stream)

try:

return int(result)

except ValueError:

return result

def getcolumns(stream):

"""Function that return the colums of a SELECT query"""

pipe = Pipeline()

pipe.append(ColumnsSelect())

return pipe(stream)

class IsType(object):

"""Functor that return is the statement is of a specific type"""

def __init__(self, type):

self.type = type

def __call__(self, stream):

for token_type, value in stream:

if token_type not in Whitespace:

return token_type in Keyword and value == self.type