From 895f021a0515dbf948efb1dfe960c0aa63cd160d Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Wed, 27 May 2009 20:58:12 +0200 Subject: Grouping of function/procedure calls. --- sqlparse/sql.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'sqlparse/sql.py') diff --git a/sqlparse/sql.py b/sqlparse/sql.py index 7c607c4..4502fa2 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -455,3 +455,17 @@ class Case(TokenList): elif in_value: ret[-1][1].append(token) return ret + + +class Function(TokenList): + """A function or procedure call.""" + + __slots__ = ('value', 'ttype', 'tokens') + + def get_parameters(self): + """Return a list of parameters.""" + parenthesis = self.tokens[-1] + for t in parenthesis.tokens: + if isinstance(t, IdentifierList): + return t.get_identifiers() + return [] -- cgit v1.2.1