summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/sql/functions.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2018-08-22 11:13:54 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2018-08-22 11:13:54 -0400
commit52a3f5b7635583ae6feb084b1db654b9c65caec2 (patch)
treedd2a70b725f93d17321fd651a9ef8da35d1de829 /lib/sqlalchemy/sql/functions.py
parent32ce703a98eba8a7685e609b4a7ca86b79dd0904 (diff)
downloadsqlalchemy-52a3f5b7635583ae6feb084b1db654b9c65caec2.tar.gz
Pass desired array type from pg.array_agg to functions.array_agg
Fixed the :func:`.postgresql.array_agg` function, which is a slightly altered version of the usual :func:`.functions.array_agg` function, to also accept an incoming "type" argument without forcing an ARRAY around it, essentially the same thing that was fixed for the generic function in 1.1 in :ticket:`4107`. Fixes: #4324 Change-Id: I399a29f59c945a217cdd22c65ff0325edea8ea65
Diffstat (limited to 'lib/sqlalchemy/sql/functions.py')
-rw-r--r--lib/sqlalchemy/sql/functions.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/sqlalchemy/sql/functions.py b/lib/sqlalchemy/sql/functions.py
index 27d030d4f..5cea7750a 100644
--- a/lib/sqlalchemy/sql/functions.py
+++ b/lib/sqlalchemy/sql/functions.py
@@ -793,13 +793,14 @@ class array_agg(GenericFunction):
def __init__(self, *args, **kwargs):
args = [_literal_as_binds(c) for c in args]
+ default_array_type = kwargs.pop('_default_array_type', sqltypes.ARRAY)
if 'type_' not in kwargs:
type_from_args = _type_from_args(args)
if isinstance(type_from_args, sqltypes.ARRAY):
kwargs['type_'] = type_from_args
else:
- kwargs['type_'] = sqltypes.ARRAY(type_from_args)
+ kwargs['type_'] = default_array_type(type_from_args)
kwargs['_parsed_args'] = args
super(array_agg, self).__init__(*args, **kwargs)