3
.th                 @   sP   d dl Z d dlZd dlZd dlZddlmZmZm	Z	m
Z
mZ dd Zdd ZdS )    N   )	db_dbnamedb_userdb_passworddb_hostdb_portc          
   C   sX   |dkr(t | ddd
}|j S Q R X n,t | d}dtj|j jd S Q R X dS )z7Read and convert image files to base64-encoded strings.svgrzutf-8)encodingNrbzdata:image/jpeg;base64,)openreadbase64	b64encodedecode)	file_path	file_typefile r   6/srv/www/htdocs/dashboard-catalog-gnss/utils/common.pyreplace_image   s
    r   c             C   s   zyPt jtttttd}|j }|j|  |j	 }dd |j
D }tj||d}|S  t jk
r } ztjd|  d S d }~X nX W d |r|j  |r|j  X d S )N)Zdbnameuserpasswordhostportc             S   s   g | ]}|d  qS )r   r   ).0descr   r   r   
<listcomp>   s    zload_data.<locals>.<listcomp>)columnsz Error connecting to PostgreSQL: )psycopg2connectr   r   r   r   r   cursorexecutefetchalldescriptionpd	DataFrameErrorsterrorclose)query
connectionr!   recordsr   dfer   r   r   	load_data   s(    
r0   )r   r   	streamlitr(   pandasr%   credentialsr   r   r   r   r   r   r0   r   r   r   r   <module>   s   	