ó ě $Sc@södZdZdZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z ddlmZd„Zdefd „ƒYZd ejfd „ƒYZd „Zd ejfd„ƒYZdS(sYaroslav Halchenkos%Copyright (c) 2013 Yaroslav HalchenkotGPLi˙˙˙˙N(tbasenametdirname(tStringIOcCsat|ƒ}|jdƒr(|d }n|tddgƒkr]tt|ƒƒd|}n|S(saCustom function to include directory name if filename is too common Also strip .py at the end s.pyiý˙˙˙tbaset__init__t.(RtendswithtsetR(tsR((s&/usr/share/fail2ban/testcases/utils.pyt mbasename#s   t TraceBackcBs#eZdZed„Zd„ZRS(s7Customized traceback to be included in debug messages cCsd|_||_dS(sžInitialize TrackBack metric Parameters ---------- compress : bool if True then prefix common with previous invocation gets replaced with ... tN(t_TraceBack__prevt_TraceBack__compress(tselftcompress((s&/usr/share/fail2ban/testcases/utils.pyR3s c Cs˛tjddƒd }g|D]3}t|dƒt|dƒt|dƒg^q}g|D]=}|ddkpƒ|djdƒs]|d|d g^q]}|dg}xV|dD]J}|d|d dkrő|d dcd |d7s%s:%ss>[^>]*$R s...(sunittestslogging.__init__(t tracebackt extract_stackR RtstrRtappendtjoinRtostpatht commonprefixR tretsubtlen( Rtftbtxtentriestet entries_outtentrytsftbt prev_nextt common_prefixtcommon_prefix2((s&/usr/share/fail2ban/testcases/utils.pyt__call__?s(@ -  1   (t__name__t __module__t__doc__tFalseRR)(((s&/usr/share/fail2ban/testcases/utils.pyR /s tFormatterWithTraceBackcBs eZdZd„Zd„ZRS(s}Custom formatter which expands %(tb) and %(tbc) with tracebacks TODO: might need locking in case of compressed tracebacks cOs>tjj|d|||Žd|k}td|ƒ|_dS(Ntfmts%(tbc)sR(tloggingt FormatterRR t_tb(RR/targstkwargsR((s&/usr/share/fail2ban/testcases/utils.pyR`s cCs)|jƒ|_|_tjj||ƒS(N(R2ttbcttbR0R1tformat(Rtrecord((s&/usr/share/fail2ban/testcases/utils.pyR7es(R*R+R,RR7(((s&/usr/share/fail2ban/testcases/utils.pyR.[s cCsdS(N((((s&/usr/share/fail2ban/testcases/utils.pyt mtimesleepistLogCaptureTestCasecBs#eZd„Zd„Zd„ZRS(cCsetjdƒ}|j|_|j|_tƒ|_tj|jƒg|_|j t tdƒƒdS(Ntfail2bantDEBUG( R0t getLoggertlevelt _old_levelthandlerst _old_handlersRt_logt StreamHandlertsetLeveltgetattr(RtlogSys((s&/usr/share/fail2ban/testcases/utils.pytsetUpps    cCs+tjdƒ}|j|_|j|_dS(sCall after every test case.R;N(R0R=RAR@R?R>(RRF((s&/usr/share/fail2ban/testcases/utils.pyttearDown~s cCs||jjƒkS(N(RBtgetvalue(RR ((s&/usr/share/fail2ban/testcases/utils.pyt _is_logged…s(R*R+RGRHRJ(((s&/usr/share/fail2ban/testcases/utils.pyR:ns  (t __author__t __copyright__t __license__RR0RRttempfiletsysttimeRtos.pathRRRR tobjectR R1R.R9tTestCaseR:(((s&/usr/share/fail2ban/testcases/utils.pyts` ,