U
    e,P                  
   @  sh  U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlmZ d dlmZmZmZmZ d dlZd dlmZ d dlmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dl m!Z! ed Z"ed Z#ed Z$ed Z%ed Z&ej'ej(ej)ej*ej+edZ,de-d< ddddZ.de-d< dddddZ/de-d< ddddZ0d e-d!< dd"d#d$dZ1d%e-d&< d'd(d)d*gZ2d+e-d,< e	j3Z4d-e-d.< d/d0d1d2dd3d4d5d6d7d8d9d:d;d<d9d=d;d7d8gd>d0d?d@d>id<gd>d0d?dAdBZ5dCe-dD< e6dEZ7dFdGdHd-d-dGdHdIdJdKdLZ8dMdNdOdPdQZ9dRdRdSdTdUdVZ:dWdRdXdYdZZ;G d[d\ d\Z<dS )]    )annotationsN)Path)Any	AwaitableCallableLiteral)ASGIApplication)ImportFromStringErrorimport_from_string)TRACE_LOG_LEVEL)ASGI2Middleware)MessageLoggerMiddleware)ProxyHeadersMiddleware)WSGIMiddleware)autoZh11Z	httptools)r   noneZ
websocketsZwsproto)r   onoff)r   r   asyncioZuvloop)r   asgi3asgi2wsgi)criticalerrorwarninginfodebugtracezdict[str, int]
LOG_LEVELSz,uvicorn.protocols.http.auto:AutoHTTPProtocolz+uvicorn.protocols.http.h11_impl:H11Protocolz7uvicorn.protocols.http.httptools_impl:HttpToolsProtocolzdict[HTTPProtocolType, str]HTTP_PROTOCOLSz8uvicorn.protocols.websockets.auto:AutoWebSocketsProtocolz>uvicorn.protocols.websockets.websockets_impl:WebSocketProtocolz4uvicorn.protocols.websockets.wsproto_impl:WSProtocolz dict[WSProtocolType, str | None]WS_PROTOCOLSzuvicorn.lifespan.on:LifespanOnz uvicorn.lifespan.off:LifespanOffzdict[LifespanType, str]LIFESPANz"uvicorn.loops.auto:auto_loop_setupz#uvicorn.loops.asyncio:asyncio_setupz!uvicorn.loops.uvloop:uvloop_setupzdict[LoopSetupType, str | None]LOOP_SETUPSr   r   r   r   zlist[InterfaceType]
INTERFACESintSSL_PROTOCOL_VERSION   Fz uvicorn.logging.DefaultFormatterz%(levelprefix)s %(message)s)()fmt
use_colorszuvicorn.logging.AccessFormatterzD%(levelprefix)s %(client_addr)s - "%(request_line)s" %(status_code)s)r'   r(   )defaultaccessr*   zlogging.StreamHandlerzext://sys.stderr)	formatterclassstreamr+   zext://sys.stdoutINFO)handlerslevel	propagater1   )Zuvicornuvicorn.erroruvicorn.access)versiondisable_existing_loggers
formattersr0   loggerszdict[str, Any]LOGGING_CONFIGr3   zstr | os.PathLike[str]str | os.PathLike[str] | None
str | Nonezssl.SSLContext)certfilekeyfilepasswordssl_version	cert_reqsca_certsciphersreturnc           	        sX   t |} r fddnd }|| || t ||_|rF|| |rT|| |S )Nc                     s    S N rE   r>   rE   2/tmp/pip-unpacked-wheel-cq1xfpla/uvicorn/config.py<lambda>o       z$create_ssl_context.<locals>.<lambda>)ssl
SSLContextload_cert_chain
VerifyModeverify_modeload_verify_locationsset_ciphers)	r<   r=   r>   r?   r@   rA   rB   ctxget_passwordrE   rF   rG   create_ssl_contexte   s    	


rS   r   bool)pathrC   c                 C  s6   z|   s|  } |  W S  tk
r0   Y dS X d S )NF)is_absoluteresolveis_dirOSError)rU   rE   rE   rG   rX   y   s    
rX   z	list[str]ztuple[list[str], list[Path]])patterns_listdirectories_listrC   c           
      C  sX  t ttt| }|  }t }| D ]V}|dkr8q*|| tt|r^|t| q*||D ]}t|rh|| qhq*t t|}t tt|}t tdd |}t dd |D }g }t	t
|D ]`}t	|d t
|D ]H}	|| ||	 jkr|||	  q||	 || jkr|||  qqt t|t|}t t||fS )Nz.*c                 S  s   |   S rD   )rW   )xrE   rE   rG   rH      rI   z)resolve_reload_patterns.<locals>.<lambda>c                 S  s   h | ]}t |r|qS rE   )rX   ).0Zreload_pathrE   rE   rG   	<setcomp>   s      z*resolve_reload_patterns.<locals>.<setcomp>r&   )listsetmapr   copycwdappendrX   globrangelenparents
difference)
rZ   r[   directoriespatternsZcurrent_working_directorypatternmatchchildrenjkrE   rE   rG   resolve_reload_patterns   s2    
rq   list[str] | str | None)dirsrC   c                 C  s(   | d krg S t | tr| gS tt| S rD   )
isinstancestrr_   r`   )rs   rE   rE   rG   _normalize_dirs   s
    
