���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)�ABC�abstractmethod)�URL�TCPSite�UnixSite�SockSite�
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�_server�shutdown_timeoutg N@� ssl_contextN�backlog� 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 )�self�runnerr 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_site�closeZ wait_closed�shutdownr 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! �host�portr 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)N�https�http�schemer4 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)N�sslr r1 r2 )r3 r% �asyncio�get_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 �hasattr�socket�familyrF 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 �_sites�handle_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_handler�signal�SIGINTr �SIGTERM�NotImplementedError� _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