DT_Try.py
上传用户:gyjinxi
上传日期:2007-01-04
资源大小:159k
文件大小:11k
源码类别:

WEB邮件程序

开发平台:

Python

  1. ##############################################################################
  2. # Zope Public License (ZPL) Version 1.0
  3. # -------------------------------------
  4. # Copyright (c) Digital Creations.  All rights reserved.
  5. # This license has been certified as Open Source(tm).
  6. # Redistribution and use in source and binary forms, with or without
  7. # modification, are permitted provided that the following conditions are
  8. # met:
  9. # 1. Redistributions in source code must retain the above copyright
  10. #    notice, this list of conditions, and the following disclaimer.
  11. # 2. Redistributions in binary form must reproduce the above copyright
  12. #    notice, this list of conditions, and the following disclaimer in
  13. #    the documentation and/or other materials provided with the
  14. #    distribution.
  15. # 3. Digital Creations requests that attribution be given to Zope
  16. #    in any manner possible. Zope includes a "Powered by Zope"
  17. #    button that is installed by default. While it is not a license
  18. #    violation to remove this button, it is requested that the
  19. #    attribution remain. A significant investment has been put
  20. #    into Zope, and this effort will continue if the Zope community
  21. #    continues to grow. This is one way to assure that growth.
  22. # 4. All advertising materials and documentation mentioning
  23. #    features derived from or use of this software must display
  24. #    the following acknowledgement:
  25. #      "This product includes software developed by Digital Creations
  26. #      for use in the Z Object Publishing Environment
  27. #      (http://www.zope.org/)."
  28. #    In the event that the product being advertised includes an
  29. #    intact Zope distribution (with copyright and license included)
  30. #    then this clause is waived.
  31. # 5. Names associated with Zope or Digital Creations must not be used to
  32. #    endorse or promote products derived from this software without
  33. #    prior written permission from Digital Creations.
  34. # 6. Modified redistributions of any form whatsoever must retain
  35. #    the following acknowledgment:
  36. #      "This product includes software developed by Digital Creations
  37. #      for use in the Z Object Publishing Environment
  38. #      (http://www.zope.org/)."
  39. #    Intact (re-)distributions of any official Zope release do not
  40. #    require an external acknowledgement.
  41. # 7. Modifications are encouraged but must be packaged separately as
  42. #    patches to official Zope releases.  Distributions that do not
  43. #    clearly separate the patches from the original work must be clearly
  44. #    labeled as unofficial distributions.  Modifications which do not
  45. #    carry the name Zope may be packaged in any form, as long as they
  46. #    conform to all of the clauses above.
  47. # Disclaimer
  48. #   THIS SOFTWARE IS PROVIDED BY DIGITAL CREATIONS ``AS IS'' AND ANY
  49. #   EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  50. #   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  51. #   PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL DIGITAL CREATIONS OR ITS
  52. #   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  53. #   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  54. #   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  55. #   USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  56. #   ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  57. #   OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  58. #   OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  59. #   SUCH DAMAGE.
  60. # This software consists of contributions made by Digital Creations and
  61. # many individuals on behalf of Digital Creations.  Specific
  62. # attributions are listed in the accompanying credits file.
  63. ##############################################################################
  64. import string, sys, traceback
  65. from cStringIO import StringIO
  66. from DT_Util import ParseError, parse_params, render_blocks
  67. from DT_Util import namespace, InstanceDict
  68. from DT_Return import DTReturn
  69. class Try:
  70.     """Zope DTML Exception handling
  71.     
  72.     usage:
  73.     
  74.     <!--#try-->
  75.     <!--#except SomeError AnotherError-->
  76.     <!--#except YetAnotherError-->
  77.     <!--#except-->
  78.     <!--#else-->
  79.     <!--#/try-->
  80.       
  81.     or:
  82.       
  83.     <!--#try-->
  84.     <!--#finally-->
  85.     <!--#/try-->
  86.     
  87.     The DTML try tag functions quite like Python's try command.
  88.     
  89.     The contents of the try tag are rendered. If an exception is raised,
  90.     then control switches to the except blocks. The first except block to
  91.     match the type of the error raised is rendered. If an except block has
  92.     no name then it matches all raised errors.
  93.     
  94.     The try tag understands class-based exceptions, as well as string-based
  95.     exceptions. Note: the 'raise' tag raises string-based exceptions.
  96.     
  97.     Inside the except blocks information about the error is available via
  98.     three variables.
  99.     
  100.       'error_type' -- This variable is the name of the exception caught.
  101.     
  102.       'error_value' -- This is the caught exception's value.
  103.     
  104.       'error_tb' -- This is a traceback for the caught exception.
  105.       
  106.     The optional else block is rendered when no exception occurs in the
  107.     try block. Exceptions in the else block are not handled by the preceding
  108.     except blocks.
  109.     The try..finally form specifies a `cleanup` block, to be rendered even
  110.     when an exception occurs. Note that any rendered result is discarded if
  111.     an exception occurs in either the try or finally blocks. The finally block
  112.     is only of any use if you need to clean up something that will not be
  113.     cleaned up by the transaction abort code.
  114.     The finally block will always be called, wether there was an exception in
  115.     the try block or not, or wether or not you used a return tag in the try
  116.     block. Note that any output of the finally block is discarded if you use a
  117.     return tag in the try block.
  118.     If an exception occurs in the try block, and an exception occurs in the
  119.     finally block, or you use the return tag in that block, any information
  120.     about that first exception is lost. No information about the first
  121.     exception is available in the finally block. Also, if you use a return tag
  122.     in the try block, and an exception occurs in the finally block or you use
  123.     a return tag there as well, the result returned in the try block will be
  124.     lost.
  125.     Original version by Jordan B. Baker.
  126.     
  127.     Try..finally and try..else implementation by Martijn Pieters.
  128.     """
  129.     
  130.     name = 'try'
  131.     blockContinuations = 'except', 'else', 'finally'
  132.     finallyBlock=None
  133.     elseBlock=None
  134.     def __init__(self, blocks):
  135.         tname, args, section = blocks[0]
  136.         self.args = parse_params(args)
  137.         self.section = section.blocks
  138.         # Find out if this is a try..finally type
  139.         if len(blocks) == 2 and blocks[1][0] == 'finally':
  140.             self.finallyBlock = blocks[1][2].blocks
  141.         # This is a try [except]* [else] block.
  142.         else:
  143.             # store handlers as tuples (name,block)
  144.             self.handlers = []
  145.             defaultHandlerFound = 0
  146.             for tname,nargs,nsection in blocks[1:]:
  147.                 if tname == 'else':
  148.                     if not self.elseBlock is None:
  149.                         raise ParseError, (
  150.                             'No more than one else block is allowed',
  151.                             self.name)
  152.                     self.elseBlock = nsection.blocks
  153.                 elif tname == 'finally':
  154.                     raise ParseError, (
  155.                         'A try..finally combination cannot contain '
  156.                         'any other else, except or finally blocks',
  157.                         self.name)
  158.                 else:
  159.                     if not self.elseBlock is None:
  160.                         raise ParseError, (
  161.                             'The else block should be the last block '
  162.                             'in a try tag', self.name)
  163.                     for errname in string.split(nargs):
  164.                         self.handlers.append((errname,nsection.blocks))
  165.                     if string.strip(nargs)=='':
  166.                         if defaultHandlerFound:
  167.                             raise ParseError, (
  168.                                 'Only one default exception handler '
  169.                                 'is allowed', self.name)
  170.                         else:
  171.                             defaultHandlerFound = 1
  172.                             self.handlers.append(('',nsection.blocks))
  173.     def render(self, md):
  174.         if (self.finallyBlock is None):
  175.             return self.render_try_except(md)
  176.         else:
  177.             return self.render_try_finally(md)
  178.     def render_try_except(self, md):
  179.         result = ''
  180.         # first we try to render the first block
  181.         try:
  182.             result = render_blocks(self.section, md)
  183.         except DTReturn:
  184.             raise
  185.         except:
  186.             # but an error occurs.. save the info.
  187.             t,v = sys.exc_info()[:2]
  188.             if type(t)==type(''):
  189.                 errname = t
  190.             else:
  191.                 errname = t.__name__
  192.             handler = self.find_handler(t)
  193.                                     
  194.             if handler is None:
  195.                 # we didn't find a handler, so reraise the error
  196.                 raise
  197.             # found the handler block, now render it
  198.             try:
  199.                 f=StringIO()
  200.                 traceback.print_exc(100,f)
  201.                 error_tb=f.getvalue()
  202.                 ns = namespace(self, error_type=errname, error_value=v,
  203.                     error_tb=error_tb)[0]
  204.                 md._push(InstanceDict(ns,md))
  205.                 return render_blocks(handler, md)
  206.             finally:
  207.                 md._pop(1)
  208.         else:
  209.             # No errors have occured, render the optional else block
  210.             if (self.elseBlock is None):
  211.                 return result
  212.             else:
  213.                 return result + render_blocks(self.elseBlock, md)
  214.                
  215.     def render_try_finally(self, md):
  216.         result = ''
  217.         # first try to render the first block
  218.         try:
  219.             result = render_blocks(self.section, md)
  220.         # Then handle finally block
  221.         finally:
  222.             result = result + render_blocks(self.finallyBlock, md)
  223.         return result
  224.     def find_handler(self,exception):
  225.         "recursively search for a handler for a given exception"
  226.         if type(exception)==type(''):
  227.             for e,h in self.handlers:
  228.                 if exception==e or e=='':
  229.                     return h
  230.             else:
  231.                 return None
  232.         for e,h in self.handlers:
  233.             if e==exception.__name__ or e=='' or self.match_base(exception,e):
  234.                 return h    
  235.         return None 
  236.     def match_base(self,exception,name):
  237.         for base in exception.__bases__:
  238.             if base.__name__==name or self.match_base(base,name):
  239.                 return 1
  240.         return None
  241.         
  242.     __call__ = render