43 lines
1.1 KiB
PL/PgSQL
43 lines
1.1 KiB
PL/PgSQL
--
|
|
-- Test named and nameless parameters
|
|
--
|
|
|
|
CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$
|
|
return args[0] + args[1]
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$
|
|
assert a0 == args[0]
|
|
assert a1 == args[1]
|
|
return True
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$
|
|
assert u == args[0]
|
|
if isinstance(u, dict):
|
|
# stringify dict the hard way because otherwise the order is implementation-dependent
|
|
u_keys = list(u.keys())
|
|
u_keys.sort()
|
|
s = '{' + ', '.join([repr(k) + ': ' + repr(u[k]) for k in u_keys]) + '}'
|
|
else:
|
|
s = str(u)
|
|
return s
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
-- use deliberately wrong parameter names
|
|
CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$
|
|
try:
|
|
assert a1 == args[0]
|
|
return False
|
|
except NameError as e:
|
|
assert e.args[0].find("a1") > -1
|
|
return True
|
|
$$ LANGUAGE plpython3u;
|
|
|
|
|
|
SELECT test_param_names0(2,7);
|
|
SELECT test_param_names1(1,'text');
|
|
SELECT test_param_names2(users) from users;
|
|
SELECT test_param_names2(NULL);
|
|
SELECT test_param_names3(1);
|