from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker


# MYSQL Configs
DB_PWD = "root:123456@localhost"
DB_NAME = "wr_data"

# Sqlalchemy Export
engine = create_engine("mysql+pymysql://{}/{}?charset=utf8mb4".format(DB_PWD, DB_NAME))
Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()


def get_db():
    try:
        db = Session()
        yield db
    finally:
        db.close()