From 788eeb9bd5516c330a551d71664b45c69f846dce Mon Sep 17 00:00:00 2001 From: Takafumi Arakaki Date: Fri, 24 May 2013 19:10:19 +0200 Subject: [PATCH] Simplify how post_mortem is launched --- sith.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/sith.py b/sith.py index 659bbb3b..2f1eda18 100755 --- a/sith.py +++ b/sith.py @@ -247,16 +247,10 @@ class AttackApp(object): try: func(**kwds) except: - exc_info = sys.exc_info() - if debugger == 'pdb': - import pdb - pdb.post_mortem(exc_info[2]) - elif debugger == 'ipdb': - import ipdb - ipdb.post_mortem(exc_info[2]) - elif debugger == 'pudb': - import pudb - pudb.post_mortem(exc_info) + if debugger: + einfo = sys.exc_info() + pdb = __import__(debugger) + pdb.post_mortem(einfo if debugger == 'pudb' else einfo[2]) def add_parser(self, attacker_class, *args, **kwds): attacker = attacker_class()