U
    Eaw                     @   s   d dl mZ d dlmZmZ d dlZdZdZe Z	ej
rVejejdddd	Ze Z	n"ejeeedddd	Ze Z	d
d ZeedddZdS )    )settings)	test_datas3Nztaskcall-prod-dataz#string_universes/label_universe.txtlabelT")Z	index_colZskipinitialspaceZ	quotecharc                 C   sb   zt | |  W S  tk
r\   | dd} | t | krJt | |   Y S t tj |   Y S X d S )N'r   )label_storeKeyErrorstripr   LANGUAGE_CODE)r   lang r   9/var/www/html/taskcallweb/translators/label_translator.py	get_label   s    r   )labelsr   c                 C   s:   |t krtj}t }| D ]}t | | ||< q||d< |S )Nr   )r   r   r   dict)r   r   contextitemr   r   r   get_context!   s    r   )taskcallwebr   utilsr   r   ZpandasZuniv_bucketZ	univ_pathr   r   	TEST_MODEZread_csvZlabel_file_locationpdZto_dictZread_bytes_ior   liststrr   r   r   r   r   <module>   s   
 
