import datetime
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime, UniqueConstraint, Index


Base = declarative_base()


class Users(Base):
    __tablename__ = 'users'  # 数据库表名
    id = Column(Integer, primary_key=True)  # id 主键
    name = Column(String(32), index=True, nullable=False)  # name 不为空 唯一 创建索引


def init_db():
    # 创建表
    engine = create_engine(
        'mysql+pymysql://root:199199@127.0.0.1:3306/blog?charset=utf8',
        max_overflow=2,  # 超过连接池大小外最多创建的数量,
        pool_size=5,  # 连接池的大小
        pool_timeout=30,  # 池中没有线程最多等待的时间
        pool_recycle=-1,  # 多久之后对线程中的线程进行一次连接的回收(重置)

    )

    Base.metadata.create_all(engine)


def drop_db():
    """
    根据类删除数据库表
    """
    engine = create_engine(
        'mysql+pymysql://root:199199@127.0.0.1:3306/blog',
        max_overflow=2,  # 超过连接池大小外最多创建的数量,
        pool_size=5,  # 连接池的大小
        pool_timeout=30,  # 池中没有线程最多等待的时间
        pool_recycle=-1,  # 多久之后对线程中的线程进行一次连接的回收(重置)

    )
    Base.metadata.drop_all(engine)


if __name__ == '__main__':
    init_db()
    # drop_db()