3
\F� � @ s� d dl Z d dlZd dlZd dlZd dlZy d dlZW n ek
rL dZY nX dZd dlm Z
d dlm Z dddd d
d d d
gZ dZ
G dd� de�ZG dd� de�ZG dd � d e�ZG dd
� d
e�ZG dd � d e�ZG dd � d e�ZdZdZdddddd d d d d d!d"h Zd#d$d%d&d'd(d)gZd(d)d*�Zd+Zejd,d-d.d/d0g�Z ejd1d2d3d4g�Z d5d
� Z d6d7� Z dkd8d9�Z dld:d;�Z dmd<d=�Z!e�r�d>d?� Z"G d@dA� dA�Z#G dBd� de#�Z$e�r�G dCdD� dDe#�Z%e j&dD� e'dEk�r|d dl(Z(e(j)dFdG�Z*e*j+dHdIdJdKdL� e*j+dMdNdOdPdL� e*j+dQdRdne,dTeef dU� e*j+dVdWdXe,dYdU� e*j+dZd[d\dd]d^� e*j-� Z.e.j/Z/e.j�s�e/dok�rpeZ/e$e.j0e/d_�Z1n e/dpk�r�eZ/e%e.j0e/d_�Z1e1j2� Z3d`e3k�r�e1j4� e1j5e.j5�\Z6Z7Z8Z9Z:e;dae:dbe7dce8dde9� dedf� Z<e=e,e9�e.j> dS �Z8e1j?e8e9�\Z6Z@xbe@D ]Z\ZAZBe eBd$ �jCdgdS�d ZDe eBd# �ZEe,eBd) �ZFe;dhjGeAe<eDdi�e<eEdj�eF�� �qW e1jH� dS )q� NFT)�
decode_header)�_GLOBAL_DEFAULT_TIMEOUT�NNTP� NNTPError�NNTPReplyError�NNTPTemporaryError�NNTPPermanentError�NNTPProtocolError�
NNTPDataErrorr i c @ s e Zd Zdd� ZdS )r c G s> t j| f|�� y|d | _W n tk
r8 d| _Y nX d S )Nr zNo response given)� Exception�__init__�response�
IndexError)�self�args� r � /usr/lib64/python3.6/nntplib.pyr b s
zNNTPError.__init__N)�__name__�
__module__� __qualname__r r r r r r ` s c @ s e Zd ZdS )r N)r r r r r r r r i s c @ s e Zd ZdS )r N)r r r r r r r r m s c @ s e Zd ZdS )r N)r r r r r r r r q s c @ s e Zd ZdS )r N)r r r r r r r r u s c @ s e Zd ZdS )r
N)r r r r r r r r
y s �w i3 Z100Z101�211�215Z220Z221Z222Z224Z225Z230Z231Z282�subject�from�datez
message-idZ
referencesz:bytesz:lines)�bytes�liness
� GroupInfo�group�last�first�flag� ArticleInfoZnumber�
message_idr c C sN g }x>t | �D ]2\}}t|t�r6|j|j|p.d�� q|j|� qW dj|�S )N�ascii� )�_email_decode_header�
isinstancer �append�decode�join)Z
header_str�parts�v�encr r r r � s
c C s� g }xh| D ]`}|d dkr<|dd � j d�\}}}d| }n|j d�\}}}|j� }tj||�}|j|� q
W t}t|�t|�k r�td��|d t|�� |kr�td��|S )Nr �:� z$LIST OVERVIEW.FMT response too shortz*LIST OVERVIEW.FMT redefines default fields)� partition�lower�_OVERVIEW_FMT_ALTERNATIVES�getr) �_DEFAULT_OVERVIEW_FMT�lenr
)r �fmt�line�name�_�suffixZdefaultsr r r �_parse_overview_fmt� s
r< c C s� t t�}g }x�| D ]�}i }|jd�^}}t|�}x�t|�D ]�\} }
| t |�krPq:|| } | jd�} | |kr�| r�| d }
|
r�|
d t |
�� j� |
kr�td��|
r�|
t |
�d � nd }
|
||| < q:W |j||f� qW |S )N� r/ z: z?OVER/XOVER response doesn't include names of additional headers) r6 r5 �split�int� enumerate�
startswithr2 r
r) )r r7 Zdata_process_funcZ
n_defaultsZoverviewr8 ZfieldsZarticle_number�tokens�i�tokenZ
field_nameZ is_metadata�hr r r �_parse_overview� s&
rF c C s� |d kr | dd � }| d d � } t |d d� �}t |dd� �}t |dd � �}t | d d
� �}t | d d � �}t | d
d � �}|dk r�|d7 }n|dk r�|d7 }tj||||||�S )N� � � �F i� �d il i����i�������rL ���rM )r? �datetime)�date_str�time_strZhoursZminutesZseconds�yearZmonthZdayr r r �_parse_datetime� s
rR c C sP t | tj�sd}n
dj| �}| j}|r<|d }dj|| �}n dj|| �}||fS )NZ000000z({0.hour:02d}{0.minute:02d}{0.second:02d}rK z {0:02d}{1.month:02d}{1.day:02d}z {0:04d}{1.month:02d}{1.day:02d})r( rN �formatrQ )ZdtZlegacyrP �yrO r r r �_unparse_datetime� s
rU c C s |d krt j� }|j| |d�S )N)Zserver_hostname)�sslZ_create_stdlib_contextZ wrap_socket)�sock�contextZhostnamer r r � _encrypt_on s rY c @ s e Zd ZdZdZdefdd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z e Z dd� Z
dd� Zdhdd�Zdd� Zdidd�Zdd � Zdjd d �Zdkd d �Zd!d"� Zd#d$� Zd%d&� Zdd'�d(d)�Zdd'�d*d+�Zdldd'�d,d-�Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z dd'�d6d7�Z d8d9� Z d:d;� Z!dmd<d=�Z"d>d?� Z#d@dA� Z$dndBdC�Z%dodd'�dDdE�Z&dpdd'�dFdG�Z'dqdd'�dHdI�Z(dJdK� Z)dd'�dLdM�Z*dd'�dNdO�Z+dd'�dPdQ�Z,dd'�dRdS�Z-dTdU� Z.dVdW� Z/dXdY� Z0dZd[� Z1d\d]� Z2d^d_� Z3d`da� Z4drdbdc�Z5ddde� Z6e7�rdsdfdg�Z8dS )t� _NNTPBasezutf-8�surrogateescapeNc C sj || _ || _d| _| j� | _d | _| j� d| _|rZd| jkrZ| j� | jsZd | _| j� d| _ d| _
d S )Nr F�READER) �host�file� debugging�_getresp�welcome�_caps�getcapabilities�readermode_afterauth�_setreadermode�tls_on�
authenticated)r r^ r] �
readermode�timeoutr r r r 8 s
z_NNTPBase.__init__c C s | S )Nr )r r r r � __enter__g s z_NNTPBase.__enter__c sR � fdd�}|� rNz*y � j � W n ttfk
r8 Y nX W d |� rL� j� X d S )Nc s
t � d�S )Nr^ )�hasattrr )r r r �<lambda>k s z$_NNTPBase.__exit__.<locals>.<lambda>)�quit�OSError�EOFError�_close)r r Z is_connectedr )r r �__exit__j s
z_NNTPBase.__exit__c C s | j rtdt| j�� | jS )Nz *welcome*)r_ �print�reprra )r r r r �
getwelcomeu s z_NNTPBase.getwelcomec
C s� | j d kr�d| _d | _y| j� \}}W n ttfk
rD i | _ Y n<X || _ d|krhttt|d ��| _d|kr�dj |d �| _| j S )Nr0 �VERSIONZIMPLEMENTATION� )
rb � nntp_versionZnntp_implementation� capabilitiesr r �max�mapr? r+ )r �resp�capsr r r rc ~ s
z_NNTPBase.getcapabilitiesc C s
|| _ d S )N)r_ )r �levelr r r �set_debuglevel� s z_NNTPBase.set_debuglevelc C s: |t }| jdkr tdt|�� | jj|� | jj� d S )Nr0 z*put*)�_CRLFr_ rr rs r^ �write�flush)r r8 r r r �_putline� s
z_NNTPBase._putlinec C s2 | j rtdt|�� |j| j| j�}| j|� d S )Nz*cmd*)r_ rr rs �encode�encoding�errorsr� )r r8 r r r �_putcmd� s z_NNTPBase._putcmdTc C s� | j jtd �}t|�tkr$td��| jdkr<tdt|�� |sDt�|r�|dd � t krf|d d� }n |dd � t kr�|d d� }|S ) Nr0 z
line too longz*get*rH rM rM ���r� )
r^ �readline�_MAXLINEr6 r
r_ rr rs ro r )r Z
strip_crlfr8 r r r �_getline� s
z_NNTPBase._getlinec C sl | j � }| jr tdt|�� |j| j| j�}|d d� }|dkrHt|��|dkrXt|��|dkrht |��|S )Nz*resp*r0 �4�5Z123)
r� r_ rr rs r* r� r� r r r )r r{ �cr r r r` � s z_NNTPBase._getrespc
C s� d }z�t |ttf�r"t|d� }}| j� }|d d� tkrBt|��g }|d k r�dt df}xv| jd�}||krpP |j d�r�|dd � }|j
|� q\W n<d}x6| j� }||kr�P |j d�r�|dd � }|j |� q�W W d |r�|j � X ||fS )N�wb� � .s .
Fs ..r0 )
r( �strr �openr` � _LONGRESPr r r� rA r� r) �close)r r^ Z
openedFiler{ r Z terminatorsr8 �
terminatorr r r � _getlongresp� s8
z_NNTPBase._getlongrespc C s | j |� | j� S )N)r� r` )r r8 r r r � _shortcmd� s
z_NNTPBase._shortcmdc C s | j |� | j|�S )N)r� r� )r r8 r^ r r r �_longcmd s
z_NNTPBase._longcmdc s. � j |� � j|�\}}|� fdd�|D �fS )Nc s g | ]}|j � j� j��qS r )r* r� r� )�.0r8 )r r r �
<listcomp> s z,_NNTPBase._longcmdstring.<locals>.<listcomp>)r� r� )r r8 r^ r{ �listr )r r �_longcmdstring s
z_NNTPBase._longcmdstringc C sb y| j S tk
r Y nX y| jd�\}}W n tk
rN td d � }Y n
X t|�}|| _ |S )NzLIST OVERVIEW.FMT)Z_cachedoverviewfmt�AttributeErrorr� r r5 r< )r r{ r r7 r r r �_getoverviewfmt s z_NNTPBase._getoverviewfmtc C s dd� |D �S )Nc S s g | ]}t |j� � �qS r )r r> )r� r8 r r r r� $ s z(_NNTPBase._grouplist.<locals>.<listcomp>r )r r r r r �
_grouplist" s z_NNTPBase._grouplistc C s< i }| j d�\}}x |D ]}|j� ^}}|||< qW ||fS )NZ CAPABILITIES)r� r> )r r| r{ r r8 r9 rB r r r rx & s
z_NNTPBase.capabilities)r^ c C sb t |tjtjf�s$tdj|jj���t|| jdk �\}}dj||�}| j ||�\}}|| j
|�fS )NzAthe date parameter must be a date or datetime object, not '{:40}'rH zNEWGROUPS {0} {1}) r( rN r � TypeErrorrS � __class__r rU rw r� r� )r r r^ rO rP �cmdr{ r r r r � newgroups4 s z_NNTPBase.newgroupsc C sR t |tjtjf�s$tdj|jj���t|| jdk �\}}dj|||�}| j ||�S )NzAthe date parameter must be a date or datetime object, not '{:40}'rH zNEWNEWS {0} {1} {2})
r( rN r r� rS r� r rU rw r� )r r r r^ rO rP r� r r r �newnewsD s z_NNTPBase.newnewsc C s4 |d k rd| }nd}| j ||�\}}|| j|�fS )Nz LIST ACTIVE ZLIST)r� r� )r �
group_patternr^ �commandr{ r r r r r� T s
z_NNTPBase.listc C s� t jd�}| jd| �\}}|jd�s8| jd| �\}}i }x>|D ]6}|j|j� �}|rB|jdd�\} }
|sp|
S |
|| < qBW |r�||fS dS d S )Nz ^(?P<group>[^ ]+)[ ]+(.*)$zLIST NEWSGROUPS r zXGTITLE r0 rH r& )�re�compiler� rA �search�stripr ) r r� Z
return_all�line_patr{ r �groups�raw_line�matchr9 Zdescr r r �_getdescriptionsc s
z_NNTPBase._getdescriptionsc C s | j |d�S )NF)r� )r r r r r � descriptionz s z_NNTPBase.descriptionc C s | j |d�S )NT)r� )r r� r r r � descriptions� s z_NNTPBase.descriptionsc C s� | j d| �}|jd�s t|��|j� }d } }}t|�}|dkr�|d }|dkr�|d }|dkr�|d }|dkr�|d j� }|t|�t|�t|�|fS )NzGROUP r r r0 rH r� rI )r� rA r r> r6 r2 r? )r r9 r{ �words�countr! r �nr r r r � s
z_NNTPBase.groupc C s | j d|�S )NZHELP)r� )r r^ r r r �help� s z_NNTPBase.helpc C s8 |j d�st|��|j� }t|d �}|d }|||fS )NZ22r0 rH )rA r r> r? )r r{ r� �art_numr$ r r r �
_statparse� s
z_NNTPBase._statparsec C s | j |�}| j|�S )N)r� r� )r r8 r{ r r r �_statcmd� s
z_NNTPBase._statcmdc C s" |r| j dj|��S | j d�S d S )NzSTAT {0}ZSTAT)r� rS )r � message_specr r r �stat� s z_NNTPBase.statc C s
| j d�S )NZNEXT)r� )r r r r �next� s z_NNTPBase.nextc C s
| j d�S )NZLAST)r� )r r r r r � s z_NNTPBase.lastc C s0 | j ||�\}}| j|�\}}}|t|||�fS )N)r� r� r# )r r8 r^ r{ r r� r$ r r r �_artcmd� s z_NNTPBase._artcmdc C s$ |d k rdj |�}nd}| j||�S )NzHEAD {0}ZHEAD)rS r� )r r� r^ r� r r r �head� s z_NNTPBase.headc C s$ |d k rdj |�}nd}| j||�S )NzBODY {0}ZBODY)rS r� )r r� r^ r� r r r �body� s z_NNTPBase.bodyc C s$ |d k rdj |�}nd}| j||�S )Nz ARTICLE {0}ZARTICLE)rS r� )r r� r^ r� r r r �article� s z_NNTPBase.articlec C s
| j d�S )NZSLAVE)r� )r r r r �slave s z_NNTPBase.slavec sD t jd�� | jdj||�|�\}}� fdd��|�fdd�|D �fS )Nz^([0-9]+) ?(.*)
?z XHDR {0} {1}c s � j | �}|r|jdd�S | S )Nr0 rH )r� r )r8 �m)�patr r �
remove_number s
z%_NNTPBase.xhdr.<locals>.remove_numberc s g | ] }� |��qS r r )r� r8 )r� r r r� s z"_NNTPBase.xhdr.<locals>.<listcomp>)r� r� r� rS )r Zhdrr� r^ r{ r r )r� r� r �xhdr
s
z_NNTPBase.xhdrc C s. | j dj||�|�\}}| j� }|t||�fS )Nz
XOVER {0}-{1})r� rS r� rF )r �start�endr^ r{ r r7 r r r �xover s
z_NNTPBase.xoverc C sx d| j krdnd}t|ttf�r>|\}}|dj||p6d�7 }n|d k rR|d | }| j||�\}}| j� }|t||�fS )NZOVERZXOVERz {0}-{1}r& rv )rb r( �tupler� rS r� r� rF ) r r� r^ r� r� r� r{ r r7 r r r �over( s z_NNTPBase.overc C sj t jdtd� tjd�}| jd| |�\}}g }x0|D ](}|j|j� �}|r6|j|j dd�� q6W ||fS )NzFThe XGTITLE extension is not actively used, use descriptions() insteadrH z^([^ ]+)[ ]+(.*)$zXGTITLE r0 )
�warnings�warn�DeprecationWarningr� r� r� r� r� r) r ) r r r^ r� r{ Z raw_linesr r� r� r r r �xgtitleC s
z_NNTPBase.xgtitlec C sj t jdtd� | jdj|��}|jd�s0t|��y|j� \}}W n tk
r\ t|��Y n
X ||fS d S )Nz(The XPATH extension is not actively usedrH z XPATH {0}Z223) r� r� r� r� rS rA r r> �
ValueError)r �idr{ Zresp_num�pathr r r �xpathU s
z_NNTPBase.xpathc C sb | j d�}|jd�s t|��|j� }t|�dkr8t|��|d }t|�dkrTt|��|t|d �fS )NZDATEZ111rH r0 � )r� rA r r> r6 r
rR )r r{ �elemr r r r r i s
z_NNTPBase.datec C s� | j |�}|jd�s t|��t|ttf�r2|j� }xB|D ]:}|jt�sT|j d�t }|jd�rfd| }| j
j |� q8W | j
j d� | j
j � | j
� S )N�3s
r� s .
)r� rA r r( r � bytearray�
splitlines�endswithr �rstripr^ r� r� r` )r r� �fr{ r8 r r r �_postz s
z_NNTPBase._postc C s | j d|�S )NZPOST)r� )r �datar r r �post� s z_NNTPBase.postc C s | j dj|�|�S )Nz IHAVE {0})r� rS )r r$ r� r r r �ihave� s z_NNTPBase.ihavec C s | j j� | ` d S )N)r^ r� )r r r r rp � s
z_NNTPBase._closec
C s z| j d�}W d | j� X |S )NZQUIT)r� rp )r r{ r r r rm � s
z_NNTPBase.quitc C s� | j rtd��| r"| r"td��y>|r^| r^dd l}|j� }|j| j�}|r^|d }|d }W n tk
rt Y nX |s~d S | jd| �}|jd�r�|s�t|��n | jd| �}|jd�s�t |��d | _
| j � | j r�d | j
kr�| j
� d | _
| j � d S )
NzAlready logged in.z7At least one of `user` and `usenetrc` must be specifiedr rH zauthinfo user Z381zauthinfo pass Z281r\ )rg r� �netrcZauthenticatorsr] rn r� rA r r rb rc rd re )r �user�password�usenetrcr� Z credentialsZauthr{ r r r �login� s<
z_NNTPBase.loginc C s` y| j d�| _W nJ tk
r$ Y n8 tk
rZ } z |jjd�rHd| _n� W Y d d }~X nX d S )Nz mode readerZ480T)r� ra r r r
rA rd )r �er r r re � s z_NNTPBase._setreadermodec C s| | j rtd��| jr td��| jd�}|jd�rp| jj� t| j|| j �| _| jj
d�| _d| _ d | _ | j � nt
d��d S )NzTLS is already enabled.z+TLS cannot be started after authentication.�STARTTLSZ382�rwbTzTLS failed to start.)rf r� rg r� rA r^ r� rY rW r] �makefilerb rc r )r rX r{ r r r �starttls� s
z_NNTPBase.starttls)T)N)N)N)N)N)N)N)N)N)NNT)N)9r r r r� r� r r rj rq rt rc r~ �debugr� r� r� r` r� r� r� r� r� r� rx 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� rp rm r� re � _have_sslr� r r r r rZ ( sj
.
.
)
rZ c @ s* e Zd Zeddddefdd�Zdd� ZdS )r NFc C s� || _ || _tj||f|�| _d }y8| jjd�}tj| ||||� |sJ|rX| j|||� W n$ |rn|j � | jj � � Y nX d S )Nr� )
r] �port�socket�create_connectionrW r� rZ r r� r� ) r r] r� r� r� rh r� ri r^ r r r r � s
z
NNTP.__init__c
C s zt j| � W d | jj� X d S )N)rZ rp rW r� )r r r r rp ! s z NNTP._close)r r r � NNTP_PORTr r rp r r r r r � s "c @ s, e Zd Zedddddefdd�Zdd� ZdS )�NNTP_SSLNFc
C s� t j||f|�| _d } yJt| j||�| _| jjd�} tj| | |||d� |sP|r^| j|||� W n$ | rt| j� | jj� � Y nX d S )Nr� )rh ri ) r� r� rW rY r� rZ r r� r� )
r r] r� r� r� Z ssl_contextrh r� ri r^ r r r r + s
zNNTP_SSL.__init__c
C s zt j| � W d | jj� X d S )N)rZ rp rW r� )r r r r rp A s zNNTP_SSL._close)r r r �
NNTP_SSL_PORTr r rp r r r r r� ) s
r� �__main__zJ nntplib built-in demo - display the latest articles in a newsgroup)r� z-gz--groupzgmane.comp.python.generalz3group to fetch messages from (default: %(default)s))�defaultr� z-sz--serverznews.gmane.orgz+NNTP server hostname (default: %(default)s)z-pz--portr0 z#NNTP port number (default: %s / %s))r� �typer� z-nz
--nb-articles�
z2number of articles to fetch (default: %(default)s)z-Sz--ssl�
store_truezuse NNTP over SSL)�actionr� r� )r] r� r� ZGroupZhaszarticles, rangeZtoc C s$ t | �|kr | d |d � d } | S )NrI z...)r6 )�sZlimr r r �cutl s r� �<z{:7} {:20} {:42} ({})� �* )N)N)Fr� r� r� )Ir� r� � collectionsrN r� rV � ImportErrorr� Z email.headerr r' r �__all__r� r r r r r r r
r� r� r� r5 r3 r �
namedtupler r# r< rF rR rU rY rZ r r� r) r �argparse�ArgumentParser�parser� add_argumentr? �
parse_argsr r� Zserverr� rc r| r� r r{ r� r! r r9 rr r� r� Z nb_articlesr� Z overviewsZartnumr� r> Zauthorr r rS rm r r r r �<module>B s�
X-