U gH@sddlmZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z mZejdkrndd lmZe reeegefZGd d d Zd d dddZdS)) annotationsN)CallableIterableMapping)AbstractContextManager) TracebackType) TYPE_CHECKINGAny) )BaseExceptionGroupc@sLeZdZddddZddddZd d d d d ddZdd dddZdS)_Catcherz2Mapping[tuple[type[BaseException], ...], _Handler]) handler_mapcCs ||_dSN) _handler_map)selfrr9/tmp/pip-unpacked-wheel-9cx8664k/exceptiongroup/_catch.py__init__sz_Catcher.__init__None)returncCsdSrr)rrrr __enter__sz_Catcher.__enter__ztype[BaseException] | NonezBaseException | NonezTracebackType | Nonebool)etypeexctbrcCsj|dk rf||}||krdS|dkr*dSt|tr`z ||jWntk r^|j|_YnX||dS)NFT)handle_exception isinstancer __cause__ __context__)rrrrZ unhandledrrr__exit__s   z_Catcher.__exit__ BaseException)rrc Csbt|tr|}n td|g}g}|jD]\}}||\}}|rz*z|Wntk rj||}YnXWnltk r}z$||kr||n ||jW5d}~XYnNtk r}z||W5d}~XYn$Xt |rt d|d|d||s*q q*|r0t |dkr&|dStd|S|rZt |jdkrZ|jd|krZ|S|SdS)NzError trying to handle z with z,. Exception handler must be a sync function.r r) rr ritemssplitappendextend exceptionsr"inspect iscoroutine TypeErrorlen) rrZexcgroupZnew_exceptionsZ exc_typeshandlermatchedresultnew_excrrrr2sL       z_Catcher.handle_exceptionN)__name__ __module__ __qualname__rrr!rrrrrrsrzFMapping[type[BaseException] | Iterable[type[BaseException]], _Handler]zAbstractContextManager[None]) __handlersrcCst|tstdi}|D]\}}t|trBt|trB|f}nt|trVt|}ntdt |sntd|D]2}t|trt|tstdt|t rrtdqr|||<qt |S)Nzthe argument must be a mappingzCeach key must be either an exception classes or an iterable thereofzhandlers must be callablezHcatching ExceptionGroup with catch() is not allowed. Use except instead.) rrr+r$type issubclassr"rtuplecallabler r)r4rZtype_or_iterabler-iterableexc_typerrrcatch`s>       r;) __future__rr)syscollections.abcrrr contextlibrtypesrtypingrr version_info _exceptionsr Z_Handlerrr;rrrrs     O