U ~HjE@sRddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z ddl m Z mZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZdd lmZ m!Z!m"Z"ddlm#Z#m$Z$m%Z%ddl m&Z&ddl'm(Z(dZ)dZ*ea+ej,ej-Z.Z/da0e1a2da3da4iZ5e6ee6e ddZ7ddZ8d(ddZ9Gddde:Zde dfddZ;ee"Gddde:ZGd"d#d#e:Z?d$d%Z@d&d'ZAe?t+ZBeBjCZCeBjDZDeBjEZEeBjFZFeBjGZGeBjHZHeBjIZIeBjJZJeBjKZKeBjLZLeBjMZMeBjNZNeBjOZOeBjPZPeBjQZQeBjRZReBjSZSeBjTZTejUZUejVZVdS))N)partialFailure) maybeDeferredDeferred DeferredList)succeedfail) IReactorTime)provider) IFailedFutureILogger log_levels)guess_stream_needs_encoding)_Config _BatchedTimer)_util)Logger formatEvent ILogObserver)globalLogBeginner formatTimeLogLevel)ensureDeferred)iscoroutinefunctionTFinfocOsdSN)argskwargsrr,/tmp/pip-unpacked-wheel-a0uj37kx/txaio/tx.py_no_opLsr"cCst|dSr) _categoriesupdate) categoriesrrr!add_log_categoriesPsr&cCs,|dk r$tjdk r$tj|k r$tdttS)NzTwisted has only a single, global reactor. You passed in a reactor different from the one already configured in txaio.config.loop)configloop RuntimeError_TxApi)r(rrr! with_configTs r+c@sTeZdZdddZdddZddZdd Zdd d Zd dZdddZ ddZ dS)rNcCsH|s td|||d|_d|_|r0||n |tt|dS)Nz$Should not be instantiated directly.)observer namespaceF)AssertionError_logger_log_level_set_explicitly set_log_level_set_log_level _log_level_loggersadd)selflevelloggerr-r,rrr!__init__gs   zLogger.__init__cCs|Srr)r6Zoselftyperrr!__get__uszLogger.__get__cOsDd|kr,|dtkr,t}t|d|d<|jj|f||dS)NZ log_categoryformat)r#tuplegetr/emitr6r7rr rrr!_log}sz Logger._logcOsJt|jt|krdS|dkr.|j||St|}|j|f||S)Ntrace)rindexr3_tracer lookupByNamerAr@rrr!r?s   z Logger.emitTcCs||||_dS)zt Set the log level. If keep is True, then it will not change along with global log changes. N)r2r0)r6r7Zkeeprrr!r1s zLogger.set_log_levelc Cst|}ttD]\}}|dkr$q||krjt||d}|tkrH|dkrTt||t|dkrt|dtqt||dtdfkr|dkrt|d|jnt||t|jt ||dkrt|d|j q||_ dS)NnoneerrorfailurerB) rrC enumerategetattrr"setattrrDrrArrE_failurer3)r6r7Z desired_indexidxnamecurrentrrr!r2s&   zLogger._set_log_levelcOs|jj|f||Sr)r/rH)r6r<rkwrrr!rLszLogger._failurecOs|j|ddi|dS)NZ txaio_traceT)debug)r6rrPrrr!rDsz Logger._trace)NNNN)N)T)N) __name__ __module__ __qualname__r9r;rAr?r1r2rLrDrrrr!res     rcCsltj}d|jkr2|jdj}d|j|j}n&|jd}|j j dkrX|d|j j }t ||||d}|S)Nr6z{0}.{1}rRz.)r7r-r8r,) inspect currentframef_backf_locals __class__r<rSrR f_globalsf_codeco_namer)r7r8r,cfclsr-rrr! make_loggers     r`c@sFeZdZdZejejejejej ej dZ ddZ ddZ ddZ d S) _LogObserverzQ Internal helper. An observer which formats events to a given file. )criticalrGwarnrrQrBcCs||_t||_d|_dSr)_filer_encode_levels)r6outrrr!r9s z_LogObserver.__init__cs4jdkr*ttfddtD_|jkS)Ncs,g|]$}t|kr|dkrj|qS)rF)rrCto_tx).0Zlvlr6Z target_levelrr! sz2_LogObserver._acceptable_level..)rfrrCr3)r6r7rrjr!_acceptable_levels    z_LogObserver._acceptable_levelcCs|ddkrJdt|dt|dtj}|jr<|d}|j|nNd|kr| |drdt|dt |tj}|jr|d}|j|dS)NZ log_formatz {0} {1}{2}Zlog_timeZ log_failureutf8Z log_level) r<rfailure_format_tracebackoslinesepreencoderdwriterlr)r6eventmsgrrr!__call__s$      z_LogObserver.__call__N)rRrSrT__doc__rrbrGrcrrQrhr9rlrurrrr!ras  racCsb|tkrtd|dttr&dSda|att|rBt|ag}trT| tt |dS)z[ Start logging to the file-like object in ``out``. By default, this is stdout. z'Invalid log level '{0}'; valid are: {1}z, NT) rr)r<join_started_loggingr3set_global_log_levelra _observerappendrZbeginLoggingTo)rgr7Z _observersrrr! start_loggings$ r|c@seZdZddZddZddZddZeed fd d Zd d Z d,ddZ ddZ ddZ ddZ d-ddZddZd.ddZd/ddZd0dd Zd1d!d"Zd#d$Zd2d&d'Zd(d)Zd*d+Zd S)3r*cCs ||_dSr)_config)r6r'rrr!r96sz_TxApi.__init__cCs<zd|jjj|WStk r6d|YSXdS)z_ :param fail: must be an IFailedFuture returns a unicode error-message z{0}: {1}z+Failed to produce failure message for "{0}"N)r<valuerZrRZgetErrorMessage Exceptionr6r rrr!failure_message9sz_TxApi.failure_messagecCs|jS)z\ :param fail: must be an IFailedFuture returns a traceback instance )tbrrrr!failure_tracebackFsz_TxApi.failure_tracebackcCsBzt}|j|d|WStk r<d|YSXdS)zP :param fail: must be an IFailedFuture returns a string )filez,Failed to format failure traceback for '{0}'N)ioStringIOZprintTracebackgetvaluerr<)r6r frrr!rnMs   z_TxApi.failure_format_tracebackNcCsL|tk r|tk rtdt|d}|tk r6t||n|tk rHt|||S)Nz"Cannot have both result and error.) canceller) _unspecified ValueErrorrresolvereject)r6resultrGrrrrr! create_futureYs   z_TxApi.create_futurecCst|Sr)r)r6rrrr!create_future_successdsz_TxApi.create_future_successcCs tt|Sr)r create_failure)r6rGrrr!create_future_errorgsz_TxApi.create_future_errorc OsXt|rHzt|||WStk rF}zt|WYSd}~XYnXt|f||Sr)rr TypeErrorrr)r6funrr errr! as_futurejs z_TxApi.as_futurecCs t|tSr) isinstancer)r6objrrr! is_futuressz_TxApi.is_futurecOst|j||f||Sr)r _get_loop callLater)r6delayrrr rrr! call_latervsz_TxApi.call_laterdcs,fdd}fdd}t|d|||dS)a& Creates and returns an object implementing :class:`txaio.IBatchedTimer`. :param bucket_seconds: the number of seconds in each bucket. That is, a value of 5 means that any timeout within a 5 second window will be in the same bucket, and get notified at the same time. This is only accurate to "milliseconds". :param chunk_size: when "doing" the callbacks in a particular bucket, this controls how many we do at once before yielding to the reactor. cs Sr)rsecondsrr6rr! get_secondssz._TxApi.make_batched_timer..get_secondscsj||f||Sr)rr)rrrr rrr!create_delayed_callsz6_TxApi.make_batched_timer..create_delayed_callg@@)Zseconds_providerZdelayed_call_creatorr)r6Zbucket_seconds chunk_sizerrrrr!make_batched_timerys  z_TxApi.make_batched_timercCs|jSr)called)r6futurerrr! is_calledsz_TxApi.is_calledcCs||dSr)callback)r6rrrrr!rsz_TxApi.resolvecCsD|dkrt}n&t|tr$t|}nt|ts6td||dS)Nz&reject requires a Failure or Exception)rrrrr)errback)r6rrGrrr!rs   z _TxApi.rejectcCs |dSr)cancel)r6rrtrrr!rsz _TxApi.cancelcCs|r t|StS)z Create a Failure instance. if ``exception`` is None (the default), we MUST be inside an "except" block. This encapsulates the exception into an object that implements IFailedFuture r)r6 exceptionrrr!rsz_TxApi.create_failurecCs<|dk s t|dkr,|dk s t||n ||||S)z] callback or errback may be None, but at least one must be non-None. N)r.Z addErrbackZ addCallbacks)r6rrrrrr! add_callbackss     z_TxApi.add_callbacksTcs,fdd}tt|d}t||d|S)Ncs0g}|D]"\}}|||ss|q|Sr)r{ZraiseException)resZrtnokr~consume_exceptionsrr! completeds    z _TxApi.gather..completed)Z consumeErrors)rlistr)r6Zfuturesrrdlrrr!gathers  z _TxApi.gathercCst}|||jd|S)z Inline sleep for use in co-routines. :param delay: Time to sleep in seconds. :type delay: float N)rrrr)r6rdrrr!sleepsz _TxApi.sleepcCs(|jjdkr ddlm}||j_|jjS)z! internal helper Nr)reactor)r}r(Ztwisted.internetr)r6rrrr!rs  z_TxApi._get_loop)N)r)N)N)N)N)T)rRrSrTr9rrrnrrrrrrrrrrrrrrrrrrrrr!r*4s(           r*cCs"tD]}|js||q|adS)zH Set the global log level on all loggers instantiated by txaio. N)r4r0r2r3)r7itemrrr!rys rycCstSr)r3rrrr!get_global_log_levelsr)N)WrrosysweakrefrV functoolsrZtwisted.python.failurerZtwisted.internet.deferrrrrr Ztwisted.internet.interfacesr Zzope.interfacer Ztxaio.interfacesr r rZ txaio._iotyperZtxaiorZ txaio._commonrrZtwisted.loggerrZ_LoggerrrrrrrZasynciorZ using_twistedZ using_asyncior'stderrstdout_stderrZ_stdoutrzWeakSetr4r3rxr#registerr"r&r+objectr`rar|rr*ryrZ _default_apirrrnrrrrrrrrrrrrrrrtime_nsperf_counter_nsrrrr!s             \<;