提交: 23f55a39ea529fb2111b99944cd457733b95c3a2 - allura (git)

allura

Commit MetaInfo

• 修訂: 23f55a39ea529fb2111b99944cd457733b95c3a2 (tree)
• 時間: 2012-07-04 02:52:42
• 作者: Cory Johns <johnsca@geek...>
• Commiter: Cory Johns

Log Message

Suppress warnings when running scripts

Signed-off-by: Cory Johns <[e-mail]>

Change Summary

• modified: Allura/allura/command/script.py

差異

--- a/Allura/allura/command/script.py
+++ b/Allura/allura/command/script.py
@@ -1,10 +1,12 @@
 import sys
 import os.path
 import cProfile
+import warnings
 
 from pylons import c
 import pylons
 import webob
+from sqlalchemy import exc
 
 from ming.orm import session
 from allura.lib import helpers as h
@@ -23,20 +25,22 @@ class ScriptCommand(base.Command):
                       help='Drop to a debugger on error')
 
     def command(self):
-        self.basic_setup()
-        request = webob.Request.blank('--script--', environ={
-                'paste.registry':self.registry})
-        self.registry.register(pylons.request, request)
-        if self.options.pdb:
-            base.log.info('Installing exception hook')
-            sys.excepthook = utils.postmortem_hook
-        with open(self.args[1]) as fp:
-            ns = dict(__name__='__main__')
-            sys.argv = self.args[1:]
-            if self.options.profile:
-                cProfile.run(fp, '%s.profile' % os.path.basename(self.args[1]))
-            else:
-                exec fp in ns
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", category=exc.SAWarning)
+            self.basic_setup()
+            request = webob.Request.blank('--script--', environ={
+                    'paste.registry':self.registry})
+            self.registry.register(pylons.request, request)
+            if self.options.pdb:
+                base.log.info('Installing exception hook')
+                sys.excepthook = utils.postmortem_hook
+            with open(self.args[1]) as fp:
+                ns = dict(__name__='__main__')
+                sys.argv = self.args[1:]
+                if self.options.profile:
+                    cProfile.run(fp, '%s.profile' % os.path.basename(self.args[1]))
+                else:
+                    exec fp in ns
 
 class SetToolAccessCommand(base.Command):
     min_args=3