U g|@sddlZGdddZdS)Nc@sTeZdZdZddZeddZejddZeddZejd dZd d Z d S) CWMS_Sitesz1 Store CWMS site information c Cs||_t|_t|dd}t|}d}|D]v}|dkrTtdd||d7}|d|jkr~|d|j|d|d <nt|d |di|j|d<|d7}q.td t|d |j|_ W5QRXd S) z_ Initialize the CWMS_Sites object with a given filename r)moderzColumn names are z, officeZ usace_gage_idZgagez Processed z lines.N) sourcedict_office_name1_to_indexopencsvZ DictReaderprintjoinstroffice_name1_to_index)selfZ csvSitefileZ csvsite_fileZcsvsite_readerZ line_countrowrM/lfs/h1/ops/prod/packages/nwm.v3.0.13/ush/ace_download/analysis/CWMS_Sites.py__init__s.   zCWMS_Sites.__init__cCs|jSN_sourcerrrrr7szCWMS_Sites.sourcecCs ||_dSrr)rsrrrr;scCs|jSrr rrrrr@sz CWMS_Sites.office_name1_to_indexcCs ||_dSrr)rorrrrDscCsd||jks ||j|krBtd|d|d|jdtd|d||j||S)NzCannot find station for  z in the site file: .zFound station: )rkeysKeyErrorrr )rrZname1rrrgetIndexHs zCWMS_Sites.getIndexN) __name__ __module__ __qualname____doc__rpropertyrsetterrr rrrrrs    r)r rrrrrs