src/pl/plpython/expected/plpython_params.out

   1 --
   2 -- Test named and nameless parameters
   3 --
   4 CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$
   5 return args[0] + args[1]
   6 $$ LANGUAGE plpythonu;
   7 CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$
   8 assert a0 == args[0]
   9 assert a1 == args[1]
  10 return True
  11 $$ LANGUAGE plpythonu;
  12 CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$
  13 assert u == args[0]
  14 if isinstance(u, dict):
  15     # stringify dict the hard way because otherwise the order is implementation-dependent
  16     u_keys = list(u.keys())
  17     u_keys.sort()
  18     s = '{' + ', '.join([repr(k) + ': ' + repr(u[k]) for k in u_keys]) + '}'
  19 else:
  20     s = str(u)
  21 return s
  22 $$ LANGUAGE plpythonu;
  23 -- use deliberately wrong parameter names
  24 CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$
  25 try:
  26         assert a1 == args[0]
  27         return False
  28 except NameError as e:
  29         assert e.args[0].find("a1") > -1
  30         return True
  31 $$ LANGUAGE plpythonu;
  32 SELECT test_param_names0(2,7);
  33  test_param_names0
  34 -------------------
  35                  9
  36 (1 row)
  37
  38 SELECT test_param_names1(1,'text');
  39  test_param_names1
  40 -------------------
  41  t
  42 (1 row)
  43
  44 SELECT test_param_names2(users) from users;
  45                            test_param_names2
  46 -----------------------------------------------------------------------
  47  {'fname': 'jane', 'lname': 'doe', 'userid': 1, 'username': 'j_doe'}
  48  {'fname': 'john', 'lname': 'doe', 'userid': 2, 'username': 'johnd'}
  49  {'fname': 'willem', 'lname': 'doe', 'userid': 3, 'username': 'w_doe'}
  50  {'fname': 'rick', 'lname': 'smith', 'userid': 4, 'username': 'slash'}
  51 (4 rows)
  52
  53 SELECT test_param_names2(NULL);
  54  test_param_names2
  55 -------------------
  56  None
  57 (1 row)
  58
  59 SELECT test_param_names3(1);
  60  test_param_names3
  61 -------------------
  62  t
  63 (1 row)
  64