rv   c                2   @  sZ  e Zd Zddddddddddddddeddddd	dd
ddddddddddddddddddeejdddd	df/dddddddddddddddddddddd d!d d ddddd dddddddd"dddddddd#ddd$0d%d&Zed'd(d)d*Z	edd(d+d,Z
edd(d-d.Zd/d(d0d1Zd/d(d2d3Zd/d(d4d5Zd6d(d7d8Zedd(d9d:ZdS );Config	127.0.0.1i@  Nr   i       g      4@TFg      ? i         TLSv1z*ASGIApplication | Callable[..., Any] | strru   r$   r;   z
int | NoneLoopSetupTypez)type[asyncio.Protocol] | HTTPProtocolTypez'type[asyncio.Protocol] | WSProtocolTypezfloat | NonerT   LifespanTyper:   zdict[str, Any] | str | Nonezstr | int | Nonezbool | NoneInterfaceTyperr   floatz%Callable[..., Awaitable[None]] | Nonezlist[tuple[str, str]] | None)0apphostportudsfdloophttpwsws_max_sizews_max_queuews_ping_intervalws_ping_timeoutws_per_message_deflatelifespanenv_file
log_config	log_level
access_logr)   	interfacereloadreload_dirsreload_delayreload_includesreload_excludesworkersproxy_headersserver_headerdate_headerforwarded_allow_ips	root_pathlimit_concurrencylimit_max_requestsbacklogtimeout_keep_alivetimeout_notifytimeout_graceful_shutdowncallback_notifyssl_keyfilessl_certfilessl_keyfile_passwordr?   ssl_cert_reqsssl_ca_certsssl_ciphersheadersfactoryh11_max_incomplete_event_sizec1           6   
   C  s  || _ || _|| _|| _|| _|| _|| _|| _|	| _|
| _	|| _
|| _|| _|| _|| _|| _|| _|| _|| _|| _|| _|pd| _|| _|| _|| _|| _| | _|!| _|"| _|#| _|$| _|%| _|&| _ |'| _!|(| _"|)| _#|*| _$|+| _%|,| _&|-| _'|.pg | _(g | _)|/| _*|0| _+d| _,| -  g | _.g | _/g | _0g | _1|sH|sH|rZ| j2sZt34d | j2rlt5|}t5|}t5|}t6||\| _0| _.t6|g \| _1| _/| j.7 }1| j/D ]P}2|1D ]D}3|2|3ks|2|3j8krz| j.9|3 W n t:k
r   Y nX qq| j1D ]}4|4| j0kr| j09|4 q| j.sP|r@t34d| t;t<= g| _.t3>dt?t@tAtB| j. |d k	rddlCmD}5 t3>d| |5|d	 |d krd
t<jEkrtFt<jEd
 | _|  |d krt<jEGdd| _Hn|| _H| jr| jdkrt34d d S )Nr&   FzcCurrent configuration will not reload as not all conditions are met, please refer to documentation.zeProvided reload directories %s did not contain valid directories, watching current working directory.z/Will watch for changes in these directories: %sr   )load_dotenvzLoading environment from '%s')Zdotenv_pathZWEB_CONCURRENCYZFORWARDED_ALLOW_IPSrx   z4"workers" flag is ignored when reloading is enabled.)Ir   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r)   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r?   r   r   r   r   encoded_headersr   r   loadedconfigure_loggingr   Zreload_dirs_excludesr   r   should_reloadloggerr   rv   rq   rb   rh   remove
ValueErrorr   osgetcwdr   sortedr_   ra   ru   Zdotenvr   environr$   getr   )6selfr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r)   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r?   r   r   r   r   r   r   Zreload_dirs_tmp	directoryZreload_directoryrl   r   rE   rE   rG   __init__   s    3







