
    Of"                     l    S SK r S SKrS SK7  S SK7  S SK7  S SKJr  S SKJr  S SK	J
r
   " S S\5      rg)    N)*ElementWrapper)map_tag)LazyConcatenationc                   v    \ rS rSrSS jrSS jrSS jrSS jrSS jrSS jr	S	 r
S
 rSS jr\S 5       rSrg)NPSChatCorpusReader   Nc                 <    [         R                  XX#5        X@l        g N)XMLCorpusReader__init___tagset)selfrootfileids
wrap_etreetagsets        =/usr/lib/python3/dist-packages/nltk/corpus/reader/nps_chat.pyr   NPSChatCorpusReader.__init__   s      WA    c           
      "   U R                   (       a@  [        U R                  U5       Vs/ s H  n[        USU R                  5      PM     sn5      $ [        U R                  U5       Vs/ s H  n[        US5      PM     sn5      $ s  snf s  snf )NzSession/Posts/Post)_wrap_etreeconcatabspathsXMLCorpusView	_wrap_eltr   r   fileids      r   	xml_postsNPSChatCorpusReader.xml_posts   s     #'--"8"8 "&*>O"8   #'--"8"8 "&*>?"8 s   !B*Bc           
          [        U R                  U5       Vs/ s H  n[        USU R                  5      PM     sn5      $ s  snf )NSession/Posts/Post/terminals)r   r   r   _elt_to_wordsr   s      r   postsNPSChatCorpusReader.posts(   sQ    
 #mmG4	 5F :D<N<N 5	
 	
s   !Ac           
         ^ ^ U U4S jn[        T R                  U5       Vs/ s H  n[        USU5      PM     sn5      $ s  snf )Nc                 (   > TR                  XT5      $ r   )_elt_to_tagged_words)elthandlerr   r   s     r   reader0NPSChatCorpusReader.tagged_posts.<locals>.reader3   s    ,,S6BBr   r#   )r   r   r   )r   r   r   r,   r   s   ` `  r   tagged_posts NPSChatCorpusReader.tagged_posts2   sM    	C  #mmG44F f&DfM4
 	
s   A c                 6    [        U R                  U5      5      $ r   )r   r%   )r   r   s     r   wordsNPSChatCorpusReader.words=   s     G!455r   c                 6    [        U R                  X5      5      $ r   )r   r.   )r   r   r   s      r   tagged_words NPSChatCorpusReader.tagged_words@   s     !2!27!CDDr   c                     [        U5      $ r   r   )r   r*   r+   s      r   r   NPSChatCorpusReader._wrap_eltC   s    c""r   c                     UR                  S5       Vs/ s H   o0R                  UR                  S   5      PM"     sn$ s  snf )Ntword)findall_simplify_usernameattrib)r   r*   r+   r9   s       r   r$   !NPSChatCorpusReader._elt_to_wordsF   s7    CF;;sCSTCSa''(89CSTTTs   '>c           
      8   UR                  S5       Vs/ s H0  nU R                  UR                  S   5      UR                  S   4PM2     nnU(       a;  X0R                  :w  a,  U VVs/ s H  u  pdU[	        U R                  X45      4PM     nnnU$ s  snf s  snnf )Nr9   r:   pos)r;   r<   r=   r   r   )r   r*   r+   r   r9   tagged_postws          r   r)   (NPSChatCorpusReader._elt_to_tagged_wordsI   s     [[%
% $$QXXf%56H% 	 
 f,DODO&1GDLL&45K   

s   7B($Bc                     SU ;   a  SU R                  SS5      S   -   n U $ [        U [        5      (       a  U R                  S5      n U $ )NUserU   ascii)split
isinstancebytesdecode)r:   s    r   r<   &NPSChatCorpusReader._simplify_usernameT   sL    T>FA.q11D  e$$;;w'Dr   )r   )FNr   )NN)__name__
__module____qualname____firstlineno__r   r    r%   r.   r1   r4   r   r$   r)   staticmethodr<   __static_attributes__ r   r   r	   r	      sE     
	
6E#U	  r   r	   )retextwrapnltk.corpus.reader.apinltk.corpus.reader.utilnltk.corpus.reader.xmldocsnltk.internalsr   nltk.tagr   	nltk.utilr   r   r	   rT   r   r   <module>r]      s-    
  $ % ( )  'G/ Gr   