U
     ig                     @   sD   d dl Z d dlmZ d dlmZ d dlmZ e  G dd dZdS )    N)create_engine)sessionmaker)load_dotenvc                   @   s$   e Zd Zdd Zdd Zdd ZdS )Databasec                 C   sJ   t d| _t d| _t d| _t d| _t d| _|  | _d S )NZDB_USERZDB_PASSWORDZDB_HOSTZDB_PORTZDB_NAME)	osgetenvuserpasswordserverportdatabasegetconnectionengineself r   6/var/www/html/context360service/connection/database.py__init__
   s    zDatabase.__init__c                 C   s   t d| j| j| j| jS )Nzmysql+pymysql://{0}:{1}@{2}/{3})r   formatr   r	   r
   r   r   r   r   r   r      s    zDatabase.getconnectionc                 C   s   t | jd}| S )N)bind)r   r   )r   sessionr   r   r   setConnection   s    zDatabase.setConnectionN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s   r   )r   Z
sqlalchemyr   Zsqlalchemy.ormr   dotenvr   r   r   r   r   r   <module>   s
   