
    xi;                     `    S r SSKrSSKrSSKrSSKrSSKJr  SSKJr  SSK	J
r
   " S S5      rg)ztrain for plumis    N)Fitli)Vemli)Runerc                   0    \ rS rSrSS jrS rS rS rSrg)	Trumi   c                     Xl         [        U5      U l        [        U5      U l        U R                  R                  5       U l        X l        SU l        X0l	        SSS.U l
        S U l        0 U l        g )Nnr   )minuplus)er   vemlir   fitli	train_sigsig
do_verbose
is_derivedno_pillsnatfitsseen_papids)selferimpr   r   s       !/home/ernad/ernad/python/trumi.py__init__Trumi.__init__   sb    5\
5\
::'')$   !1-
    c                     SnSnX0l         U HJ  nUc  UnUS-  S:  a  [        U S3SS9  O[        U 5        US-  nX&   U l        U R                  Xa5      nML     US-  S:w  a  [        S	5        U R                   $ )
zfeeds plumis into train_filer   N        )end   
)r   print
plumi_fufi
feed_issue)r   
train_fileplumi_dafusr   count_takenfirst_issuedate	issuedates          r   write_trainTrumi.write_train(   s    &$I&"+Q"1o2.%1K)4DO//)@K % ?a$Kr   c                    U R                   nU R                  (       a  [        SU-   5        SU R                  -   S-   nUR                  S   S-   U-   U-   n[
        R                  R                  U5      (       dj  [        SU SU R                   S3S	-   [        R                  S
9  UR                  5         [        S[        R                  S
9  [        R                  " S5        [        R                  " U R                  S5      nSU l        SU R                  ;   a  SU l        UR!                  5       n["        R$                  " U5      U l        SnU H  n	U	R)                  5       R+                  S5      (       a  U R,                  S==   S-  ss'   OzU	R)                  5       R+                  S5      (       a  U R,                  S==   S-  ss'   O>[        SU	R)                  5       -   [        R                  S
9  [        R                  " 5         XR/                  U	R)                  5       R1                  5       U5      -  nM     UR3                  U5        U R,                  $ )zfeeds issue into train fileztrumi: feed_issue adds _z.json.gzr   /ztrumi does not see z of sig r    z' this should have done by post_process.fileztrain ends.r#   rr
   z/forpi/ir!   +r   -r   ztrumi: bad plumi line )r   r   r%   r   dirsospathisfilesysstderrcloseexitgzipopenr&   r   	readlinesfilerloadr   decode
startswithr   feed_plumi_linerstripwrite)
r   r,   r(   r   r"   
vemli_fufi
plumi_fileplumi_linesvemstlines
             r   r'   Trumi.feed_issue<   s    FF??+i78DHHnz)VVG_s*Y6<
ww~~j))'
|8DHH:QG;<zz# -cjj1HHQKYYt4
'!DO **,ZZ
+
D{{}'',,V$)$))#..V$)$.>::'
))$++-*>*>*@)LLE   	||r   c                     U R                   nUSS nUSS  nXS;  a  [        US-   U-   [        R                  S9  g U R                  U R
                  U'   US-   X5   -   S-   nXbS-   U R                  -   S-   -  nU$ )Nr         z is not in the vemli of r2   r    r$   )r   r%   r<   r=   r   r   )r   rN   r,   vemlisindicpapidrM   s          r   rG   Trumi.feed_plumi_linee   s    Qq	QR %44y@zz#"&//fm+c1S4??2T99r   )
r   r   r   r   r   r   r&   r   r   r   N)FF)	__name__
__module____qualname____firstlineno__r   r-   r'   rG   __static_attributes__ r   r   r   r      s    & ('Rr   r   )__doc__r@   r9   r<   rC   r   r   r   r   runerr   r   r\   r   r   <module>r_      s/      	 

    _ _r   