pychatbot / repository /ConfigDatabase.py
kltn20133118's picture
Update repository/ConfigDatabase.py
7f8ee40 verified
raw
history blame
1.37 kB
from sqlalchemy import create_engine, URL
from sqlalchemy.orm import DeclarativeBase
Base = DeclarativeBase()
from sqlalchemy.engine import create_engine, URL
from dotenv import load_dotenv
import os
load_dotenv()
MYSQL_USER_NAME=os.getenv('MYSQL_USER_NAME')
MYSQL_PASSWORD=os.getenv('MYSQL_PASSWOR')
MYSQL_PORT=os.getenv('MYSQL_PORT')
MYSQL_DATABASE=os.getenv('MYSQL_DATABASE')
MYSQL_HOST=os.getenv('MYSQL_HOST')
#IF USE DOCKER HOST = host.docker.internal
def get_db_engine():
dsn = URL.create(
drivername="mysql+pymysql",
username=MYSQL_USER_NAME,
password=MYSQL_PASSWORD,
host=MYSQL_HOST,
port=MYSQL_PORT,
database=MYSQL_DATABASE
)
connect_args = {
"ssl_verify_cert": True,
"ssl_verify_identity": True,
"ssl_ca": 'cert.pem',
}
return create_engine(
dsn,
connect_args=connect_args,
)
def get_db_engine1():
dsn = URL.create(
drivername="mysql+pymysql",
username=MYSQL_USER_NAME,
password=MYSQL_PASSWORD,
host=MYSQL_HOST,
port=MYSQL_PORT,
database=MYSQL_DATABASE
)
connect_args = {
"ssl_verify_cert": True,
"ssl_verify_identity": True,
"ssl_ca": 'cert.pem',
}
return create_engine(
dsn,
connect_args=connect_args,
)