Z/ @ s d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d Z Gd d
d
e Z
d d
Z Gd d d e Z Gd d d e Z
Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z d S) N)ABCabstractmethod)URLTCPSiteUnixSiteSockSite
BaseRunner AppRunner ServerRunner GracefulExitc @ s e Z d Z d Z d S)r N)__name__
__module__ __qualname__code r r /web_runner.pyr
s c C s
t d S)N)r r r r r _raise_graceful_exit s r c @ sj e Z d Z d Z d d d d d
d d d
Z e e d d Z e d d Z d d Z d S)BaseSite_runner_shutdown_timeout _ssl_context_backlog_servershutdown_timeoutg N@ ssl_contextNbacklog c C sL | j d k r t d | | _ | | _ | | _ | | _ d | _ d S)Nz(Call runner.setup() before making a site)server RuntimeErrorr r r r r )selfrunnerr r r r r r __init__ s zBaseSite.__init__c C s d S)Nr )r r r r name$ s z
BaseSite.namec s | j j | d S)N)r _reg_site)r r r r start) s zBaseSite.startc s | j j | | j d k r3 | j j | d S| j j | j j Id H| j j Id H| j j j | j Id H| j j | d S)N) r _check_siter _unreg_sitecloseZ wait_closedshutdownr r )r r r r stop- s
z
BaseSite.stop)r r r r r )
r
r r __slots__r" propertyr r# r% r* r r r r r s r c sp e Z d Z d Z d d d d d d d d
d d d d f d
d Z e d d Z f d d Z S)r _host_port_reuse_address _reuse_portNr g N@r r r
reuse_address
reuse_portc s} t j | d | d | d | | d k r4 d } | | _ | d k r^ | j rX d n d } | | _ | | _ | | _ d S)Nr r r z0.0.0.0i i )superr" r- r r. r/ r0 ) r r! hostportr r r r1 r2 ) __class__r r r" < s
zTCPSite.__init__c C s= | j r d n d } t t j d | d | j d | j S)Nhttpshttpschemer4 r5 )r strr buildr- r. )r r9 r r r r# K s z TCPSite.namec so t j Id Ht j } | j | j j | j | j d | j d | j
d | j d | j Id H| _
d S)Nsslr r1 r2 )r3 r% asyncioget_event_loop
create_serverr r r- r. r r r/ r0 r )r loop)r6 r r r% P s z
TCPSite.start)r- r. r/ r0 )r
r r r+ r" r, r# r% r r )r6 r r 9 s c s^ e Z d Z d Z d d d d d d f d d Z e d
d Z f d d
Z S)r _pathr g N@r Nr r c s/ t j | d | d | d | | | _ d S)Nr r r )r3 r" rA )r r! pathr r r )r6 r r r" ] s
zUnixSite.__init__c C s( | j r d n d } d j | | j S)Nr7 r8 z
{}://unix:{}:)r formatrA )r r9 r r r r# d s z
UnixSite.namec sW t j Id Ht j } | j | j j | j d | j d | j Id H| _
d S)Nr< r ) r3 r% r= r> Zcreate_unix_serverr r rA r r r )r r@ )r6 r r r% i s
zUnixSite.start)rA )r
r r r+ r" r, r# r% r r )r6 r r Z s c s^ e Z d Z d Z d d d d d d f d d
Z e d d Z f d
d Z S)r _sock_namer g N@r Nr r c
s t j | d | d | d | | | _ | j r: d n d } t t d r| | j t j k r| d j | | j } n@ | j d d \ } } t
t j d | d
| d | } | | _
d S) Nr r r r7 r8 AF_UNIXz
{}://unix:{}: r9 r4 r5 )r3 r" rD r hasattrsocketfamilyrF rC getsocknamer: r r; rE )
r r! sockr r r r9 r# r4 r5 )r6 r r r" t s
! $zSockSite.__init__c C s | j S)N)rE )r r r r r# s z
SockSite.namec sZ t j Id Ht j } | j | j j d | j d | j d | j Id H| _
d S)NrL r< r ) r3 r% r= r> r? r r rD r r r )r r@ )r6 r r r% s
zSockSite.start)rD rE )r
r r r+ r" r, r# r% r r )r6 r r q s c @ s e Z d Z d Z d d d d Z e d d
Z e d d Z d
d Z e d d Z
d d Z e d d Z e d d Z
d d Z d d Z d d Z d S) r _handle_signals_kwargsr _siteshandle_signalsFc K s+ | | _ | | _ d | _ t | _ d S)N)rM rN r setrO )r rP kwargsr r r r" s zBaseRunner.__init__c C s | j S)N)r )r r r r r s zBaseRunner.serverc C s
t | j S)N)rQ rO )r r r r sites s zBaseRunner.sitesc sl t j } | j rT y* | j t j t | j t j t Wn t k
rS Yn X| j Id H| _
d S)N) r= r> rM Zadd_signal_handlersignalSIGINTr SIGTERMNotImplementedError _make_serverr )r r@ r r r setup s
zBaseRunner.setupc s d S)Nr )r r r r r) s zBaseRunner.shutdownc s t j } | j d k r d Sx& t | j D] } | j Id Hq/ W| j Id Hd | _ | j r y$ | j t j
| j t j Wn t k
r Yn Xd S)N)
r= r> r listrO r* _cleanup_serverrM Zremove_signal_handlerrT rU rV rW )r r@ siter r r cleanup s
zBaseRunner.cleanupc s d S)Nr )r r r r rX s zBaseRunner._make_serverc s d S)Nr )r r r r r[ s zBaseRunner._cleanup_serverc C s; | | j k r' t d j | | | j j | d S)Nz*Site {} is already registered in runner {})rO r rC add)r r\ r r r r$ s zBaseRunner._reg_sitec C s+ | | j k r' t d j | | d S)Nz&Site {} is not registered in runner {})rO r rC )r r\ r r r r& s zBaseRunner._check_sitec C s; | | j k r' t d j | | | j j | d S)Nz&Site {} is not registered in runner {})rO r rC remove)r r\ r r r r' s zBaseRunner._unreg_siteN)rM rN r rO )r
r r r+ r" r, r rS rY r r) r] rX r[ r$ r& r' r r r r r s
c sX e Z d Z d Z d
Z d d f d d Z d d Z d d
Z d d Z S)r
zLow-level web server runner _web_serverrP Fc s# t j d | | | | _ d S)NrP )r3 r" r` )r Z
web_serverrP rR )r6 r r r" s zServerRunner.__init__c s d S)Nr )r r r r r) s zServerRunner.shutdownc s | j S)N)r` )r r r r rX s zServerRunner._make_serverc s d S)Nr )r r r r r[ s z ServerRunner._cleanup_server)r` ) r
r r __doc__r+ r" r) rX r[ r r )r6 r r
s c sj e Z d Z d Z d Z d d f d d Z e d d Z d d
Z d d Z d
d Z
S)r zWeb Application runner_apprP Fc s# t j d | | | | _ d S)NrP )r3 r" rb )r apprP rR )r6 r r r" s zAppRunner.__init__c C s | j S)N)rb )r r r r rc s z
AppRunner.appc s | j j Id Hd S)N)rb r) )r r r r r) s zAppRunner.shutdownc sd t j } | j j | | j j j | j j Id H| j j | j j d | | j S)Nr@ ) r= r> rb Z _set_loopZ
on_startupZfreezeZstartupZ make_handlerrN )r r@ r r r rX s
zAppRunner._make_serverc s | j j Id Hd S)N)rb r] )r r r r r[
s zAppRunner._cleanup_server)rb ) r
r r ra r+ r" r, rc r) rX r[ r r )r6 r r s )r r r r r r
r )r= rT rI abcr r Zyarlr __all__
SystemExitr r r r r r r r
r r r r r <module> s $! S