mirror of https://github.com/postgres/postgres
of the previous monolithic setup-create-run sequence, that was apparently inherited from a previous test infrastructure, but makes working with the tests and adding new ones weird.REL8_5_ALPHA1_BRANCH
Peter Eisentraut
16 years ago
parent
ef7574eb01
commit
9d9848668f
@ -0,0 +1,8 @@ |
|||||||
|
Guide to alternative expected files: |
||||||
|
|
||||||
|
plpython_error_2.out Python 2.2, 2.3, 2.4 |
||||||
|
plpython_error.out Python 2.5, 2.6 |
||||||
|
|
||||||
|
plpython_unicode_2.out Python 2.2 |
||||||
|
plpython_unicode_3.out Python 2.3, 2.4 |
||||||
|
plpython_unicode.out Python 2.5, 2.6 |
||||||
@ -1,38 +0,0 @@ |
|||||||
-- test error handling, i forgot to restore Warn_restart in |
|
||||||
-- the trigger handler once. the errors and subsequent core dump were |
|
||||||
-- interesting. |
|
||||||
SELECT invalid_type_uncaught('rick'); |
|
||||||
WARNING: PL/Python: in PL/Python function "invalid_type_uncaught" |
|
||||||
DETAIL: plpy.SPIError: unrecognized error in PLy_spi_prepare |
|
||||||
ERROR: type "test" does not exist |
|
||||||
SELECT invalid_type_caught('rick'); |
|
||||||
WARNING: PL/Python: in PL/Python function "invalid_type_caught" |
|
||||||
DETAIL: plpy.SPIError: unrecognized error in PLy_spi_prepare |
|
||||||
ERROR: type "test" does not exist |
|
||||||
SELECT invalid_type_reraised('rick'); |
|
||||||
WARNING: PL/Python: in PL/Python function "invalid_type_reraised" |
|
||||||
DETAIL: plpy.SPIError: unrecognized error in PLy_spi_prepare |
|
||||||
ERROR: type "test" does not exist |
|
||||||
SELECT valid_type('rick'); |
|
||||||
valid_type |
|
||||||
------------ |
|
||||||
|
|
||||||
(1 row) |
|
||||||
|
|
||||||
-- |
|
||||||
-- Test Unicode error handling. |
|
||||||
-- |
|
||||||
SELECT unicode_return_error(); |
|
||||||
ERROR: PL/Python: could not create string representation of Python object in PL/Python function "unicode_return_error" while creating return value |
|
||||||
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character '\u80' in position 0: ordinal not in range(128) |
|
||||||
INSERT INTO unicode_test (testvalue) VALUES ('test'); |
|
||||||
ERROR: PL/Python: could not compute string representation of Python object in PL/Python function "unicode_trigger_error" while modifying trigger row |
|
||||||
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character '\u80' in position 0: ordinal not in range(128) |
|
||||||
SELECT unicode_plan_error1(); |
|
||||||
WARNING: PL/Python: in PL/Python function "unicode_plan_error1" |
|
||||||
DETAIL: plpy.Error: unrecognized error in PLy_spi_execute_plan |
|
||||||
ERROR: PL/Python: PL/Python function "unicode_plan_error1" could not execute plan |
|
||||||
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character '\u80' in position 0: ordinal not in range(128) |
|
||||||
SELECT unicode_plan_error2(); |
|
||||||
ERROR: PL/Python: PL/Python function "unicode_plan_error2" could not execute plan |
|
||||||
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character '\u80' in position 0: ordinal not in range(128) |
|
||||||
@ -1,482 +0,0 @@ |
|||||||
CREATE FUNCTION global_test_one() returns text |
|
||||||
AS |
|
||||||
'if not SD.has_key("global_test"): |
|
||||||
SD["global_test"] = "set by global_test_one" |
|
||||||
if not GD.has_key("global_test"): |
|
||||||
GD["global_test"] = "set by global_test_one" |
|
||||||
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION global_test_two() returns text |
|
||||||
AS |
|
||||||
'if not SD.has_key("global_test"): |
|
||||||
SD["global_test"] = "set by global_test_two" |
|
||||||
if not GD.has_key("global_test"): |
|
||||||
GD["global_test"] = "set by global_test_two" |
|
||||||
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION static_test() returns int4 |
|
||||||
AS |
|
||||||
'if SD.has_key("call"): |
|
||||||
SD["call"] = SD["call"] + 1 |
|
||||||
else: |
|
||||||
SD["call"] = 1 |
|
||||||
return SD["call"] |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
-- import python modules |
|
||||||
CREATE FUNCTION import_fail() returns text |
|
||||||
AS |
|
||||||
'try: |
|
||||||
import foosocket |
|
||||||
except Exception, ex: |
|
||||||
plpy.notice("import socket failed -- %s" % str(ex)) |
|
||||||
return "failed as expected" |
|
||||||
return "succeeded, that wasn''t supposed to happen"' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION import_succeed() returns text |
|
||||||
AS |
|
||||||
'try: |
|
||||||
import array |
|
||||||
import bisect |
|
||||||
import calendar |
|
||||||
import cmath |
|
||||||
import errno |
|
||||||
import math |
|
||||||
import md5 |
|
||||||
import operator |
|
||||||
import random |
|
||||||
import re |
|
||||||
import sha |
|
||||||
import string |
|
||||||
import time |
|
||||||
except Exception, ex: |
|
||||||
plpy.notice("import failed -- %s" % str(ex)) |
|
||||||
return "failed, that wasn''t supposed to happen" |
|
||||||
return "succeeded, as expected"' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION import_test_one(p text) RETURNS text |
|
||||||
AS |
|
||||||
'import sha |
|
||||||
digest = sha.new(p) |
|
||||||
return digest.hexdigest()' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION import_test_two(u users) RETURNS text |
|
||||||
AS |
|
||||||
'import sha |
|
||||||
plain = u["fname"] + u["lname"] |
|
||||||
digest = sha.new(plain); |
|
||||||
return "sha hash of " + plain + " is " + digest.hexdigest()' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION argument_test_one(u users, a1 text, a2 text) RETURNS text |
|
||||||
AS |
|
||||||
'keys = u.keys() |
|
||||||
keys.sort() |
|
||||||
out = [] |
|
||||||
for key in keys: |
|
||||||
out.append("%s: %s" % (key, u[key])) |
|
||||||
words = a1 + " " + a2 + " => {" + ", ".join(out) + "}" |
|
||||||
return words' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
-- these triggers are dedicated to HPHC of RI who |
|
||||||
-- decided that my kid's name was william not willem, and |
|
||||||
-- vigorously resisted all efforts at correction. they have |
|
||||||
-- since gone bankrupt... |
|
||||||
CREATE FUNCTION users_insert() returns trigger |
|
||||||
AS |
|
||||||
'if TD["new"]["fname"] == None or TD["new"]["lname"] == None: |
|
||||||
return "SKIP" |
|
||||||
if TD["new"]["username"] == None: |
|
||||||
TD["new"]["username"] = TD["new"]["fname"][:1] + "_" + TD["new"]["lname"] |
|
||||||
rv = "MODIFY" |
|
||||||
else: |
|
||||||
rv = None |
|
||||||
if TD["new"]["fname"] == "william": |
|
||||||
TD["new"]["fname"] = TD["args"][0] |
|
||||||
rv = "MODIFY" |
|
||||||
return rv' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION users_update() returns trigger |
|
||||||
AS |
|
||||||
'if TD["event"] == "UPDATE": |
|
||||||
if TD["old"]["fname"] != TD["new"]["fname"] and TD["old"]["fname"] == TD["args"][0]: |
|
||||||
return "SKIP" |
|
||||||
return None' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION users_delete() RETURNS trigger |
|
||||||
AS |
|
||||||
'if TD["old"]["fname"] == TD["args"][0]: |
|
||||||
return "SKIP" |
|
||||||
return None' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE TRIGGER users_insert_trig BEFORE INSERT ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_insert ('willem'); |
|
||||||
CREATE TRIGGER users_update_trig BEFORE UPDATE ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_update ('willem'); |
|
||||||
CREATE TRIGGER users_delete_trig BEFORE DELETE ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_delete ('willem'); |
|
||||||
-- dump trigger data |
|
||||||
CREATE TABLE trigger_test |
|
||||||
(i int, v text ); |
|
||||||
CREATE FUNCTION trigger_data() returns trigger language plpythonu as $$ |
|
||||||
|
|
||||||
if TD.has_key('relid'): |
|
||||||
TD['relid'] = "bogus:12345" |
|
||||||
|
|
||||||
skeys = TD.keys() |
|
||||||
skeys.sort() |
|
||||||
for key in skeys: |
|
||||||
val = TD[key] |
|
||||||
plpy.notice("TD[" + key + "] => " + str(val)) |
|
||||||
|
|
||||||
return None |
|
||||||
|
|
||||||
$$; |
|
||||||
CREATE TRIGGER show_trigger_data_trig |
|
||||||
BEFORE INSERT OR UPDATE OR DELETE ON trigger_test |
|
||||||
FOR EACH ROW EXECUTE PROCEDURE trigger_data(23,'skidoo'); |
|
||||||
insert into trigger_test values(1,'insert'); |
|
||||||
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[event] => INSERT',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[level] => ROW',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[name] => show_trigger_data_trig',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ("TD[new] => {'i': 1, 'v': 'insert'}",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[old] => None',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[relid] => bogus:12345',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_name] => trigger_test',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_schema] => public',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[when] => BEFORE',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
update trigger_test set v = 'update' where i = 1; |
|
||||||
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[event] => UPDATE',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[level] => ROW',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[name] => show_trigger_data_trig',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ("TD[new] => {'i': 1, 'v': 'update'}",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ("TD[old] => {'i': 1, 'v': 'insert'}",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[relid] => bogus:12345',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_name] => trigger_test',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_schema] => public',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[when] => BEFORE',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
delete from trigger_test; |
|
||||||
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[event] => DELETE',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[level] => ROW',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[name] => show_trigger_data_trig',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[new] => None',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ("TD[old] => {'i': 1, 'v': 'update'}",) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[relid] => bogus:12345',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_name] => trigger_test',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[table_schema] => public',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
NOTICE: ('TD[when] => BEFORE',) |
|
||||||
CONTEXT: PL/Python function "trigger_data" |
|
||||||
|
|
||||||
DROP TRIGGER show_trigger_data_trig on trigger_test; |
|
||||||
|
|
||||||
DROP FUNCTION trigger_data(); |
|
||||||
-- nested calls |
|
||||||
-- |
|
||||||
CREATE FUNCTION nested_call_one(a text) RETURNS text |
|
||||||
AS |
|
||||||
'q = "SELECT nested_call_two(''%s'')" % a |
|
||||||
r = plpy.execute(q) |
|
||||||
return r[0]' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
CREATE FUNCTION nested_call_two(a text) RETURNS text |
|
||||||
AS |
|
||||||
'q = "SELECT nested_call_three(''%s'')" % a |
|
||||||
r = plpy.execute(q) |
|
||||||
return r[0]' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
CREATE FUNCTION nested_call_three(a text) RETURNS text |
|
||||||
AS |
|
||||||
'return a' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
-- some spi stuff |
|
||||||
CREATE FUNCTION spi_prepared_plan_test_one(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("myplan"): |
|
||||||
q = "SELECT count(*) FROM users WHERE lname = $1" |
|
||||||
SD["myplan"] = plpy.prepare(q, [ "text" ]) |
|
||||||
try: |
|
||||||
rv = plpy.execute(SD["myplan"], [a]) |
|
||||||
return "there are " + str(rv[0]["count"]) + " " + str(a) + "s" |
|
||||||
except Exception, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION spi_prepared_plan_test_nested(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("myplan"): |
|
||||||
q = "SELECT spi_prepared_plan_test_one(''%s'') as count" % a |
|
||||||
SD["myplan"] = plpy.prepare(q) |
|
||||||
try: |
|
||||||
rv = plpy.execute(SD["myplan"]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["count"] |
|
||||||
except Exception, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* really stupid function just to get the module loaded |
|
||||||
*/ |
|
||||||
CREATE FUNCTION stupid() RETURNS text AS 'return "zarkon"' LANGUAGE plpythonu; |
|
||||||
/* a typo |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_uncaught(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* for what it's worth catch the exception generated by |
|
||||||
* the typo, and return None |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_caught(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
try: |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
except plpy.SPIError, ex: |
|
||||||
plpy.notice(str(ex)) |
|
||||||
return None |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* for what it's worth catch the exception generated by |
|
||||||
* the typo, and reraise it as a plain error |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_reraised(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
try: |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
except plpy.SPIError, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* no typo no messing about |
|
||||||
*/ |
|
||||||
CREATE FUNCTION valid_type(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
SD["plan"] = plpy.prepare("SELECT fname FROM users WHERE lname = $1", [ "text" ]) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* Flat out syntax error |
|
||||||
*/ |
|
||||||
CREATE FUNCTION sql_syntax_error() RETURNS text |
|
||||||
AS |
|
||||||
'plpy.execute("syntax error")' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* check the handling of uncaught python exceptions |
|
||||||
*/ |
|
||||||
CREATE FUNCTION exception_index_invalid(text) RETURNS text |
|
||||||
AS |
|
||||||
'return args[1]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* check handling of nested exceptions |
|
||||||
*/ |
|
||||||
CREATE FUNCTION exception_index_invalid_nested() RETURNS text |
|
||||||
AS |
|
||||||
'rv = plpy.execute("SELECT test5(''foo'')") |
|
||||||
return rv[0]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION join_sequences(s sequences) RETURNS text |
|
||||||
AS |
|
||||||
'if not s["multipart"]: |
|
||||||
return s["sequence"] |
|
||||||
q = "SELECT sequence FROM xsequences WHERE pid = ''%s''" % s["pid"] |
|
||||||
rv = plpy.execute(q) |
|
||||||
seq = s["sequence"] |
|
||||||
for r in rv: |
|
||||||
seq = seq + r["sequence"] |
|
||||||
return seq |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
-- |
|
||||||
-- Universal Newline Support |
|
||||||
-- |
|
||||||
CREATE OR REPLACE FUNCTION newline_lf() RETURNS integer AS |
|
||||||
E'x = 100\ny = 23\nreturn x + y\n' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE OR REPLACE FUNCTION newline_cr() RETURNS integer AS |
|
||||||
E'x = 100\ry = 23\rreturn x + y\r' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
CREATE OR REPLACE FUNCTION newline_crlf() RETURNS integer AS |
|
||||||
E'x = 100\r\ny = 23\r\nreturn x + y\r\n' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
-- |
|
||||||
-- Unicode error handling |
|
||||||
-- |
|
||||||
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
|
||||||
return u"\\x80" |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
|
||||||
TD["new"]["testvalue"] = u"\\x80" |
|
||||||
return "MODIFY" |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
|
||||||
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
|
||||||
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
|
||||||
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
|
||||||
rv = plpy.execute(plan, [u"\\x80"], 1) |
|
||||||
return rv[0]["testvalue"] |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
|
||||||
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
|
||||||
rv = plpy.execute(plan, u"\\x80", 1) |
|
||||||
return rv[0]["testvalue1"] |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
-- Tests for functions that return void |
|
||||||
CREATE FUNCTION test_void_func1() RETURNS void AS $$ |
|
||||||
x = 10 |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
-- illegal: can't return non-None value in void-returning func |
|
||||||
CREATE FUNCTION test_void_func2() RETURNS void AS $$ |
|
||||||
return 10 |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_return_none() RETURNS int AS $$ |
|
||||||
None |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
-- |
|
||||||
-- Test named and nameless parameters |
|
||||||
-- |
|
||||||
CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$ |
|
||||||
return args[0] + args[1] |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$ |
|
||||||
assert a0 == args[0] |
|
||||||
assert a1 == args[1] |
|
||||||
return True |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$ |
|
||||||
assert u == args[0] |
|
||||||
return str(u) |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
-- use deliberately wrong parameter names |
|
||||||
CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$ |
|
||||||
try: |
|
||||||
assert a1 == args[0] |
|
||||||
return False |
|
||||||
except NameError, e: |
|
||||||
assert e.args[0].find("a1") > -1 |
|
||||||
return True |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
-- |
|
||||||
-- Test returning SETOF |
|
||||||
-- |
|
||||||
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 xrange(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 returning tuples |
|
||||||
-- |
|
||||||
CREATE FUNCTION test_table_record_as(typ text, first text, second integer, retnull boolean) RETURNS table_record AS $$ |
|
||||||
if retnull: |
|
||||||
return None |
|
||||||
if typ == 'dict': |
|
||||||
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
|
||||||
elif typ == 'tuple': |
|
||||||
return ( first, second ) |
|
||||||
elif typ == 'list': |
|
||||||
return [ first, second ] |
|
||||||
elif typ == 'obj': |
|
||||||
class type_record: pass |
|
||||||
type_record.first = first |
|
||||||
type_record.second = second |
|
||||||
return type_record |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_type_record_as(typ text, first text, second integer, retnull boolean) RETURNS type_record AS $$ |
|
||||||
if retnull: |
|
||||||
return None |
|
||||||
if typ == 'dict': |
|
||||||
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
|
||||||
elif typ == 'tuple': |
|
||||||
return ( first, second ) |
|
||||||
elif typ == 'list': |
|
||||||
return [ first, second ] |
|
||||||
elif typ == 'obj': |
|
||||||
class type_record: pass |
|
||||||
type_record.first = first |
|
||||||
type_record.second = second |
|
||||||
return type_record |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_in_out_params(first in text, second out text) AS $$ |
|
||||||
return first + '_in_to_out'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
-- this doesn't work yet :-( |
|
||||||
CREATE FUNCTION test_in_out_params_multi(first in text, |
|
||||||
second out text, third out text) AS $$ |
|
||||||
return first + '_record_in_to_out'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
CREATE FUNCTION test_inout_params(first inout text) AS $$ |
|
||||||
return first + '_inout'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
@ -0,0 +1,52 @@ |
|||||||
|
-- |
||||||
|
-- check static and global data (SD and GD) |
||||||
|
-- |
||||||
|
CREATE FUNCTION global_test_one() returns text |
||||||
|
AS |
||||||
|
'if not SD.has_key("global_test"): |
||||||
|
SD["global_test"] = "set by global_test_one" |
||||||
|
if not GD.has_key("global_test"): |
||||||
|
GD["global_test"] = "set by global_test_one" |
||||||
|
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION global_test_two() returns text |
||||||
|
AS |
||||||
|
'if not SD.has_key("global_test"): |
||||||
|
SD["global_test"] = "set by global_test_two" |
||||||
|
if not GD.has_key("global_test"): |
||||||
|
GD["global_test"] = "set by global_test_two" |
||||||
|
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION static_test() returns int4 |
||||||
|
AS |
||||||
|
'if SD.has_key("call"): |
||||||
|
SD["call"] = SD["call"] + 1 |
||||||
|
else: |
||||||
|
SD["call"] = 1 |
||||||
|
return SD["call"] |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
SELECT static_test(); |
||||||
|
static_test |
||||||
|
------------- |
||||||
|
1 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT static_test(); |
||||||
|
static_test |
||||||
|
------------- |
||||||
|
2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT global_test_one(); |
||||||
|
global_test_one |
||||||
|
-------------------------------------------------------- |
||||||
|
SD: set by global_test_one, GD: set by global_test_one |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT global_test_two(); |
||||||
|
global_test_two |
||||||
|
-------------------------------------------------------- |
||||||
|
SD: set by global_test_two, GD: set by global_test_one |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -0,0 +1,76 @@ |
|||||||
|
-- import python modules |
||||||
|
CREATE FUNCTION import_fail() returns text |
||||||
|
AS |
||||||
|
'try: |
||||||
|
import foosocket |
||||||
|
except Exception, ex: |
||||||
|
plpy.notice("import socket failed -- %s" % str(ex)) |
||||||
|
return "failed as expected" |
||||||
|
return "succeeded, that wasn''t supposed to happen"' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION import_succeed() returns text |
||||||
|
AS |
||||||
|
'try: |
||||||
|
import array |
||||||
|
import bisect |
||||||
|
import calendar |
||||||
|
import cmath |
||||||
|
import errno |
||||||
|
import math |
||||||
|
import md5 |
||||||
|
import operator |
||||||
|
import random |
||||||
|
import re |
||||||
|
import sha |
||||||
|
import string |
||||||
|
import time |
||||||
|
except Exception, ex: |
||||||
|
plpy.notice("import failed -- %s" % str(ex)) |
||||||
|
return "failed, that wasn''t supposed to happen" |
||||||
|
return "succeeded, as expected"' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION import_test_one(p text) RETURNS text |
||||||
|
AS |
||||||
|
'import sha |
||||||
|
digest = sha.new(p) |
||||||
|
return digest.hexdigest()' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION import_test_two(u users) RETURNS text |
||||||
|
AS |
||||||
|
'import sha |
||||||
|
plain = u["fname"] + u["lname"] |
||||||
|
digest = sha.new(plain); |
||||||
|
return "sha hash of " + plain + " is " + digest.hexdigest()' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
-- import python modules |
||||||
|
-- |
||||||
|
SELECT import_fail(); |
||||||
|
NOTICE: ('import socket failed -- No module named foosocket',) |
||||||
|
CONTEXT: PL/Python function "import_fail" |
||||||
|
import_fail |
||||||
|
-------------------- |
||||||
|
failed as expected |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT import_succeed(); |
||||||
|
import_succeed |
||||||
|
------------------------ |
||||||
|
succeeded, as expected |
||||||
|
(1 row) |
||||||
|
|
||||||
|
-- test import and simple argument handling |
||||||
|
-- |
||||||
|
SELECT import_test_one('sha hash of this string'); |
||||||
|
import_test_one |
||||||
|
------------------------------------------ |
||||||
|
a04e23cb9b1a09cd1051a04a7c571aae0f90346c |
||||||
|
(1 row) |
||||||
|
|
||||||
|
-- test import and tuple argument handling |
||||||
|
-- |
||||||
|
select import_test_two(users) from users where fname = 'willem'; |
||||||
|
import_test_two |
||||||
|
------------------------------------------------------------------- |
||||||
|
sha hash of willemdoe is 3cde6b574953b0ca937b4d76ebc40d534d910759 |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -0,0 +1,30 @@ |
|||||||
|
-- |
||||||
|
-- Universal Newline Support |
||||||
|
-- |
||||||
|
CREATE OR REPLACE FUNCTION newline_lf() RETURNS integer AS |
||||||
|
E'x = 100\ny = 23\nreturn x + y\n' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE OR REPLACE FUNCTION newline_cr() RETURNS integer AS |
||||||
|
E'x = 100\ry = 23\rreturn x + y\r' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE OR REPLACE FUNCTION newline_crlf() RETURNS integer AS |
||||||
|
E'x = 100\r\ny = 23\r\nreturn x + y\r\n' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
SELECT newline_lf(); |
||||||
|
newline_lf |
||||||
|
------------ |
||||||
|
123 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT newline_cr(); |
||||||
|
newline_cr |
||||||
|
------------ |
||||||
|
123 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT newline_crlf(); |
||||||
|
newline_crlf |
||||||
|
-------------- |
||||||
|
123 |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -0,0 +1,51 @@ |
|||||||
|
-- |
||||||
|
-- Test named and nameless parameters |
||||||
|
-- |
||||||
|
CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$ |
||||||
|
return args[0] + args[1] |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$ |
||||||
|
assert a0 == args[0] |
||||||
|
assert a1 == args[1] |
||||||
|
return True |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$ |
||||||
|
assert u == args[0] |
||||||
|
return str(u) |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
-- use deliberately wrong parameter names |
||||||
|
CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$ |
||||||
|
try: |
||||||
|
assert a1 == args[0] |
||||||
|
return False |
||||||
|
except NameError, e: |
||||||
|
assert e.args[0].find("a1") > -1 |
||||||
|
return True |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
SELECT test_param_names0(2,7); |
||||||
|
test_param_names0 |
||||||
|
------------------- |
||||||
|
9 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT test_param_names1(1,'text'); |
||||||
|
test_param_names1 |
||||||
|
------------------- |
||||||
|
t |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT test_param_names2(users) from users; |
||||||
|
test_param_names2 |
||||||
|
----------------------------------------------------------------------- |
||||||
|
{'lname': 'doe', 'username': 'j_doe', 'userid': 1, 'fname': 'jane'} |
||||||
|
{'lname': 'doe', 'username': 'johnd', 'userid': 2, 'fname': 'john'} |
||||||
|
{'lname': 'doe', 'username': 'w_doe', 'userid': 3, 'fname': 'willem'} |
||||||
|
{'lname': 'smith', 'username': 'slash', 'userid': 4, 'fname': 'rick'} |
||||||
|
(4 rows) |
||||||
|
|
||||||
|
SELECT test_param_names3(1); |
||||||
|
test_param_names3 |
||||||
|
------------------- |
||||||
|
t |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -0,0 +1,300 @@ |
|||||||
|
-- |
||||||
|
-- Test returning tuples |
||||||
|
-- |
||||||
|
CREATE FUNCTION test_table_record_as(typ text, first text, second integer, retnull boolean) RETURNS table_record AS $$ |
||||||
|
if retnull: |
||||||
|
return None |
||||||
|
if typ == 'dict': |
||||||
|
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
||||||
|
elif typ == 'tuple': |
||||||
|
return ( first, second ) |
||||||
|
elif typ == 'list': |
||||||
|
return [ first, second ] |
||||||
|
elif typ == 'obj': |
||||||
|
class type_record: pass |
||||||
|
type_record.first = first |
||||||
|
type_record.second = second |
||||||
|
return type_record |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_type_record_as(typ text, first text, second integer, retnull boolean) RETURNS type_record AS $$ |
||||||
|
if retnull: |
||||||
|
return None |
||||||
|
if typ == 'dict': |
||||||
|
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
||||||
|
elif typ == 'tuple': |
||||||
|
return ( first, second ) |
||||||
|
elif typ == 'list': |
||||||
|
return [ first, second ] |
||||||
|
elif typ == 'obj': |
||||||
|
class type_record: pass |
||||||
|
type_record.first = first |
||||||
|
type_record.second = second |
||||||
|
return type_record |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_in_out_params(first in text, second out text) AS $$ |
||||||
|
return first + '_in_to_out'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
-- this doesn't work yet :-( |
||||||
|
CREATE FUNCTION test_in_out_params_multi(first in text, |
||||||
|
second out text, third out text) AS $$ |
||||||
|
return first + '_record_in_to_out'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_inout_params(first inout text) AS $$ |
||||||
|
return first + '_inout'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
-- Test tuple returning functions |
||||||
|
SELECT * FROM test_table_record_as('dict', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('dict', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('dict', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('dict', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('dict', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', null, null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', 'one', null, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
one | |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', null, 2, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| 2 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', 'three', 3, false); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
three | 3 |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', null, null, true); |
||||||
|
first | second |
||||||
|
-------+-------- |
||||||
|
| |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT * FROM test_in_out_params('test_in'); |
||||||
|
second |
||||||
|
------------------- |
||||||
|
test_in_in_to_out |
||||||
|
(1 row) |
||||||
|
|
||||||
|
-- this doesn't work yet :-( |
||||||
|
SELECT * FROM test_in_out_params_multi('test_in'); |
||||||
|
ERROR: PL/Python functions cannot return type record |
||||||
|
SELECT * FROM test_inout_params('test_in'); |
||||||
|
first |
||||||
|
--------------- |
||||||
|
test_in_inout |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -0,0 +1,102 @@ |
|||||||
|
-- |
||||||
|
-- Test returning SETOF |
||||||
|
-- |
||||||
|
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 xrange(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) |
||||||
|
|
||||||
@ -0,0 +1,113 @@ |
|||||||
|
-- nested calls |
||||||
|
-- |
||||||
|
CREATE FUNCTION nested_call_one(a text) RETURNS text |
||||||
|
AS |
||||||
|
'q = "SELECT nested_call_two(''%s'')" % a |
||||||
|
r = plpy.execute(q) |
||||||
|
return r[0]' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
CREATE FUNCTION nested_call_two(a text) RETURNS text |
||||||
|
AS |
||||||
|
'q = "SELECT nested_call_three(''%s'')" % a |
||||||
|
r = plpy.execute(q) |
||||||
|
return r[0]' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
CREATE FUNCTION nested_call_three(a text) RETURNS text |
||||||
|
AS |
||||||
|
'return a' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
-- some spi stuff |
||||||
|
CREATE FUNCTION spi_prepared_plan_test_one(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("myplan"): |
||||||
|
q = "SELECT count(*) FROM users WHERE lname = $1" |
||||||
|
SD["myplan"] = plpy.prepare(q, [ "text" ]) |
||||||
|
try: |
||||||
|
rv = plpy.execute(SD["myplan"], [a]) |
||||||
|
return "there are " + str(rv[0]["count"]) + " " + str(a) + "s" |
||||||
|
except Exception, ex: |
||||||
|
plpy.error(str(ex)) |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION spi_prepared_plan_test_nested(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("myplan"): |
||||||
|
q = "SELECT spi_prepared_plan_test_one(''%s'') as count" % a |
||||||
|
SD["myplan"] = plpy.prepare(q) |
||||||
|
try: |
||||||
|
rv = plpy.execute(SD["myplan"]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["count"] |
||||||
|
except Exception, ex: |
||||||
|
plpy.error(str(ex)) |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION join_sequences(s sequences) RETURNS text |
||||||
|
AS |
||||||
|
'if not s["multipart"]: |
||||||
|
return s["sequence"] |
||||||
|
q = "SELECT sequence FROM xsequences WHERE pid = ''%s''" % s["pid"] |
||||||
|
rv = plpy.execute(q) |
||||||
|
seq = s["sequence"] |
||||||
|
for r in rv: |
||||||
|
seq = seq + r["sequence"] |
||||||
|
return seq |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
-- spi and nested calls |
||||||
|
-- |
||||||
|
select nested_call_one('pass this along'); |
||||||
|
nested_call_one |
||||||
|
----------------------------------------------------------------- |
||||||
|
{'nested_call_two': "{'nested_call_three': 'pass this along'}"} |
||||||
|
(1 row) |
||||||
|
|
||||||
|
select spi_prepared_plan_test_one('doe'); |
||||||
|
spi_prepared_plan_test_one |
||||||
|
---------------------------- |
||||||
|
there are 3 does |
||||||
|
(1 row) |
||||||
|
|
||||||
|
select spi_prepared_plan_test_one('smith'); |
||||||
|
spi_prepared_plan_test_one |
||||||
|
---------------------------- |
||||||
|
there are 1 smiths |
||||||
|
(1 row) |
||||||
|
|
||||||
|
select spi_prepared_plan_test_nested('smith'); |
||||||
|
spi_prepared_plan_test_nested |
||||||
|
------------------------------- |
||||||
|
there are 1 smiths |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT join_sequences(sequences) FROM sequences; |
||||||
|
join_sequences |
||||||
|
---------------- |
||||||
|
ABCDEFGHIJKL |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
(6 rows) |
||||||
|
|
||||||
|
SELECT join_sequences(sequences) FROM sequences |
||||||
|
WHERE join_sequences(sequences) ~* '^A'; |
||||||
|
join_sequences |
||||||
|
---------------- |
||||||
|
ABCDEFGHIJKL |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
ABCDEF |
||||||
|
(6 rows) |
||||||
|
|
||||||
|
SELECT join_sequences(sequences) FROM sequences |
||||||
|
WHERE join_sequences(sequences) ~* '^B'; |
||||||
|
join_sequences |
||||||
|
---------------- |
||||||
|
(0 rows) |
||||||
|
|
||||||
@ -0,0 +1,153 @@ |
|||||||
|
-- these triggers are dedicated to HPHC of RI who |
||||||
|
-- decided that my kid's name was william not willem, and |
||||||
|
-- vigorously resisted all efforts at correction. they have |
||||||
|
-- since gone bankrupt... |
||||||
|
CREATE FUNCTION users_insert() returns trigger |
||||||
|
AS |
||||||
|
'if TD["new"]["fname"] == None or TD["new"]["lname"] == None: |
||||||
|
return "SKIP" |
||||||
|
if TD["new"]["username"] == None: |
||||||
|
TD["new"]["username"] = TD["new"]["fname"][:1] + "_" + TD["new"]["lname"] |
||||||
|
rv = "MODIFY" |
||||||
|
else: |
||||||
|
rv = None |
||||||
|
if TD["new"]["fname"] == "william": |
||||||
|
TD["new"]["fname"] = TD["args"][0] |
||||||
|
rv = "MODIFY" |
||||||
|
return rv' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION users_update() returns trigger |
||||||
|
AS |
||||||
|
'if TD["event"] == "UPDATE": |
||||||
|
if TD["old"]["fname"] != TD["new"]["fname"] and TD["old"]["fname"] == TD["args"][0]: |
||||||
|
return "SKIP" |
||||||
|
return None' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION users_delete() RETURNS trigger |
||||||
|
AS |
||||||
|
'if TD["old"]["fname"] == TD["args"][0]: |
||||||
|
return "SKIP" |
||||||
|
return None' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
CREATE TRIGGER users_insert_trig BEFORE INSERT ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_insert ('willem'); |
||||||
|
CREATE TRIGGER users_update_trig BEFORE UPDATE ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_update ('willem'); |
||||||
|
CREATE TRIGGER users_delete_trig BEFORE DELETE ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_delete ('willem'); |
||||||
|
-- quick peek at the table |
||||||
|
-- |
||||||
|
SELECT * FROM users; |
||||||
|
fname | lname | username | userid |
||||||
|
--------+-------+----------+-------- |
||||||
|
jane | doe | j_doe | 1 |
||||||
|
john | doe | johnd | 2 |
||||||
|
willem | doe | w_doe | 3 |
||||||
|
rick | smith | slash | 4 |
||||||
|
(4 rows) |
||||||
|
|
||||||
|
-- should fail |
||||||
|
-- |
||||||
|
UPDATE users SET fname = 'william' WHERE fname = 'willem'; |
||||||
|
-- should modify william to willem and create username |
||||||
|
-- |
||||||
|
INSERT INTO users (fname, lname) VALUES ('william', 'smith'); |
||||||
|
INSERT INTO users (fname, lname, username) VALUES ('charles', 'darwin', 'beagle'); |
||||||
|
SELECT * FROM users; |
||||||
|
fname | lname | username | userid |
||||||
|
---------+--------+----------+-------- |
||||||
|
jane | doe | j_doe | 1 |
||||||
|
john | doe | johnd | 2 |
||||||
|
willem | doe | w_doe | 3 |
||||||
|
rick | smith | slash | 4 |
||||||
|
willem | smith | w_smith | 5 |
||||||
|
charles | darwin | beagle | 6 |
||||||
|
(6 rows) |
||||||
|
|
||||||
|
-- dump trigger data |
||||||
|
CREATE TABLE trigger_test |
||||||
|
(i int, v text ); |
||||||
|
CREATE FUNCTION trigger_data() returns trigger language plpythonu as $$ |
||||||
|
|
||||||
|
if TD.has_key('relid'): |
||||||
|
TD['relid'] = "bogus:12345" |
||||||
|
|
||||||
|
skeys = TD.keys() |
||||||
|
skeys.sort() |
||||||
|
for key in skeys: |
||||||
|
val = TD[key] |
||||||
|
plpy.notice("TD[" + key + "] => " + str(val)) |
||||||
|
|
||||||
|
return None |
||||||
|
|
||||||
|
$$; |
||||||
|
CREATE TRIGGER show_trigger_data_trig |
||||||
|
BEFORE INSERT OR UPDATE OR DELETE ON trigger_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE trigger_data(23,'skidoo'); |
||||||
|
insert into trigger_test values(1,'insert'); |
||||||
|
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[event] => INSERT',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[level] => ROW',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[name] => show_trigger_data_trig',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ("TD[new] => {'i': 1, 'v': 'insert'}",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[old] => None',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[relid] => bogus:12345',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_name] => trigger_test',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_schema] => public',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[when] => BEFORE',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
update trigger_test set v = 'update' where i = 1; |
||||||
|
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[event] => UPDATE',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[level] => ROW',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[name] => show_trigger_data_trig',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ("TD[new] => {'i': 1, 'v': 'update'}",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ("TD[old] => {'i': 1, 'v': 'insert'}",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[relid] => bogus:12345',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_name] => trigger_test',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_schema] => public',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[when] => BEFORE',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
delete from trigger_test; |
||||||
|
NOTICE: ("TD[args] => ['23', 'skidoo']",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[event] => DELETE',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[level] => ROW',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[name] => show_trigger_data_trig',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[new] => None',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ("TD[old] => {'i': 1, 'v': 'update'}",) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[relid] => bogus:12345',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_name] => trigger_test',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[table_schema] => public',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
NOTICE: ('TD[when] => BEFORE',) |
||||||
|
CONTEXT: PL/Python function "trigger_data" |
||||||
|
|
||||||
|
DROP TRIGGER show_trigger_data_trig on trigger_test; |
||||||
|
|
||||||
|
DROP FUNCTION trigger_data(); |
||||||
@ -1,30 +1,28 @@ |
|||||||
-- test error handling, i forgot to restore Warn_restart in |
|
||||||
-- the trigger handler once. the errors and subsequent core dump were |
|
||||||
-- interesting. |
|
||||||
SELECT invalid_type_uncaught('rick'); |
|
||||||
WARNING: PL/Python: <class 'plpy.SPIError'>: unrecognized error in PLy_spi_prepare |
|
||||||
CONTEXT: PL/Python function "invalid_type_uncaught" |
|
||||||
ERROR: type "test" does not exist |
|
||||||
CONTEXT: PL/Python function "invalid_type_uncaught" |
|
||||||
SELECT invalid_type_caught('rick'); |
|
||||||
WARNING: PL/Python: <class 'plpy.SPIError'>: unrecognized error in PLy_spi_prepare |
|
||||||
CONTEXT: PL/Python function "invalid_type_caught" |
|
||||||
ERROR: type "test" does not exist |
|
||||||
CONTEXT: PL/Python function "invalid_type_caught" |
|
||||||
SELECT invalid_type_reraised('rick'); |
|
||||||
WARNING: PL/Python: <class 'plpy.SPIError'>: unrecognized error in PLy_spi_prepare |
|
||||||
CONTEXT: PL/Python function "invalid_type_reraised" |
|
||||||
ERROR: type "test" does not exist |
|
||||||
CONTEXT: PL/Python function "invalid_type_reraised" |
|
||||||
SELECT valid_type('rick'); |
|
||||||
valid_type |
|
||||||
------------ |
|
||||||
|
|
||||||
(1 row) |
|
||||||
|
|
||||||
-- |
-- |
||||||
-- Test Unicode error handling. |
-- Unicode handling |
||||||
-- |
-- |
||||||
|
CREATE TABLE unicode_test ( |
||||||
|
testvalue text NOT NULL |
||||||
|
); |
||||||
|
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
||||||
|
return u"\\x80" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
||||||
|
TD["new"]["testvalue"] = u"\\x80" |
||||||
|
return "MODIFY" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
||||||
|
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
||||||
|
rv = plpy.execute(plan, [u"\\x80"], 1) |
||||||
|
return rv[0]["testvalue"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
||||||
|
rv = plpy.execute(plan, u"\\x80", 1) |
||||||
|
return rv[0]["testvalue1"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
SELECT unicode_return_error(); |
SELECT unicode_return_error(); |
||||||
ERROR: PL/Python: could not create string representation of Python object, while creating return value |
ERROR: PL/Python: could not create string representation of Python object, while creating return value |
||||||
DETAIL: <type 'exceptions.UnicodeEncodeError'>: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
DETAIL: <type 'exceptions.UnicodeEncodeError'>: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
||||||
@ -0,0 +1,43 @@ |
|||||||
|
-- |
||||||
|
-- Unicode handling |
||||||
|
-- |
||||||
|
CREATE TABLE unicode_test ( |
||||||
|
testvalue text NOT NULL |
||||||
|
); |
||||||
|
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
||||||
|
return u"\\x80" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
||||||
|
TD["new"]["testvalue"] = u"\\x80" |
||||||
|
return "MODIFY" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
||||||
|
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
||||||
|
rv = plpy.execute(plan, [u"\\x80"], 1) |
||||||
|
return rv[0]["testvalue"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
||||||
|
rv = plpy.execute(plan, u"\\x80", 1) |
||||||
|
return rv[0]["testvalue1"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
SELECT unicode_return_error(); |
||||||
|
ERROR: PL/Python: could not create string representation of Python object, while creating return value |
||||||
|
DETAIL: exceptions.UnicodeError: ASCII encoding error: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_return_error" |
||||||
|
INSERT INTO unicode_test (testvalue) VALUES ('test'); |
||||||
|
ERROR: PL/Python: could not compute string representation of Python object, while modifying trigger row |
||||||
|
DETAIL: exceptions.UnicodeError: ASCII encoding error: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_trigger_error" |
||||||
|
SELECT unicode_plan_error1(); |
||||||
|
WARNING: PL/Python: plpy.Error: unrecognized error in PLy_spi_execute_plan |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error1" |
||||||
|
ERROR: PL/Python: could not execute plan |
||||||
|
DETAIL: exceptions.UnicodeError: ASCII encoding error: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error1" |
||||||
|
SELECT unicode_plan_error2(); |
||||||
|
ERROR: PL/Python: could not execute plan |
||||||
|
DETAIL: exceptions.UnicodeError: ASCII encoding error: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error2" |
||||||
@ -0,0 +1,43 @@ |
|||||||
|
-- |
||||||
|
-- Unicode handling |
||||||
|
-- |
||||||
|
CREATE TABLE unicode_test ( |
||||||
|
testvalue text NOT NULL |
||||||
|
); |
||||||
|
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
||||||
|
return u"\\x80" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
||||||
|
TD["new"]["testvalue"] = u"\\x80" |
||||||
|
return "MODIFY" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
||||||
|
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
||||||
|
rv = plpy.execute(plan, [u"\\x80"], 1) |
||||||
|
return rv[0]["testvalue"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
||||||
|
rv = plpy.execute(plan, u"\\x80", 1) |
||||||
|
return rv[0]["testvalue1"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
SELECT unicode_return_error(); |
||||||
|
ERROR: PL/Python: could not create string representation of Python object, while creating return value |
||||||
|
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_return_error" |
||||||
|
INSERT INTO unicode_test (testvalue) VALUES ('test'); |
||||||
|
ERROR: PL/Python: could not compute string representation of Python object, while modifying trigger row |
||||||
|
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_trigger_error" |
||||||
|
SELECT unicode_plan_error1(); |
||||||
|
WARNING: PL/Python: plpy.Error: unrecognized error in PLy_spi_execute_plan |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error1" |
||||||
|
ERROR: PL/Python: could not execute plan |
||||||
|
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error1" |
||||||
|
SELECT unicode_plan_error2(); |
||||||
|
ERROR: PL/Python: could not execute plan |
||||||
|
DETAIL: exceptions.UnicodeEncodeError: 'ascii' codec can't encode character u'\x80' in position 0: ordinal not in range(128) |
||||||
|
CONTEXT: PL/Python function "unicode_plan_error2" |
||||||
@ -0,0 +1,29 @@ |
|||||||
|
-- |
||||||
|
-- Tests for functions that return void |
||||||
|
-- |
||||||
|
CREATE FUNCTION test_void_func1() RETURNS void AS $$ |
||||||
|
x = 10 |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
-- illegal: can't return non-None value in void-returning func |
||||||
|
CREATE FUNCTION test_void_func2() RETURNS void AS $$ |
||||||
|
return 10 |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
CREATE FUNCTION test_return_none() RETURNS int AS $$ |
||||||
|
None |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
-- Tests for functions returning void |
||||||
|
SELECT test_void_func1(), test_void_func1() IS NULL AS "is null"; |
||||||
|
test_void_func1 | is null |
||||||
|
-----------------+--------- |
||||||
|
| f |
||||||
|
(1 row) |
||||||
|
|
||||||
|
SELECT test_void_func2(); -- should fail |
||||||
|
ERROR: PL/Python function with return type "void" did not return None |
||||||
|
CONTEXT: PL/Python function "test_void_func2" |
||||||
|
SELECT test_return_none(), test_return_none() IS NULL AS "is null"; |
||||||
|
test_return_none | is null |
||||||
|
------------------+--------- |
||||||
|
| t |
||||||
|
(1 row) |
||||||
|
|
||||||
@ -1,18 +1,109 @@ |
|||||||
|
|
||||||
-- test error handling, i forgot to restore Warn_restart in |
-- test error handling, i forgot to restore Warn_restart in |
||||||
-- the trigger handler once. the errors and subsequent core dump were |
-- the trigger handler once. the errors and subsequent core dump were |
||||||
-- interesting. |
-- interesting. |
||||||
|
|
||||||
|
/* Flat out syntax error |
||||||
|
*/ |
||||||
|
CREATE FUNCTION sql_syntax_error() RETURNS text |
||||||
|
AS |
||||||
|
'plpy.execute("syntax error")' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
SELECT sql_syntax_error(); |
||||||
|
|
||||||
|
|
||||||
|
/* check the handling of uncaught python exceptions |
||||||
|
*/ |
||||||
|
CREATE FUNCTION exception_index_invalid(text) RETURNS text |
||||||
|
AS |
||||||
|
'return args[1]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
SELECT exception_index_invalid('test'); |
||||||
|
|
||||||
|
|
||||||
|
/* check handling of nested exceptions |
||||||
|
*/ |
||||||
|
CREATE FUNCTION exception_index_invalid_nested() RETURNS text |
||||||
|
AS |
||||||
|
'rv = plpy.execute("SELECT test5(''foo'')") |
||||||
|
return rv[0]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
SELECT exception_index_invalid_nested(); |
||||||
|
|
||||||
|
|
||||||
|
/* a typo |
||||||
|
*/ |
||||||
|
CREATE FUNCTION invalid_type_uncaught(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("plan"): |
||||||
|
q = "SELECT fname FROM users WHERE lname = $1" |
||||||
|
SD["plan"] = plpy.prepare(q, [ "test" ]) |
||||||
|
rv = plpy.execute(SD["plan"], [ a ]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["fname"] |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
SELECT invalid_type_uncaught('rick'); |
SELECT invalid_type_uncaught('rick'); |
||||||
|
|
||||||
|
|
||||||
|
/* for what it's worth catch the exception generated by |
||||||
|
* the typo, and return None |
||||||
|
*/ |
||||||
|
CREATE FUNCTION invalid_type_caught(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("plan"): |
||||||
|
q = "SELECT fname FROM users WHERE lname = $1" |
||||||
|
try: |
||||||
|
SD["plan"] = plpy.prepare(q, [ "test" ]) |
||||||
|
except plpy.SPIError, ex: |
||||||
|
plpy.notice(str(ex)) |
||||||
|
return None |
||||||
|
rv = plpy.execute(SD["plan"], [ a ]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["fname"] |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
SELECT invalid_type_caught('rick'); |
SELECT invalid_type_caught('rick'); |
||||||
|
|
||||||
|
|
||||||
|
/* for what it's worth catch the exception generated by |
||||||
|
* the typo, and reraise it as a plain error |
||||||
|
*/ |
||||||
|
CREATE FUNCTION invalid_type_reraised(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("plan"): |
||||||
|
q = "SELECT fname FROM users WHERE lname = $1" |
||||||
|
try: |
||||||
|
SD["plan"] = plpy.prepare(q, [ "test" ]) |
||||||
|
except plpy.SPIError, ex: |
||||||
|
plpy.error(str(ex)) |
||||||
|
rv = plpy.execute(SD["plan"], [ a ]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["fname"] |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
SELECT invalid_type_reraised('rick'); |
SELECT invalid_type_reraised('rick'); |
||||||
SELECT valid_type('rick'); |
|
||||||
|
|
||||||
-- |
|
||||||
-- Test Unicode error handling. |
|
||||||
-- |
|
||||||
|
|
||||||
SELECT unicode_return_error(); |
/* no typo no messing about |
||||||
INSERT INTO unicode_test (testvalue) VALUES ('test'); |
*/ |
||||||
SELECT unicode_plan_error1(); |
CREATE FUNCTION valid_type(a text) RETURNS text |
||||||
SELECT unicode_plan_error2(); |
AS |
||||||
|
'if not SD.has_key("plan"): |
||||||
|
SD["plan"] = plpy.prepare("SELECT fname FROM users WHERE lname = $1", [ "text" ]) |
||||||
|
rv = plpy.execute(SD["plan"], [ a ]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["fname"] |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
SELECT valid_type('rick'); |
||||||
|
|||||||
@ -1,499 +0,0 @@ |
|||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION global_test_one() returns text |
|
||||||
AS |
|
||||||
'if not SD.has_key("global_test"): |
|
||||||
SD["global_test"] = "set by global_test_one" |
|
||||||
if not GD.has_key("global_test"): |
|
||||||
GD["global_test"] = "set by global_test_one" |
|
||||||
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION global_test_two() returns text |
|
||||||
AS |
|
||||||
'if not SD.has_key("global_test"): |
|
||||||
SD["global_test"] = "set by global_test_two" |
|
||||||
if not GD.has_key("global_test"): |
|
||||||
GD["global_test"] = "set by global_test_two" |
|
||||||
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION static_test() returns int4 |
|
||||||
AS |
|
||||||
'if SD.has_key("call"): |
|
||||||
SD["call"] = SD["call"] + 1 |
|
||||||
else: |
|
||||||
SD["call"] = 1 |
|
||||||
return SD["call"] |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- import python modules |
|
||||||
|
|
||||||
CREATE FUNCTION import_fail() returns text |
|
||||||
AS |
|
||||||
'try: |
|
||||||
import foosocket |
|
||||||
except Exception, ex: |
|
||||||
plpy.notice("import socket failed -- %s" % str(ex)) |
|
||||||
return "failed as expected" |
|
||||||
return "succeeded, that wasn''t supposed to happen"' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION import_succeed() returns text |
|
||||||
AS |
|
||||||
'try: |
|
||||||
import array |
|
||||||
import bisect |
|
||||||
import calendar |
|
||||||
import cmath |
|
||||||
import errno |
|
||||||
import math |
|
||||||
import md5 |
|
||||||
import operator |
|
||||||
import random |
|
||||||
import re |
|
||||||
import sha |
|
||||||
import string |
|
||||||
import time |
|
||||||
except Exception, ex: |
|
||||||
plpy.notice("import failed -- %s" % str(ex)) |
|
||||||
return "failed, that wasn''t supposed to happen" |
|
||||||
return "succeeded, as expected"' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION import_test_one(p text) RETURNS text |
|
||||||
AS |
|
||||||
'import sha |
|
||||||
digest = sha.new(p) |
|
||||||
return digest.hexdigest()' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION import_test_two(u users) RETURNS text |
|
||||||
AS |
|
||||||
'import sha |
|
||||||
plain = u["fname"] + u["lname"] |
|
||||||
digest = sha.new(plain); |
|
||||||
return "sha hash of " + plain + " is " + digest.hexdigest()' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION argument_test_one(u users, a1 text, a2 text) RETURNS text |
|
||||||
AS |
|
||||||
'keys = u.keys() |
|
||||||
keys.sort() |
|
||||||
out = [] |
|
||||||
for key in keys: |
|
||||||
out.append("%s: %s" % (key, u[key])) |
|
||||||
words = a1 + " " + a2 + " => {" + ", ".join(out) + "}" |
|
||||||
return words' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
-- these triggers are dedicated to HPHC of RI who |
|
||||||
-- decided that my kid's name was william not willem, and |
|
||||||
-- vigorously resisted all efforts at correction. they have |
|
||||||
-- since gone bankrupt... |
|
||||||
|
|
||||||
CREATE FUNCTION users_insert() returns trigger |
|
||||||
AS |
|
||||||
'if TD["new"]["fname"] == None or TD["new"]["lname"] == None: |
|
||||||
return "SKIP" |
|
||||||
if TD["new"]["username"] == None: |
|
||||||
TD["new"]["username"] = TD["new"]["fname"][:1] + "_" + TD["new"]["lname"] |
|
||||||
rv = "MODIFY" |
|
||||||
else: |
|
||||||
rv = None |
|
||||||
if TD["new"]["fname"] == "william": |
|
||||||
TD["new"]["fname"] = TD["args"][0] |
|
||||||
rv = "MODIFY" |
|
||||||
return rv' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION users_update() returns trigger |
|
||||||
AS |
|
||||||
'if TD["event"] == "UPDATE": |
|
||||||
if TD["old"]["fname"] != TD["new"]["fname"] and TD["old"]["fname"] == TD["args"][0]: |
|
||||||
return "SKIP" |
|
||||||
return None' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION users_delete() RETURNS trigger |
|
||||||
AS |
|
||||||
'if TD["old"]["fname"] == TD["args"][0]: |
|
||||||
return "SKIP" |
|
||||||
return None' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE TRIGGER users_insert_trig BEFORE INSERT ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_insert ('willem'); |
|
||||||
|
|
||||||
CREATE TRIGGER users_update_trig BEFORE UPDATE ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_update ('willem'); |
|
||||||
|
|
||||||
CREATE TRIGGER users_delete_trig BEFORE DELETE ON users FOR EACH ROW |
|
||||||
EXECUTE PROCEDURE users_delete ('willem'); |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
-- dump trigger data |
|
||||||
|
|
||||||
CREATE TABLE trigger_test |
|
||||||
(i int, v text ); |
|
||||||
|
|
||||||
CREATE FUNCTION trigger_data() returns trigger language plpythonu as $$ |
|
||||||
|
|
||||||
if TD.has_key('relid'): |
|
||||||
TD['relid'] = "bogus:12345" |
|
||||||
|
|
||||||
skeys = TD.keys() |
|
||||||
skeys.sort() |
|
||||||
for key in skeys: |
|
||||||
val = TD[key] |
|
||||||
plpy.notice("TD[" + key + "] => " + str(val)) |
|
||||||
|
|
||||||
return None |
|
||||||
|
|
||||||
$$; |
|
||||||
|
|
||||||
CREATE TRIGGER show_trigger_data_trig |
|
||||||
BEFORE INSERT OR UPDATE OR DELETE ON trigger_test |
|
||||||
FOR EACH ROW EXECUTE PROCEDURE trigger_data(23,'skidoo'); |
|
||||||
|
|
||||||
insert into trigger_test values(1,'insert'); |
|
||||||
update trigger_test set v = 'update' where i = 1; |
|
||||||
delete from trigger_test; |
|
||||||
|
|
||||||
DROP TRIGGER show_trigger_data_trig on trigger_test; |
|
||||||
|
|
||||||
DROP FUNCTION trigger_data(); |
|
||||||
|
|
||||||
-- nested calls |
|
||||||
-- |
|
||||||
|
|
||||||
CREATE FUNCTION nested_call_one(a text) RETURNS text |
|
||||||
AS |
|
||||||
'q = "SELECT nested_call_two(''%s'')" % a |
|
||||||
r = plpy.execute(q) |
|
||||||
return r[0]' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
|
|
||||||
CREATE FUNCTION nested_call_two(a text) RETURNS text |
|
||||||
AS |
|
||||||
'q = "SELECT nested_call_three(''%s'')" % a |
|
||||||
r = plpy.execute(q) |
|
||||||
return r[0]' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
|
|
||||||
CREATE FUNCTION nested_call_three(a text) RETURNS text |
|
||||||
AS |
|
||||||
'return a' |
|
||||||
LANGUAGE plpythonu ; |
|
||||||
|
|
||||||
-- some spi stuff |
|
||||||
|
|
||||||
CREATE FUNCTION spi_prepared_plan_test_one(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("myplan"): |
|
||||||
q = "SELECT count(*) FROM users WHERE lname = $1" |
|
||||||
SD["myplan"] = plpy.prepare(q, [ "text" ]) |
|
||||||
try: |
|
||||||
rv = plpy.execute(SD["myplan"], [a]) |
|
||||||
return "there are " + str(rv[0]["count"]) + " " + str(a) + "s" |
|
||||||
except Exception, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION spi_prepared_plan_test_nested(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("myplan"): |
|
||||||
q = "SELECT spi_prepared_plan_test_one(''%s'') as count" % a |
|
||||||
SD["myplan"] = plpy.prepare(q) |
|
||||||
try: |
|
||||||
rv = plpy.execute(SD["myplan"]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["count"] |
|
||||||
except Exception, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
/* really stupid function just to get the module loaded |
|
||||||
*/ |
|
||||||
CREATE FUNCTION stupid() RETURNS text AS 'return "zarkon"' LANGUAGE plpythonu; |
|
||||||
|
|
||||||
/* a typo |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_uncaught(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
/* for what it's worth catch the exception generated by |
|
||||||
* the typo, and return None |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_caught(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
try: |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
except plpy.SPIError, ex: |
|
||||||
plpy.notice(str(ex)) |
|
||||||
return None |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
/* for what it's worth catch the exception generated by |
|
||||||
* the typo, and reraise it as a plain error |
|
||||||
*/ |
|
||||||
CREATE FUNCTION invalid_type_reraised(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
q = "SELECT fname FROM users WHERE lname = $1" |
|
||||||
try: |
|
||||||
SD["plan"] = plpy.prepare(q, [ "test" ]) |
|
||||||
except plpy.SPIError, ex: |
|
||||||
plpy.error(str(ex)) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
/* no typo no messing about |
|
||||||
*/ |
|
||||||
CREATE FUNCTION valid_type(a text) RETURNS text |
|
||||||
AS |
|
||||||
'if not SD.has_key("plan"): |
|
||||||
SD["plan"] = plpy.prepare("SELECT fname FROM users WHERE lname = $1", [ "text" ]) |
|
||||||
rv = plpy.execute(SD["plan"], [ a ]) |
|
||||||
if len(rv): |
|
||||||
return rv[0]["fname"] |
|
||||||
return None |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
/* Flat out syntax error |
|
||||||
*/ |
|
||||||
CREATE FUNCTION sql_syntax_error() RETURNS text |
|
||||||
AS |
|
||||||
'plpy.execute("syntax error")' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
/* check the handling of uncaught python exceptions |
|
||||||
*/ |
|
||||||
CREATE FUNCTION exception_index_invalid(text) RETURNS text |
|
||||||
AS |
|
||||||
'return args[1]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
/* check handling of nested exceptions |
|
||||||
*/ |
|
||||||
CREATE FUNCTION exception_index_invalid_nested() RETURNS text |
|
||||||
AS |
|
||||||
'rv = plpy.execute("SELECT test5(''foo'')") |
|
||||||
return rv[0]' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
CREATE FUNCTION join_sequences(s sequences) RETURNS text |
|
||||||
AS |
|
||||||
'if not s["multipart"]: |
|
||||||
return s["sequence"] |
|
||||||
q = "SELECT sequence FROM xsequences WHERE pid = ''%s''" % s["pid"] |
|
||||||
rv = plpy.execute(q) |
|
||||||
seq = s["sequence"] |
|
||||||
for r in rv: |
|
||||||
seq = seq + r["sequence"] |
|
||||||
return seq |
|
||||||
' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- |
|
||||||
-- Universal Newline Support |
|
||||||
-- |
|
||||||
|
|
||||||
CREATE OR REPLACE FUNCTION newline_lf() RETURNS integer AS |
|
||||||
E'x = 100\ny = 23\nreturn x + y\n' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE OR REPLACE FUNCTION newline_cr() RETURNS integer AS |
|
||||||
E'x = 100\ry = 23\rreturn x + y\r' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE OR REPLACE FUNCTION newline_crlf() RETURNS integer AS |
|
||||||
E'x = 100\r\ny = 23\r\nreturn x + y\r\n' |
|
||||||
LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- |
|
||||||
-- Unicode error handling |
|
||||||
-- |
|
||||||
|
|
||||||
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
|
||||||
return u"\\x80" |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
|
||||||
TD["new"]["testvalue"] = u"\\x80" |
|
||||||
return "MODIFY" |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
|
||||||
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
|
||||||
|
|
||||||
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
|
||||||
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
|
||||||
rv = plpy.execute(plan, [u"\\x80"], 1) |
|
||||||
return rv[0]["testvalue"] |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
|
||||||
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
|
||||||
rv = plpy.execute(plan, u"\\x80", 1) |
|
||||||
return rv[0]["testvalue1"] |
|
||||||
' LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- Tests for functions that return void |
|
||||||
|
|
||||||
CREATE FUNCTION test_void_func1() RETURNS void AS $$ |
|
||||||
x = 10 |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- illegal: can't return non-None value in void-returning func |
|
||||||
CREATE FUNCTION test_void_func2() RETURNS void AS $$ |
|
||||||
return 10 |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_return_none() RETURNS int AS $$ |
|
||||||
None |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
-- |
|
||||||
-- Test named and nameless parameters |
|
||||||
-- |
|
||||||
CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$ |
|
||||||
return args[0] + args[1] |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$ |
|
||||||
assert a0 == args[0] |
|
||||||
assert a1 == args[1] |
|
||||||
return True |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$ |
|
||||||
assert u == args[0] |
|
||||||
return str(u) |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- use deliberately wrong parameter names |
|
||||||
CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$ |
|
||||||
try: |
|
||||||
assert a1 == args[0] |
|
||||||
return False |
|
||||||
except NameError, e: |
|
||||||
assert e.args[0].find("a1") > -1 |
|
||||||
return True |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
|
|
||||||
-- |
|
||||||
-- Test returning SETOF |
|
||||||
-- |
|
||||||
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 xrange(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 returning tuples |
|
||||||
-- |
|
||||||
CREATE FUNCTION test_table_record_as(typ text, first text, second integer, retnull boolean) RETURNS table_record AS $$ |
|
||||||
if retnull: |
|
||||||
return None |
|
||||||
if typ == 'dict': |
|
||||||
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
|
||||||
elif typ == 'tuple': |
|
||||||
return ( first, second ) |
|
||||||
elif typ == 'list': |
|
||||||
return [ first, second ] |
|
||||||
elif typ == 'obj': |
|
||||||
class type_record: pass |
|
||||||
type_record.first = first |
|
||||||
type_record.second = second |
|
||||||
return type_record |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_type_record_as(typ text, first text, second integer, retnull boolean) RETURNS type_record AS $$ |
|
||||||
if retnull: |
|
||||||
return None |
|
||||||
if typ == 'dict': |
|
||||||
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
|
||||||
elif typ == 'tuple': |
|
||||||
return ( first, second ) |
|
||||||
elif typ == 'list': |
|
||||||
return [ first, second ] |
|
||||||
elif typ == 'obj': |
|
||||||
class type_record: pass |
|
||||||
type_record.first = first |
|
||||||
type_record.second = second |
|
||||||
return type_record |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_in_out_params(first in text, second out text) AS $$ |
|
||||||
return first + '_in_to_out'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
-- this doesn't work yet :-( |
|
||||||
CREATE FUNCTION test_in_out_params_multi(first in text, |
|
||||||
second out text, third out text) AS $$ |
|
||||||
return first + '_record_in_to_out'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
|
|
||||||
CREATE FUNCTION test_inout_params(first inout text) AS $$ |
|
||||||
return first + '_inout'; |
|
||||||
$$ LANGUAGE plpythonu; |
|
||||||
@ -0,0 +1,38 @@ |
|||||||
|
-- |
||||||
|
-- check static and global data (SD and GD) |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE FUNCTION global_test_one() returns text |
||||||
|
AS |
||||||
|
'if not SD.has_key("global_test"): |
||||||
|
SD["global_test"] = "set by global_test_one" |
||||||
|
if not GD.has_key("global_test"): |
||||||
|
GD["global_test"] = "set by global_test_one" |
||||||
|
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION global_test_two() returns text |
||||||
|
AS |
||||||
|
'if not SD.has_key("global_test"): |
||||||
|
SD["global_test"] = "set by global_test_two" |
||||||
|
if not GD.has_key("global_test"): |
||||||
|
GD["global_test"] = "set by global_test_two" |
||||||
|
return "SD: " + SD["global_test"] + ", GD: " + GD["global_test"]' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
CREATE FUNCTION static_test() returns int4 |
||||||
|
AS |
||||||
|
'if SD.has_key("call"): |
||||||
|
SD["call"] = SD["call"] + 1 |
||||||
|
else: |
||||||
|
SD["call"] = 1 |
||||||
|
return SD["call"] |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
SELECT static_test(); |
||||||
|
SELECT static_test(); |
||||||
|
SELECT global_test_one(); |
||||||
|
SELECT global_test_two(); |
||||||
@ -0,0 +1,63 @@ |
|||||||
|
-- import python modules |
||||||
|
|
||||||
|
CREATE FUNCTION import_fail() returns text |
||||||
|
AS |
||||||
|
'try: |
||||||
|
import foosocket |
||||||
|
except Exception, ex: |
||||||
|
plpy.notice("import socket failed -- %s" % str(ex)) |
||||||
|
return "failed as expected" |
||||||
|
return "succeeded, that wasn''t supposed to happen"' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
CREATE FUNCTION import_succeed() returns text |
||||||
|
AS |
||||||
|
'try: |
||||||
|
import array |
||||||
|
import bisect |
||||||
|
import calendar |
||||||
|
import cmath |
||||||
|
import errno |
||||||
|
import math |
||||||
|
import md5 |
||||||
|
import operator |
||||||
|
import random |
||||||
|
import re |
||||||
|
import sha |
||||||
|
import string |
||||||
|
import time |
||||||
|
except Exception, ex: |
||||||
|
plpy.notice("import failed -- %s" % str(ex)) |
||||||
|
return "failed, that wasn''t supposed to happen" |
||||||
|
return "succeeded, as expected"' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION import_test_one(p text) RETURNS text |
||||||
|
AS |
||||||
|
'import sha |
||||||
|
digest = sha.new(p) |
||||||
|
return digest.hexdigest()' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION import_test_two(u users) RETURNS text |
||||||
|
AS |
||||||
|
'import sha |
||||||
|
plain = u["fname"] + u["lname"] |
||||||
|
digest = sha.new(plain); |
||||||
|
return "sha hash of " + plain + " is " + digest.hexdigest()' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
-- import python modules |
||||||
|
-- |
||||||
|
SELECT import_fail(); |
||||||
|
SELECT import_succeed(); |
||||||
|
|
||||||
|
-- test import and simple argument handling |
||||||
|
-- |
||||||
|
SELECT import_test_one('sha hash of this string'); |
||||||
|
|
||||||
|
-- test import and tuple argument handling |
||||||
|
-- |
||||||
|
select import_test_two(users) from users where fname = 'willem'; |
||||||
@ -0,0 +1,20 @@ |
|||||||
|
-- |
||||||
|
-- Universal Newline Support |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE OR REPLACE FUNCTION newline_lf() RETURNS integer AS |
||||||
|
E'x = 100\ny = 23\nreturn x + y\n' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE OR REPLACE FUNCTION newline_cr() RETURNS integer AS |
||||||
|
E'x = 100\ry = 23\rreturn x + y\r' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE OR REPLACE FUNCTION newline_crlf() RETURNS integer AS |
||||||
|
E'x = 100\r\ny = 23\r\nreturn x + y\r\n' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
SELECT newline_lf(); |
||||||
|
SELECT newline_cr(); |
||||||
|
SELECT newline_crlf(); |
||||||
@ -0,0 +1,34 @@ |
|||||||
|
-- |
||||||
|
-- Test named and nameless parameters |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE FUNCTION test_param_names0(integer, integer) RETURNS int AS $$ |
||||||
|
return args[0] + args[1] |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_param_names1(a0 integer, a1 text) RETURNS boolean AS $$ |
||||||
|
assert a0 == args[0] |
||||||
|
assert a1 == args[1] |
||||||
|
return True |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_param_names2(u users) RETURNS text AS $$ |
||||||
|
assert u == args[0] |
||||||
|
return str(u) |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
-- use deliberately wrong parameter names |
||||||
|
CREATE FUNCTION test_param_names3(a0 integer) RETURNS boolean AS $$ |
||||||
|
try: |
||||||
|
assert a1 == args[0] |
||||||
|
return False |
||||||
|
except NameError, e: |
||||||
|
assert e.args[0].find("a1") > -1 |
||||||
|
return True |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
SELECT test_param_names0(2,7); |
||||||
|
SELECT test_param_names1(1,'text'); |
||||||
|
SELECT test_param_names2(users) from users; |
||||||
|
SELECT test_param_names3(1); |
||||||
@ -0,0 +1,104 @@ |
|||||||
|
-- |
||||||
|
-- Test returning tuples |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE FUNCTION test_table_record_as(typ text, first text, second integer, retnull boolean) RETURNS table_record AS $$ |
||||||
|
if retnull: |
||||||
|
return None |
||||||
|
if typ == 'dict': |
||||||
|
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
||||||
|
elif typ == 'tuple': |
||||||
|
return ( first, second ) |
||||||
|
elif typ == 'list': |
||||||
|
return [ first, second ] |
||||||
|
elif typ == 'obj': |
||||||
|
class type_record: pass |
||||||
|
type_record.first = first |
||||||
|
type_record.second = second |
||||||
|
return type_record |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_type_record_as(typ text, first text, second integer, retnull boolean) RETURNS type_record AS $$ |
||||||
|
if retnull: |
||||||
|
return None |
||||||
|
if typ == 'dict': |
||||||
|
return { 'first': first, 'second': second, 'additionalfield': 'must not cause trouble' } |
||||||
|
elif typ == 'tuple': |
||||||
|
return ( first, second ) |
||||||
|
elif typ == 'list': |
||||||
|
return [ first, second ] |
||||||
|
elif typ == 'obj': |
||||||
|
class type_record: pass |
||||||
|
type_record.first = first |
||||||
|
type_record.second = second |
||||||
|
return type_record |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_in_out_params(first in text, second out text) AS $$ |
||||||
|
return first + '_in_to_out'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
-- this doesn't work yet :-( |
||||||
|
CREATE FUNCTION test_in_out_params_multi(first in text, |
||||||
|
second out text, third out text) AS $$ |
||||||
|
return first + '_record_in_to_out'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_inout_params(first inout text) AS $$ |
||||||
|
return first + '_inout'; |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
-- Test tuple returning functions |
||||||
|
SELECT * FROM test_table_record_as('dict', null, null, false); |
||||||
|
SELECT * FROM test_table_record_as('dict', 'one', null, false); |
||||||
|
SELECT * FROM test_table_record_as('dict', null, 2, false); |
||||||
|
SELECT * FROM test_table_record_as('dict', 'three', 3, false); |
||||||
|
SELECT * FROM test_table_record_as('dict', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('tuple', null, null, false); |
||||||
|
SELECT * FROM test_table_record_as('tuple', 'one', null, false); |
||||||
|
SELECT * FROM test_table_record_as('tuple', null, 2, false); |
||||||
|
SELECT * FROM test_table_record_as('tuple', 'three', 3, false); |
||||||
|
SELECT * FROM test_table_record_as('tuple', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('list', null, null, false); |
||||||
|
SELECT * FROM test_table_record_as('list', 'one', null, false); |
||||||
|
SELECT * FROM test_table_record_as('list', null, 2, false); |
||||||
|
SELECT * FROM test_table_record_as('list', 'three', 3, false); |
||||||
|
SELECT * FROM test_table_record_as('list', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_table_record_as('obj', null, null, false); |
||||||
|
SELECT * FROM test_table_record_as('obj', 'one', null, false); |
||||||
|
SELECT * FROM test_table_record_as('obj', null, 2, false); |
||||||
|
SELECT * FROM test_table_record_as('obj', 'three', 3, false); |
||||||
|
SELECT * FROM test_table_record_as('obj', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('dict', null, null, false); |
||||||
|
SELECT * FROM test_type_record_as('dict', 'one', null, false); |
||||||
|
SELECT * FROM test_type_record_as('dict', null, 2, false); |
||||||
|
SELECT * FROM test_type_record_as('dict', 'three', 3, false); |
||||||
|
SELECT * FROM test_type_record_as('dict', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('tuple', null, null, false); |
||||||
|
SELECT * FROM test_type_record_as('tuple', 'one', null, false); |
||||||
|
SELECT * FROM test_type_record_as('tuple', null, 2, false); |
||||||
|
SELECT * FROM test_type_record_as('tuple', 'three', 3, false); |
||||||
|
SELECT * FROM test_type_record_as('tuple', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('list', null, null, false); |
||||||
|
SELECT * FROM test_type_record_as('list', 'one', null, false); |
||||||
|
SELECT * FROM test_type_record_as('list', null, 2, false); |
||||||
|
SELECT * FROM test_type_record_as('list', 'three', 3, false); |
||||||
|
SELECT * FROM test_type_record_as('list', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_type_record_as('obj', null, null, false); |
||||||
|
SELECT * FROM test_type_record_as('obj', 'one', null, false); |
||||||
|
SELECT * FROM test_type_record_as('obj', null, 2, false); |
||||||
|
SELECT * FROM test_type_record_as('obj', 'three', 3, false); |
||||||
|
SELECT * FROM test_type_record_as('obj', null, null, true); |
||||||
|
|
||||||
|
SELECT * FROM test_in_out_params('test_in'); |
||||||
|
-- this doesn't work yet :-( |
||||||
|
SELECT * FROM test_in_out_params_multi('test_in'); |
||||||
|
SELECT * FROM test_inout_params('test_in'); |
||||||
@ -0,0 +1,46 @@ |
|||||||
|
-- |
||||||
|
-- Test returning SETOF |
||||||
|
-- |
||||||
|
|
||||||
|
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 xrange(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'); |
||||||
|
SELECT test_setof_as_list(1, 'list'); |
||||||
|
SELECT test_setof_as_list(2, 'list'); |
||||||
|
SELECT test_setof_as_list(2, null); |
||||||
|
|
||||||
|
SELECT test_setof_as_tuple(0, 'tuple'); |
||||||
|
SELECT test_setof_as_tuple(1, 'tuple'); |
||||||
|
SELECT test_setof_as_tuple(2, 'tuple'); |
||||||
|
SELECT test_setof_as_tuple(2, null); |
||||||
|
|
||||||
|
SELECT test_setof_as_iterator(0, 'list'); |
||||||
|
SELECT test_setof_as_iterator(1, 'list'); |
||||||
|
SELECT test_setof_as_iterator(2, 'list'); |
||||||
|
SELECT test_setof_as_iterator(2, null); |
||||||
@ -0,0 +1,90 @@ |
|||||||
|
|
||||||
|
|
||||||
|
-- nested calls |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE FUNCTION nested_call_one(a text) RETURNS text |
||||||
|
AS |
||||||
|
'q = "SELECT nested_call_two(''%s'')" % a |
||||||
|
r = plpy.execute(q) |
||||||
|
return r[0]' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
|
||||||
|
CREATE FUNCTION nested_call_two(a text) RETURNS text |
||||||
|
AS |
||||||
|
'q = "SELECT nested_call_three(''%s'')" % a |
||||||
|
r = plpy.execute(q) |
||||||
|
return r[0]' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
|
||||||
|
CREATE FUNCTION nested_call_three(a text) RETURNS text |
||||||
|
AS |
||||||
|
'return a' |
||||||
|
LANGUAGE plpythonu ; |
||||||
|
|
||||||
|
-- some spi stuff |
||||||
|
|
||||||
|
CREATE FUNCTION spi_prepared_plan_test_one(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("myplan"): |
||||||
|
q = "SELECT count(*) FROM users WHERE lname = $1" |
||||||
|
SD["myplan"] = plpy.prepare(q, [ "text" ]) |
||||||
|
try: |
||||||
|
rv = plpy.execute(SD["myplan"], [a]) |
||||||
|
return "there are " + str(rv[0]["count"]) + " " + str(a) + "s" |
||||||
|
except Exception, ex: |
||||||
|
plpy.error(str(ex)) |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION spi_prepared_plan_test_nested(a text) RETURNS text |
||||||
|
AS |
||||||
|
'if not SD.has_key("myplan"): |
||||||
|
q = "SELECT spi_prepared_plan_test_one(''%s'') as count" % a |
||||||
|
SD["myplan"] = plpy.prepare(q) |
||||||
|
try: |
||||||
|
rv = plpy.execute(SD["myplan"]) |
||||||
|
if len(rv): |
||||||
|
return rv[0]["count"] |
||||||
|
except Exception, ex: |
||||||
|
plpy.error(str(ex)) |
||||||
|
return None |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
CREATE FUNCTION join_sequences(s sequences) RETURNS text |
||||||
|
AS |
||||||
|
'if not s["multipart"]: |
||||||
|
return s["sequence"] |
||||||
|
q = "SELECT sequence FROM xsequences WHERE pid = ''%s''" % s["pid"] |
||||||
|
rv = plpy.execute(q) |
||||||
|
seq = s["sequence"] |
||||||
|
for r in rv: |
||||||
|
seq = seq + r["sequence"] |
||||||
|
return seq |
||||||
|
' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
-- spi and nested calls |
||||||
|
-- |
||||||
|
select nested_call_one('pass this along'); |
||||||
|
select spi_prepared_plan_test_one('doe'); |
||||||
|
select spi_prepared_plan_test_one('smith'); |
||||||
|
select spi_prepared_plan_test_nested('smith'); |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
SELECT join_sequences(sequences) FROM sequences; |
||||||
|
SELECT join_sequences(sequences) FROM sequences |
||||||
|
WHERE join_sequences(sequences) ~* '^A'; |
||||||
|
SELECT join_sequences(sequences) FROM sequences |
||||||
|
WHERE join_sequences(sequences) ~* '^B'; |
||||||
@ -1,151 +1,21 @@ |
|||||||
-- first some tests of basic functionality |
-- first some tests of basic functionality |
||||||
-- |
|
||||||
-- better succeed |
|
||||||
-- |
|
||||||
select stupid(); |
|
||||||
|
|
||||||
-- check static and global data |
|
||||||
-- |
|
||||||
SELECT static_test(); |
|
||||||
SELECT static_test(); |
|
||||||
SELECT global_test_one(); |
|
||||||
SELECT global_test_two(); |
|
||||||
|
|
||||||
-- import python modules |
-- really stupid function just to get the module loaded |
||||||
-- |
CREATE FUNCTION stupid() RETURNS text AS 'return "zarkon"' LANGUAGE plpythonu; |
||||||
SELECT import_fail(); |
|
||||||
SELECT import_succeed(); |
|
||||||
|
|
||||||
-- test import and simple argument handling |
select stupid(); |
||||||
-- |
|
||||||
SELECT import_test_one('sha hash of this string'); |
|
||||||
|
|
||||||
-- test import and tuple argument handling |
|
||||||
-- |
|
||||||
select import_test_two(users) from users where fname = 'willem'; |
|
||||||
|
|
||||||
-- test multiple arguments |
-- test multiple arguments |
||||||
-- |
CREATE FUNCTION argument_test_one(u users, a1 text, a2 text) RETURNS text |
||||||
select argument_test_one(users, fname, lname) from users where lname = 'doe' order by 1; |
AS |
||||||
|
'keys = u.keys() |
||||||
|
keys.sort() |
||||||
-- spi and nested calls |
out = [] |
||||||
-- |
for key in keys: |
||||||
select nested_call_one('pass this along'); |
out.append("%s: %s" % (key, u[key])) |
||||||
select spi_prepared_plan_test_one('doe'); |
words = a1 + " " + a2 + " => {" + ", ".join(out) + "}" |
||||||
select spi_prepared_plan_test_one('smith'); |
return words' |
||||||
select spi_prepared_plan_test_nested('smith'); |
LANGUAGE plpythonu; |
||||||
|
|
||||||
-- quick peek at the table |
|
||||||
-- |
|
||||||
SELECT * FROM users; |
|
||||||
|
|
||||||
-- should fail |
|
||||||
-- |
|
||||||
UPDATE users SET fname = 'william' WHERE fname = 'willem'; |
|
||||||
|
|
||||||
-- should modify william to willem and create username |
|
||||||
-- |
|
||||||
INSERT INTO users (fname, lname) VALUES ('william', 'smith'); |
|
||||||
INSERT INTO users (fname, lname, username) VALUES ('charles', 'darwin', 'beagle'); |
|
||||||
|
|
||||||
SELECT * FROM users; |
|
||||||
|
|
||||||
|
|
||||||
SELECT join_sequences(sequences) FROM sequences; |
|
||||||
SELECT join_sequences(sequences) FROM sequences |
|
||||||
WHERE join_sequences(sequences) ~* '^A'; |
|
||||||
SELECT join_sequences(sequences) FROM sequences |
|
||||||
WHERE join_sequences(sequences) ~* '^B'; |
|
||||||
|
|
||||||
-- error in trigger |
|
||||||
-- |
|
||||||
|
|
||||||
-- |
|
||||||
-- Check Universal Newline Support |
|
||||||
-- |
|
||||||
|
|
||||||
SELECT newline_lf(); |
|
||||||
SELECT newline_cr(); |
|
||||||
SELECT newline_crlf(); |
|
||||||
|
|
||||||
-- Tests for functions returning void |
select argument_test_one(users, fname, lname) from users where lname = 'doe' order by 1; |
||||||
SELECT test_void_func1(), test_void_func1() IS NULL AS "is null"; |
|
||||||
SELECT test_void_func2(); -- should fail |
|
||||||
SELECT test_return_none(), test_return_none() IS NULL AS "is null"; |
|
||||||
|
|
||||||
-- Test for functions with named and nameless parameters |
|
||||||
SELECT test_param_names0(2,7); |
|
||||||
SELECT test_param_names1(1,'text'); |
|
||||||
SELECT test_param_names2(users) from users; |
|
||||||
SELECT test_param_names3(1); |
|
||||||
|
|
||||||
-- Test set returning functions |
|
||||||
SELECT test_setof_as_list(0, 'list'); |
|
||||||
SELECT test_setof_as_list(1, 'list'); |
|
||||||
SELECT test_setof_as_list(2, 'list'); |
|
||||||
SELECT test_setof_as_list(2, null); |
|
||||||
|
|
||||||
SELECT test_setof_as_tuple(0, 'tuple'); |
|
||||||
SELECT test_setof_as_tuple(1, 'tuple'); |
|
||||||
SELECT test_setof_as_tuple(2, 'tuple'); |
|
||||||
SELECT test_setof_as_tuple(2, null); |
|
||||||
|
|
||||||
SELECT test_setof_as_iterator(0, 'list'); |
|
||||||
SELECT test_setof_as_iterator(1, 'list'); |
|
||||||
SELECT test_setof_as_iterator(2, 'list'); |
|
||||||
SELECT test_setof_as_iterator(2, null); |
|
||||||
|
|
||||||
-- Test tuple returning functions |
|
||||||
SELECT * FROM test_table_record_as('dict', null, null, false); |
|
||||||
SELECT * FROM test_table_record_as('dict', 'one', null, false); |
|
||||||
SELECT * FROM test_table_record_as('dict', null, 2, false); |
|
||||||
SELECT * FROM test_table_record_as('dict', 'three', 3, false); |
|
||||||
SELECT * FROM test_table_record_as('dict', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_table_record_as('tuple', null, null, false); |
|
||||||
SELECT * FROM test_table_record_as('tuple', 'one', null, false); |
|
||||||
SELECT * FROM test_table_record_as('tuple', null, 2, false); |
|
||||||
SELECT * FROM test_table_record_as('tuple', 'three', 3, false); |
|
||||||
SELECT * FROM test_table_record_as('tuple', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_table_record_as('list', null, null, false); |
|
||||||
SELECT * FROM test_table_record_as('list', 'one', null, false); |
|
||||||
SELECT * FROM test_table_record_as('list', null, 2, false); |
|
||||||
SELECT * FROM test_table_record_as('list', 'three', 3, false); |
|
||||||
SELECT * FROM test_table_record_as('list', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_table_record_as('obj', null, null, false); |
|
||||||
SELECT * FROM test_table_record_as('obj', 'one', null, false); |
|
||||||
SELECT * FROM test_table_record_as('obj', null, 2, false); |
|
||||||
SELECT * FROM test_table_record_as('obj', 'three', 3, false); |
|
||||||
SELECT * FROM test_table_record_as('obj', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_type_record_as('dict', null, null, false); |
|
||||||
SELECT * FROM test_type_record_as('dict', 'one', null, false); |
|
||||||
SELECT * FROM test_type_record_as('dict', null, 2, false); |
|
||||||
SELECT * FROM test_type_record_as('dict', 'three', 3, false); |
|
||||||
SELECT * FROM test_type_record_as('dict', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_type_record_as('tuple', null, null, false); |
|
||||||
SELECT * FROM test_type_record_as('tuple', 'one', null, false); |
|
||||||
SELECT * FROM test_type_record_as('tuple', null, 2, false); |
|
||||||
SELECT * FROM test_type_record_as('tuple', 'three', 3, false); |
|
||||||
SELECT * FROM test_type_record_as('tuple', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_type_record_as('list', null, null, false); |
|
||||||
SELECT * FROM test_type_record_as('list', 'one', null, false); |
|
||||||
SELECT * FROM test_type_record_as('list', null, 2, false); |
|
||||||
SELECT * FROM test_type_record_as('list', 'three', 3, false); |
|
||||||
SELECT * FROM test_type_record_as('list', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_type_record_as('obj', null, null, false); |
|
||||||
SELECT * FROM test_type_record_as('obj', 'one', null, false); |
|
||||||
SELECT * FROM test_type_record_as('obj', null, 2, false); |
|
||||||
SELECT * FROM test_type_record_as('obj', 'three', 3, false); |
|
||||||
SELECT * FROM test_type_record_as('obj', null, null, true); |
|
||||||
|
|
||||||
SELECT * FROM test_in_out_params('test_in'); |
|
||||||
-- this doesn't work yet :-( |
|
||||||
SELECT * FROM test_in_out_params_multi('test_in'); |
|
||||||
SELECT * FROM test_inout_params('test_in'); |
|
||||||
|
|||||||
@ -0,0 +1,95 @@ |
|||||||
|
-- these triggers are dedicated to HPHC of RI who |
||||||
|
-- decided that my kid's name was william not willem, and |
||||||
|
-- vigorously resisted all efforts at correction. they have |
||||||
|
-- since gone bankrupt... |
||||||
|
|
||||||
|
CREATE FUNCTION users_insert() returns trigger |
||||||
|
AS |
||||||
|
'if TD["new"]["fname"] == None or TD["new"]["lname"] == None: |
||||||
|
return "SKIP" |
||||||
|
if TD["new"]["username"] == None: |
||||||
|
TD["new"]["username"] = TD["new"]["fname"][:1] + "_" + TD["new"]["lname"] |
||||||
|
rv = "MODIFY" |
||||||
|
else: |
||||||
|
rv = None |
||||||
|
if TD["new"]["fname"] == "william": |
||||||
|
TD["new"]["fname"] = TD["args"][0] |
||||||
|
rv = "MODIFY" |
||||||
|
return rv' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
CREATE FUNCTION users_update() returns trigger |
||||||
|
AS |
||||||
|
'if TD["event"] == "UPDATE": |
||||||
|
if TD["old"]["fname"] != TD["new"]["fname"] and TD["old"]["fname"] == TD["args"][0]: |
||||||
|
return "SKIP" |
||||||
|
return None' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
CREATE FUNCTION users_delete() RETURNS trigger |
||||||
|
AS |
||||||
|
'if TD["old"]["fname"] == TD["args"][0]: |
||||||
|
return "SKIP" |
||||||
|
return None' |
||||||
|
LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
CREATE TRIGGER users_insert_trig BEFORE INSERT ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_insert ('willem'); |
||||||
|
|
||||||
|
CREATE TRIGGER users_update_trig BEFORE UPDATE ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_update ('willem'); |
||||||
|
|
||||||
|
CREATE TRIGGER users_delete_trig BEFORE DELETE ON users FOR EACH ROW |
||||||
|
EXECUTE PROCEDURE users_delete ('willem'); |
||||||
|
|
||||||
|
|
||||||
|
-- quick peek at the table |
||||||
|
-- |
||||||
|
SELECT * FROM users; |
||||||
|
|
||||||
|
-- should fail |
||||||
|
-- |
||||||
|
UPDATE users SET fname = 'william' WHERE fname = 'willem'; |
||||||
|
|
||||||
|
-- should modify william to willem and create username |
||||||
|
-- |
||||||
|
INSERT INTO users (fname, lname) VALUES ('william', 'smith'); |
||||||
|
INSERT INTO users (fname, lname, username) VALUES ('charles', 'darwin', 'beagle'); |
||||||
|
|
||||||
|
SELECT * FROM users; |
||||||
|
|
||||||
|
|
||||||
|
-- dump trigger data |
||||||
|
|
||||||
|
CREATE TABLE trigger_test |
||||||
|
(i int, v text ); |
||||||
|
|
||||||
|
CREATE FUNCTION trigger_data() returns trigger language plpythonu as $$ |
||||||
|
|
||||||
|
if TD.has_key('relid'): |
||||||
|
TD['relid'] = "bogus:12345" |
||||||
|
|
||||||
|
skeys = TD.keys() |
||||||
|
skeys.sort() |
||||||
|
for key in skeys: |
||||||
|
val = TD[key] |
||||||
|
plpy.notice("TD[" + key + "] => " + str(val)) |
||||||
|
|
||||||
|
return None |
||||||
|
|
||||||
|
$$; |
||||||
|
|
||||||
|
CREATE TRIGGER show_trigger_data_trig |
||||||
|
BEFORE INSERT OR UPDATE OR DELETE ON trigger_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE trigger_data(23,'skidoo'); |
||||||
|
|
||||||
|
insert into trigger_test values(1,'insert'); |
||||||
|
update trigger_test set v = 'update' where i = 1; |
||||||
|
delete from trigger_test; |
||||||
|
|
||||||
|
DROP TRIGGER show_trigger_data_trig on trigger_test; |
||||||
|
|
||||||
|
DROP FUNCTION trigger_data(); |
||||||
@ -0,0 +1,37 @@ |
|||||||
|
-- |
||||||
|
-- Unicode handling |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE TABLE unicode_test ( |
||||||
|
testvalue text NOT NULL |
||||||
|
); |
||||||
|
|
||||||
|
CREATE FUNCTION unicode_return_error() RETURNS text AS E' |
||||||
|
return u"\\x80" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION unicode_trigger_error() RETURNS trigger AS E' |
||||||
|
TD["new"]["testvalue"] = u"\\x80" |
||||||
|
return "MODIFY" |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE TRIGGER unicode_test_bi BEFORE INSERT ON unicode_test |
||||||
|
FOR EACH ROW EXECUTE PROCEDURE unicode_trigger_error(); |
||||||
|
|
||||||
|
CREATE FUNCTION unicode_plan_error1() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue", ["text"]) |
||||||
|
rv = plpy.execute(plan, [u"\\x80"], 1) |
||||||
|
return rv[0]["testvalue"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION unicode_plan_error2() RETURNS text AS E' |
||||||
|
plan = plpy.prepare("SELECT $1 AS testvalue1, $2 AS testvalue2", ["text", "text"]) |
||||||
|
rv = plpy.execute(plan, u"\\x80", 1) |
||||||
|
return rv[0]["testvalue1"] |
||||||
|
' LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
SELECT unicode_return_error(); |
||||||
|
INSERT INTO unicode_test (testvalue) VALUES ('test'); |
||||||
|
SELECT unicode_plan_error1(); |
||||||
|
SELECT unicode_plan_error2(); |
||||||
@ -0,0 +1,22 @@ |
|||||||
|
-- |
||||||
|
-- Tests for functions that return void |
||||||
|
-- |
||||||
|
|
||||||
|
CREATE FUNCTION test_void_func1() RETURNS void AS $$ |
||||||
|
x = 10 |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
-- illegal: can't return non-None value in void-returning func |
||||||
|
CREATE FUNCTION test_void_func2() RETURNS void AS $$ |
||||||
|
return 10 |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
CREATE FUNCTION test_return_none() RETURNS int AS $$ |
||||||
|
None |
||||||
|
$$ LANGUAGE plpythonu; |
||||||
|
|
||||||
|
|
||||||
|
-- Tests for functions returning void |
||||||
|
SELECT test_void_func1(), test_void_func1() IS NULL AS "is null"; |
||||||
|
SELECT test_void_func2(); -- should fail |
||||||
|
SELECT test_return_none(), test_return_none() IS NULL AS "is null"; |
||||||
Loading…
Reference in new issue