File: //proc/thread-self/root/lib/python3/dist-packages/twisted/web/__pycache__/distrib.cpython-38.pyc
U
W[e. � @ s< d Z ddlZddlZz ddlZW n ek
r8 dZY nX ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZmZmZmZ ddlmZ dd lmZ dd
lmZmZ dd lmZ G d d
� d
e
j �Z G dd� de
j ej �Z e
� ej e � G dd� d�Z!G dd� dej"�Z#G dd� de
j$ej%�Z&G dd� dej"�Z'dS )z�
Distributed web servers.
This is going to have to be refactored so that argument parsing is done
by each subprocess and not by the main web server (i.e. GET, POST etc.).
� N)�BytesIO)�getDOMImplementation)�pb)�
SIZE_LIMIT)�http�resource�server�util�static)�Headers)�styles)�address�reactor)�Loggerc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� _ReferenceableProducerWrapperc C s
|| _ d S �N)�producer)�selfr � r �5/usr/lib/python3/dist-packages/twisted/web/distrib.py�__init__! s z&_ReferenceableProducerWrapper.__init__c C s | j �� d S r )r ZresumeProducing�r r r r �remote_resumeProducing$ s z4_ReferenceableProducerWrapper.remote_resumeProducingc C s | j �� d S r )r ZpauseProducingr r r r �remote_pauseProducing' s z3_ReferenceableProducerWrapper.remote_pauseProducingc C s | j �� d S r )r Z
stopProducingr r r r �remote_stopProducing* s z2_ReferenceableProducerWrapper.remote_stopProducingN)�__name__�
__module__� __qualname__r r r r r r r r r s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�Requestzn
A request which was received by a L{ResourceSubscription} and sent via
PB to a distributed node.
c s� dD ]H}|| �ddd��d � � �fdd��fdd�d �� � }|||< qt t|d
��|d
< tj�| |� t| j�| _| j� d �| _
| j� d �| _ | j� d
�| _ | j� d�| _
| j� d�| _| j� d�| _| j� d�| _dS )z�
Initialize this L{twisted.web.distrib.Request} based on the copied
state so that it closely resembles a L{twisted.web.server.Request}.
)�hostZclient�TCP�UNIX)ZINETr! r c s t �� �d �d �S )N� � )r
Z IPv4Addressr �Zaddrdesc�tupr r �<lambda>; s �z*Request.setCopyableState.<locals>.<lambda>c s t �� d �S )Nr" )r
Z UNIXAddressr )r% r r r& = � )r r! ZrequestHeaders�finish� setHeader� addCookie�setETag�setResponseCode�setLastModified�writeN)r �dictr �
RemoteCopy�setCopyableStater Z content_dataZcontent�remoteZ remoteMethodr( r) r* r+ r, r- �_write)r �state�kZaddrr r$ r r1 3 s&
��
zRequest.setCopyableStatec C s> d}t }| �|||� � |t 7 }|t 7 }|t|�krq:qdS )z�
Write the given bytes to the response body.
@param bytes: The bytes to write. If this is longer than 640k, it
will be split up into smaller pieces.
r N)r r3 �len)r �bytes�start�endr r r r. O s z
Request.writec C s | j �dt|�|��| j� d S )N�registerProducer)r2 �
callRemoter �
addErrback�fail)r r Z streamingr r r r: ` s ��zRequest.registerProducerc C s | j �d��| j� d S )N�unregisterProducer)r2 r; r< r= r r r r r> e s zRequest.unregisterProducerc C s | j jd|d� d S )N� )�failure)�_logr@ �r r@ r r r r= h s z Request.failN) r r r �__doc__r1 r. r: r> r= r r r r r . s r c @ s* e Zd Ze� Zdd� Zdd� Zdd� ZdS )�Issuec C s
|| _ d S r )�request)r rE r r r r q s zIssue.__init__c C s6 |t jkr2t|t�s td��| j�|� | j�� d S )Nzreturn value not a string)r � NOT_DONE_YET�
isinstance�str�AssertionErrorrE r. r( )r �resultr r r �finishedt s
zIssue.finishedc
C sL t |�}| j�t�tjddt�|� �� | j�� | j�
� | j � |� d S )NzServer Connection Lostz&Connection to distributed server lost:)
rH rE r. r Z ErrorPager ZINTERNAL_SERVER_ERRORr Z_PRE�renderr( rA �inforB r r r �failedz s ����
z Issue.failedN)r r r r rA r rK rN r r r r rD n s rD c @ sJ e Zd ZdZdZe� Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�ResourceSubscriptionr" r c C s( t j�| � || _|| _g | _d | _d S r )r �Resourcer r �port�pending� publisher)r r rQ r r r r � s
z ResourceSubscription.__init__c C s( t � | j�}d|d< d|d< g |d< |S )z<Get persistent state for this ResourceSubscription.
NrS r �waitingrR )�copy�__dict__)r r4 r r r � __getstate__� s
z!ResourceSubscription.__getstate__c C sF | j �d� |j�| j� || _d| _| jD ]}| �|� q,g | _dS )zFI've connected to a publisher; I'll now send all my requests.
zconnected to publisherr N) rA rM ZbrokerZnotifyOnDisconnect�bootedrS rT rR rL )r rS rE r r r � connected� s
z ResourceSubscription.connectedc C sD | j jd|d� d| _d| _| jD ]}|�d� |�� q"g | _dS )zXI can't connect to a publisher; I'll now reply to all pending
requests.
z3could not connect to distributed web service: {msg})�msgr Nz(Unable to connect to distributed server.)rA rM rT rS rR r. r( )r rZ rE r r r � notConnected� s �
z!ResourceSubscription.notConnectedc C s | � d� d S )Nzconnection dropped)r[ r r r r rX � s zResourceSubscription.bootedc C s� | j st| j�|� | js�d| _t�� }d}| jdkrFt�| j ||� nt�
| j| j ||� |� � }|� | j
| j� n"t|�}| j �d|�� |j|j� tjS )a: Render this request, from my server.
This will always be asynchronous, and therefore return NOT_DONE_YET.
It spins off a request to the pb client, and either adds it to the list
of pending issues or requests it immediately, depending on if the
client is already connected.
r" �
�unixrE )rS rR �appendrT r ZPBClientFactoryr r Z connectUNIXrQ Z
connectTCPZ
getRootObjectZ addCallbacksrY r[ rD r; rK rN r rF )r rE ZbfZtimeout�d�ir r r rL � s
zResourceSubscription.renderN)
r r r ZisLeafrT r rA r rW rY r[ rX rL r r r r rO � s
rO c @ s: e Zd ZdZe� Zdd� ZdZdd� Zdd� Z d d
� Z
d S ) �ResourcePublisherz�
L{ResourcePublisher} exposes a remote API which can be used to respond
to request.
@ivar site: The site which will be used for resource lookup.
@type site: L{twisted.web.server.Site}
c C s
|| _ d S r )�site)r rb r r r r � s zResourcePublisher.__init__r# c C s* | j j�d� | j j| j= | `| ` | `d S )NZweb)Z applicationZ
authorizerZremoveIdentity�servicesZ serviceNameZperspectiveNamer r r r �upgradeToVersion2� s
z#ResourcePublisher.upgradeToVersion2c C s | S r r )r �namer r r �getPerspectiveNamed� s z%ResourcePublisher.getPerspectiveNamedc C sD | j �|�}| j�|� |�|�}|tjk r>|�|� |�� tjS )zK
Look up the resource for the given request and render it.
) rb ZgetResourceForrA rM rL r rF r. r( )r rE �resrJ r r r �remote_request� s
z ResourcePublisher.remote_requestN) r r r rC r rA r ZpersistenceVersionrd rf rh r r r r ra � s ra c @ s> e Zd ZdZdZdZdZddd�Zdd � Zd
d � Z d d
� Z
dS )�
UserDirectoryz�
A resource which lists available user resources and serves them as
children.
@ivar _pwd: An object like L{pwd} which is used to enumerate users and
their home directories.
Z public_htmlz.twistd-web-pba�
<html>
<head>
<title>twisted.web.distrib.UserDirectory</title>
<style>
a
{
font-family: Lucida, Verdana, Helvetica, Arial, sans-serif;
color: #369;
text-decoration: none;
}
th
{
font-family: Lucida, Verdana, Helvetica, Arial, sans-serif;
font-weight: bold;
text-decoration: none;
text-align: left;
}
pre, code
{
font-family: "Courier New", Courier, monospace;
}
p, body, td, ol, ul, menu, blockquote, div
{
font-family: Lucida, Verdana, Helvetica, Arial, sans-serif;
color: #000;
}
</style>
</head>
<body>
<h1>twisted.web.distrib.UserDirectory</h1>
%(users)s
</body>
</html>
Nc C s" t j�| � |d krt}|| _d S r )r rP r �pwd�_pwd)r Z userDatabaser r r r 7 s zUserDirectory.__init__c
C s� g }| j �� D ]�}|\}}}}}}} |�d�d }
|
s:|}
tj�tj�|| j��rd|�||
d f� tj�|| j �} tj�| �r|d } |�| |
d f� q|S )z|
Return a list of two-tuples giving links to user resources and text to
associate with those links.
�,r z (file)�.twistdz (twistd))
rk Zgetpwall�split�os�path�exists�join� userDirNamer^ �userSocketName)
r �users�userre ZpasswdZuid�gidZgecos�dir�shellZrealname�
twistdsockZlinkNamer r r �_users> s zUserDirectory._usersc C s� t � }|�ddd�}|j}| �� D ]T\}}|�d�}|�d|d � |�|�}|�|� |�d�} | �|� |�| � q"| jd|� � i }
|
�
d�S ) zm
Render as HTML a listing of all known users with links to their
personal resources.
NZul�aZhref�/Zliru zutf-8) r ZcreateDocumentZdocumentElementr{ Z
createElementZ setAttributeZcreateTextNodeZ appendChild�templateZtoxml�encode) r rE ZdomImplZnewDocZlisting�link�textZ linkElementZtextNode�itemZhtmlDocr r r �
render_GETR s
zUserDirectory.render_GETc C s� |dkr | S d}|t |� d � |kr>|d t |� � }d}n|}d}z | j�|�\}}}} }
} } W n tk
r� t�� Y S X |r�tj�| | j �}
t
d|
�}| � ||� |S tj�| | j �}tj�
|�s�t�� S t�|�S d S )Nr? rm r" r r] )r6 rk �getpwnam�KeyErrorr Z
NoResourcero rp rr rt rO ZputChildrs rq r
ZFile)r re rE ZtdZusername�subZpw_nameZ pw_passwdZpw_uidZpw_gidZpw_gecos�pw_dirZpw_shellrz Zrsrp r r r �getChildh s,
�
zUserDirectory.getChild)N) r r r rC rs rt r~ r r{ r� r� r r r r ri s *
ri )(rC ro rU rj � ImportError�ior Zxml.dom.minidomr Ztwisted.spreadr Ztwisted.spread.bananar Z twisted.webr r r r r
Ztwisted.web.http_headersr Ztwisted.persistedr Ztwisted.internetr
r Ztwisted.loggerr Z
Referenceabler r0 r ZsetUnjellyableForClassrD rP rO ZRootZ Versionedra ri r r r r �<module> s*
>R(