From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Thu, 16 Mar 2006 23:38:14 +0000 (+0000)
Subject: expanded and integrated qvx's patch for dotted function names
X-Git-Tag: rel_0_1_5~72
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7eedf883703ce845685e14c8838cb38e87211bd2;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

expanded and integrated qvx's patch for dotted function names
---

diff --git a/lib/sqlalchemy/ansisql.py b/lib/sqlalchemy/ansisql.py
index 6a5ef8ef0d..b05872182b 100644
--- a/lib/sqlalchemy/ansisql.py
+++ b/lib/sqlalchemy/ansisql.py
@@ -224,9 +224,9 @@ class ANSICompiler(sql.Compiled):
         if len(self.select_stack):
             self.typemap.setdefault(func.name, func.type)
         if func.name.upper() in ANSI_FUNCS and not len(func.clauses):
-            self.strings[func] = func.name
+            self.strings[func] = ".".join(func.packagenames + [func.name])
         else:
-            self.strings[func] = func.name + "(" + string.join([self.get_str(c) for c in func.clauses], ', ') + ")"
+            self.strings[func] = ".".join(func.packagenames + [func.name]) + "(" + string.join([self.get_str(c) for c in func.clauses], ', ') + ")"
         
     def visit_compound_select(self, cs):
         text = string.join([self.get_str(c) for c in cs.selects], " " + cs.keyword + " ")
diff --git a/lib/sqlalchemy/engine.py b/lib/sqlalchemy/engine.py
index 44ffadfeb3..bee0e6ff6a 100644
--- a/lib/sqlalchemy/engine.py
+++ b/lib/sqlalchemy/engine.py
@@ -280,10 +280,7 @@ class SQLEngine(schema.SchemaEngine):
         return typeobj
 
     def _func(self):
-        class FunctionGateway(object):
-            def __getattr__(s, name):
-                return lambda *c, **kwargs: sql.Function(name, engine=self, *c, **kwargs)
-        return FunctionGateway()
+        return sql.FunctionGenerator(self)
     func = property(_func)
     
     def text(self, text, *args, **kwargs):
diff --git a/lib/sqlalchemy/sql.py b/lib/sqlalchemy/sql.py
index 0df2977437..24263184d1 100644
--- a/lib/sqlalchemy/sql.py
+++ b/lib/sqlalchemy/sql.py
@@ -218,7 +218,7 @@ class FunctionGateway(object):
     """returns a callable based on an attribute name, which then returns a Function 
     object with that name."""
     def __getattr__(self, name):
-        return lambda *c, **kwargs: Function(name, *c, **kwargs)
+        return getattr(FunctionGenerator(), name)
 func = FunctionGateway()
 
 def _compound_clause(keyword, *clauses):
@@ -794,6 +794,7 @@ class Function(ClauseList, ColumnElement):
     def __init__(self, name, *clauses, **kwargs):
         self.name = name
         self.type = kwargs.get('type', sqltypes.NULLTYPE)
+        self.packagenames = kwargs.get('packagenames')
         self._engine = kwargs.get('engine', None)
         if self._engine is not None:
             self.type = self._engine.type_descriptor(self.type)
@@ -827,6 +828,17 @@ class Function(ClauseList, ColumnElement):
         return select([self]).execute()
     def _compare_type(self, obj):
         return self.type
+
+class FunctionGenerator(object):
+    """generates Function objects based on getattr calls"""
+    def __init__(self, engine=None):
+        self.__engine = engine
+        self.__names = []
+    def __getattr__(self, name):
+        self.__names.append(name)
+        return self
+    def __call__(self, *c, **kwargs):
+        return Function(self.__names[-1], packagenames=self.__names[0:-1], engine=self.__engine, *c, **kwargs)     
                 
 class BinaryClause(ClauseElement):
     """represents two clauses with an operator in between"""
diff --git a/test/select.py b/test/select.py
index cec8939d3b..610e05cb40 100644
--- a/test/select.py
+++ b/test/select.py
@@ -343,12 +343,27 @@ FROM mytable, myothertable WHERE foo.id = foofoo(lala) AND datetime(foo) = Today
             "SELECT :literal + :literal_1 FROM mytable")
 
     def testfunction(self):
+        """tests the generation of functions using the func keyword"""
+        # test an expression with a function
         self.runtest(func.lala(3, 4, literal("five"), table1.c.myid) * table2.c.otherid, 
             "lala(:lala, :lala_1, :literal, mytable.myid) * myothertable.otherid")
 
+        # test it in a SELECT
         self.runtest(select([func.count(table1.c.myid)]), 
             "SELECT count(mytable.myid) FROM mytable")
 
+        # test a "dotted" function name
+        self.runtest(select([func.foo.bar.lala(table1.c.myid)]), 
+            "SELECT foo.bar.lala(mytable.myid) FROM mytable")
+
+        # test the bind parameter name with a "dotted" function name is only the name
+        # (limits the length of the bind param name)
+        self.runtest(select([func.foo.bar.lala(12)]), 
+            "SELECT foo.bar.lala(:lala)")
+
+        # test a dotted func off the engine itself
+        self.runtest(db.func.lala.hoho(7), "lala.hoho(:hoho)")
+        
     def testjoin(self):
         self.runtest(
             join(table2, table1, table1.c.myid == table2.c.otherid).select(),