src/pl/plpython/expected/plpython_setof.out

   1 --
   2 -- Test returning SETOF
   3 --
   4 CREATE FUNCTION test_setof_error() RETURNS SETOF text AS $$
   5 return 37
   6 $$ LANGUAGE plpythonu;
   7 SELECT test_setof_error();
   8 ERROR:  returned object cannot be iterated
   9 DETAIL:  PL/Python set-returning functions must return an iterable object.
  10 CONTEXT:  PL/Python function "test_setof_error"
  11 CREATE FUNCTION test_setof_as_list(count integer, content text) RETURNS SETOF text AS $$
  12 return [ content ]*count
  13 $$ LANGUAGE plpythonu;
  14 CREATE FUNCTION test_setof_as_tuple(count integer, content text) RETURNS SETOF text AS $$
  15 t = ()
  16 for i in range(count):
  17         t += ( content, )
  18 return t
  19 $$ LANGUAGE plpythonu;
  20 CREATE FUNCTION test_setof_as_iterator(count integer, content text) RETURNS SETOF text AS $$
  21 class producer:
  22         def __init__ (self, icount, icontent):
  23                 self.icontent = icontent
  24                 self.icount = icount
  25         def __iter__ (self):
  26                 return self
  27         def next (self):
  28                 if self.icount == 0:
  29                         raise StopIteration
  30                 self.icount -= 1
  31                 return self.icontent
  32 return producer(count, content)
  33 $$ LANGUAGE plpythonu;
  34 CREATE FUNCTION test_setof_spi_in_iterator() RETURNS SETOF text AS
  35 $$
  36     for s in ('Hello', 'Brave', 'New', 'World'):
  37         plpy.execute('select 1')
  38         yield s
  39         plpy.execute('select 2')
  40 $$
  41 LANGUAGE plpythonu;
  42 -- Test set returning functions
  43 SELECT test_setof_as_list(0, 'list');
  44  test_setof_as_list
  45 --------------------
  46 (0 rows)
  47
  48 SELECT test_setof_as_list(1, 'list');
  49  test_setof_as_list
  50 --------------------
  51  list
  52 (1 row)
  53
  54 SELECT test_setof_as_list(2, 'list');
  55  test_setof_as_list
  56 --------------------
  57  list
  58  list
  59 (2 rows)
  60
  61 SELECT test_setof_as_list(2, null);
  62  test_setof_as_list
  63 --------------------
  64
  65
  66 (2 rows)
  67
  68 SELECT test_setof_as_tuple(0, 'tuple');
  69  test_setof_as_tuple
  70 ---------------------
  71 (0 rows)
  72
  73 SELECT test_setof_as_tuple(1, 'tuple');
  74  test_setof_as_tuple
  75 ---------------------
  76  tuple
  77 (1 row)
  78
  79 SELECT test_setof_as_tuple(2, 'tuple');
  80  test_setof_as_tuple
  81 ---------------------
  82  tuple
  83  tuple
  84 (2 rows)
  85
  86 SELECT test_setof_as_tuple(2, null);
  87  test_setof_as_tuple
  88 ---------------------
  89
  90
  91 (2 rows)
  92
  93 SELECT test_setof_as_iterator(0, 'list');
  94  test_setof_as_iterator
  95 ------------------------
  96 (0 rows)
  97
  98 SELECT test_setof_as_iterator(1, 'list');
  99  test_setof_as_iterator
 100 ------------------------
 101  list
 102 (1 row)
 103
 104 SELECT test_setof_as_iterator(2, 'list');
 105  test_setof_as_iterator
 106 ------------------------
 107  list
 108  list
 109 (2 rows)
 110
 111 SELECT test_setof_as_iterator(2, null);
 112  test_setof_as_iterator
 113 ------------------------
 114
 115
 116 (2 rows)
 117
 118 SELECT test_setof_spi_in_iterator();
 119  test_setof_spi_in_iterator
 120 ----------------------------
 121  Hello
 122  Brave
 123  New
 124  World
 125 (4 rows)
 126
 127 -- set-returning function that modifies its parameters
 128 CREATE OR REPLACE FUNCTION ugly(x int, lim int) RETURNS SETOF int AS $$
 129 global x
 130 while x <= lim:
 131     yield x
 132     x = x + 1
 133 $$ LANGUAGE plpythonu;
 134 SELECT ugly(1, 5);
 135  ugly
 136 ------
 137     1
 138     2
 139     3
 140     4
 141     5
 142 (5 rows)
 143
 144 -- interleaved execution of such a function
 145 SELECT ugly(1,3), ugly(7,8);
 146  ugly | ugly
 147 ------+------
 148     1 |    7
 149     2 |    8
 150     3 |
 151 (3 rows)
 152
 153 -- returns set of named-composite-type tuples
 154 CREATE OR REPLACE FUNCTION get_user_records()
 155 RETURNS SETOF users
 156 AS $$
 157     return plpy.execute("SELECT * FROM users ORDER BY username")
 158 $$ LANGUAGE plpythonu;
 159 SELECT get_user_records();
 160    get_user_records
 161 ----------------------
 162  (jane,doe,j_doe,1)
 163  (john,doe,johnd,2)
 164  (rick,smith,slash,4)
 165  (willem,doe,w_doe,3)
 166 (4 rows)
 167
 168 SELECT * FROM get_user_records();
 169  fname  | lname | username | userid
 170 --------+-------+----------+--------
 171  jane   | doe   | j_doe    |      1
 172  john   | doe   | johnd    |      2
 173  rick   | smith | slash    |      4
 174  willem | doe   | w_doe    |      3
 175 (4 rows)
 176
 177 -- same, but returning set of RECORD
 178 CREATE OR REPLACE FUNCTION get_user_records2()
 179 RETURNS TABLE(fname text, lname text, username text, userid int)
 180 AS $$
 181     return plpy.execute("SELECT * FROM users ORDER BY username")
 182 $$ LANGUAGE plpythonu;
 183 SELECT get_user_records2();
 184   get_user_records2
 185 ----------------------
 186  (jane,doe,j_doe,1)
 187  (john,doe,johnd,2)
 188  (rick,smith,slash,4)
 189  (willem,doe,w_doe,3)
 190 (4 rows)
 191
 192 SELECT * FROM get_user_records2();
 193  fname  | lname | username | userid
 194 --------+-------+----------+--------
 195  jane   | doe   | j_doe    |      1
 196  john   | doe   | johnd    |      2
 197  rick   | smith | slash    |      4
 198  willem | doe   | w_doe    |      3
 199 (4 rows)
 200