src/pl/plpython/expected/plpython_import.out

   1 -- import python modules
   2 CREATE FUNCTION import_fail() returns text
   3     AS
   4 'try:
   5         import foosocket
   6 except ImportError:
   7         return "failed as expected"
   8 return "succeeded, that wasn''t supposed to happen"'
   9     LANGUAGE plpythonu;
  10 CREATE FUNCTION import_succeed() returns text
  11         AS
  12 'try:
  13   import array
  14   import bisect
  15   import calendar
  16   import cmath
  17   import errno
  18   import math
  19   import operator
  20   import random
  21   import re
  22   import string
  23   import time
  24 except Exception as ex:
  25         plpy.notice("import failed -- %s" % str(ex))
  26         return "failed, that wasn''t supposed to happen"
  27 return "succeeded, as expected"'
  28     LANGUAGE plpythonu;
  29 CREATE FUNCTION import_test_one(p text) RETURNS text
  30         AS
  31 'try:
  32     import hashlib
  33     digest = hashlib.sha1(p.encode("ascii"))
  34 except ImportError:
  35     import sha
  36     digest = sha.new(p)
  37 return digest.hexdigest()'
  38         LANGUAGE plpythonu;
  39 CREATE FUNCTION import_test_two(u users) RETURNS text
  40         AS
  41 'plain = u["fname"] + u["lname"]
  42 try:
  43     import hashlib
  44     digest = hashlib.sha1(plain.encode("ascii"))
  45 except ImportError:
  46     import sha
  47     digest = sha.new(plain);
  48 return "sha hash of " + plain + " is " + digest.hexdigest()'
  49         LANGUAGE plpythonu;
  50 -- import python modules
  51 --
  52 SELECT import_fail();
  53     import_fail
  54 --------------------
  55  failed as expected
  56 (1 row)
  57
  58 SELECT import_succeed();
  59      import_succeed
  60 ------------------------
  61  succeeded, as expected
  62 (1 row)
  63
  64 -- test import and simple argument handling
  65 --
  66 SELECT import_test_one('sha hash of this string');
  67              import_test_one
  68 ------------------------------------------
  69  a04e23cb9b1a09cd1051a04a7c571aae0f90346c
  70 (1 row)
  71
  72 -- test import and tuple argument handling
  73 --
  74 select import_test_two(users) from users where fname = 'willem';
  75                           import_test_two
  76 -------------------------------------------------------------------
  77  sha hash of willemdoe is 3cde6b574953b0ca937b4d76ebc40d534d910759
  78 (1 row)
  79