mirror of
https://github.com/postgres/postgres.git
synced 2026-06-04 06:45:49 -04:00
Peter Eisentraut
5dff93638c
This changes a bunch of incidentially used constructs in the PL/Python regression tests to equivalent constructs in cases where Python 3 no longer supports the old syntax. Support for older Python versions is unchanged.
110 lines
2.2 KiB
Plaintext
110 lines
2.2 KiB
Plaintext
--
|
|
-- Test returning SETOF
|
|
--
|
|
CREATE FUNCTION test_setof_error() RETURNS SETOF text AS $$
|
|
return 37
|
|
$$ LANGUAGE plpythonu;
|
|
SELECT test_setof_error();
|
|
ERROR: returned object cannot be iterated
|
|
DETAIL: PL/Python set-returning functions must return an iterable object.
|
|
CONTEXT: PL/Python function "test_setof_error"
|
|
CREATE FUNCTION test_setof_as_list(count integer, content text) RETURNS SETOF text AS $$
|
|
return [ content ]*count
|
|
$$ LANGUAGE plpythonu;
|
|
CREATE FUNCTION test_setof_as_tuple(count integer, content text) RETURNS SETOF text AS $$
|
|
t = ()
|
|
for i in range(count):
|
|
t += ( content, )
|
|
return t
|
|
$$ LANGUAGE plpythonu;
|
|
CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$
|
|
class producer:
|
|
def __init__ (self, icount, icontent):
|
|
self.icontent = icontent
|
|
self.icount = icount
|
|
def __iter__ (self):
|
|
return self
|
|
def next (self):
|
|
if self.icount == 0:
|
|
raise StopIteration
|
|
self.icount -= 1
|
|
return self.icontent
|
|
return producer(count, content)
|
|
$$ LANGUAGE plpythonu;
|
|
-- Test set returning functions
|
|
SELECT test_setof_as_list(0, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_list(1, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
list
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_list(2, 'list');
|
|
test_setof_as_list
|
|
--------------------
|
|
list
|
|
list
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_list(2, null);
|
|
test_setof_as_list
|
|
--------------------
|
|
|
|
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_tuple(0, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_tuple(1, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
tuple
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_tuple(2, 'tuple');
|
|
test_setof_as_tuple
|
|
---------------------
|
|
tuple
|
|
tuple
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_tuple(2, null);
|
|
test_setof_as_tuple
|
|
---------------------
|
|
|
|
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_iterator(0, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
(0 rows)
|
|
|
|
SELECT test_setof_as_iterator(1, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
list
|
|
(1 row)
|
|
|
|
SELECT test_setof_as_iterator(2, 'list');
|
|
test_setof_as_iterator
|
|
------------------------
|
|
list
|
|
list
|
|
(2 rows)
|
|
|
|
SELECT test_setof_as_iterator(2, null);
|
|
test_setof_as_iterator
|
|
------------------------
|
|
|
|
|
|
(2 rows)
|
|
|