zConfig.__init__zLiteral[('2.0', '3.0')])rC   c                 C  s   dddd}|| j  S )Nz2.0z3.0)r   r   r   )r   )r   mappingrE   rE   rG   asgi_versionS  s
    zConfig.asgi_versionc                 C  s   t | jp| jS rD   )rT   r   r   r   rE   rE   rG   is_ssl\  s    zConfig.is_sslc                 C  s   t | jp| jdkS )Nr&   )rT   r   r   r   rE   rE   rG   use_subprocess`  s    zConfig.use_subprocessNonec              	   C  sv  t td | jd k	rt| jtrd| jdkrT| j| jd d d< | j| jd d d< t j| j n| j	drt
| j}t|}t j| W 5 Q R X nT| j	drd	d l}t
| j}||}t j| W 5 Q R X nt jj| jd
d | jd k	rNt| jtrt| j }n| j}t d| t d| t d| | jd
krrg t d_d
t d_d S )NTRACE)TFr7   r*   r)   r+   z.json)z.yamlz.ymlr   F)r6   r3   r4   zuvicorn.asgi)loggingaddLevelNamer   r   rt   dictr)   config
dictConfigendswithopenjsonloadyamlZ	safe_load
fileConfigr   ru   r   	getLoggersetLevelr   r0   r2   )r   fileZloaded_configr   r   rE   rE   rG   r   d  s6    



zConfig.configure_loggingc              
   C  sr  | j r
t| jrB| jstt| j| j| j| j| j| j	| j
d| _nd | _dd | jD }dt|krt| jrtdg| n|| _t| jtrtt| j }|| _n| j| _t| jtrtt| j }|| _n| j| _tt| j | _zt| j| _W n< tk
r* } zt d|  t!"d W 5 d }~X Y nX z|  | _W nB t#k
r| } z"| j$rlt d| t!"d W 5 d }~X Y nX | j$st%d	 | j&d
krt'(| jrt)| jd}n4t'*| jrt+,| j}nt-| jdd }t+,|}|rdnd| _&| j&dkrt.| j| _d | _n| j&dkr4t/| j| _t0 t1krNt2| j| _| j3rht4| j| j5d| _d| _ d S )N)r=   r<   r>   r?   r@   rA   rB   c                 S  s(   g | ] \}}|  d |d fqS )latin1)lowerencode)r]   keyvaluerE   rE   rG   
<listcomp>  s     zConfig.load.<locals>.<listcomp>   server)r   s   uvicornzError loading ASGI app. %sr&   z"Error loading ASGI app factory: %sz_ASGI app factory detected. Using it, but please consider setting the --factory flag explicitly.r   	__await____call__r   r   r   )trusted_hostsT)6r   AssertionErrorr   r   rS   r   r   r?   r   r   r   rJ   r   r   r   r   rt   r   ru   r
   r   http_protocol_classr   r    ws_protocol_classr!   r   Zlifespan_classr   Z
loaded_appr	   r   r   sysexit	TypeErrorr   r   r   inspectisclasshasattr
isfunctionr   iscoroutinefunctiongetattrr   r   getEffectiveLevelr   r   r   r   r   )r   r   r   r   excZ
use_asgi_3callrE   rE   rG   r     s|    





zConfig.loadc                 C  s&   t t| j }|d k	r"|| jd d S )N)r   )r
   r"   r   r   )r   Z
loop_setuprE   rE   rG   setup_event_loop  s    zConfig.setup_event_loopzsocket.socketc              
   C  s  | j r| j }ttjtj}z || d}t| j | W n6 tk
rr } zt	| t
d W 5 d }~X Y nX d}d}dtj|dd d }| j g}n | jrt| jtjtj}d	}d}	dtj|	dd d }| g}ntj}
d
}| jr
d| jkr
tj}
d}tj|
d}|tjtjd z|| j| jf W n8 tk
rv } zt	| t
d W 5 d }~X Y nX d| d}dtj|dd d }| jrdnd}|| j| d g}tj|f|dd|ii |d |S )Ni  r&   z8Uvicorn running on unix socket %s (Press CTRL+C to quit)z%szUvicorn running on T)Zboldz (Press CTRL+C to quit)z3Uvicorn running on socket %s (Press CTRL+C to quit)z
%s://%s:%d:z%s://[%s]:%d)familyhttpsr   extracolor_message)r   socketAF_UNIXSOCK_STREAMbindr   chmodrY   r   r   r   r   clickstyler   fromfdgetsocknameAF_INETr   AF_INET6
setsockopt
SOL_SOCKETSO_REUSEADDRr   r   r   set_inheritable)r   rU   sockZ	uds_permsr   messageZsock_name_formatr   Zlogger_argsZfd_name_formatr   Zaddr_formatZprotocol_namerE   rE   rG   bind_socket  sN    



zConfig.bind_socketc                 C  s   t | jto| jS rD   )rt   r   ru   r   r   rE   rE   rG   r     s    zConfig.should_reload)__name__
__module____qualname__r9   r%   rJ   	CERT_NONEr   propertyr   r   r   r   r   r   r  r   rE   rE   rE   rG   rw      sz   n %&O0rw   )=
__future__r   r   r   r   r   logging.configr   r   rJ   r   pathlibr   typingr   r   r   r   r   Zuvicorn._typesr   Zuvicorn.importerr	   r
   Zuvicorn.loggingr   Zuvicorn.middleware.asgi2r   Z!uvicorn.middleware.message_loggerr   Z uvicorn.middleware.proxy_headersr   Zuvicorn.middleware.wsgir   ZHTTPProtocolTypeZWSProtocolTyper   r~   r   CRITICALERRORWARNINGr/   DEBUGr   __annotations__r   r    r!   r"   r#   PROTOCOL_TLS_SERVERr%   r9   r   r   rS   rX   rq   rv   rw   rE   rE   rE   rG   <module>   s    	!
	$