SQLAlchemy操作数据库

from sqlalchemy import create_engine
# 导入引擎


# 使用原生SQL 获取新创建数据库的信息
def conn_db1():
    # 数据库的变量
    # 主机地址：若主机不在本机上，可以写地址
    HOST = 'localhost'
    PORT = 3306
    # 操作哪个数据库
    DATA_BASE = 'flask_db'
    USER = 'root'
    PWD = 'root'

    # 生成URL
    DB_URI = f'mysql+pymysql://{USER}:{PWD}@{HOST}:{PORT}/{DATA_BASE}'
    # 链接数据库引擎
    engine = create_engine(DB_URI)
    # 执行一个SQL
    sql = 'select 1'    #[(2,)]
    #链接数据库
    conn = engine.connect()
    # 执行sql，返回结果给rs
    rs = conn.execute(sql)
    print(rs.fetchall())

# 使用原生sql 创建数据库的表
def conn_db2():
    HOST = 'localhost'
    PORT = 3306
    DATA_BASE = 'flask_db'
    USER = "root"
    PWD = 'root'
    DB_URL = f'mysql+pymysql://{USER}:{PWD}@{HOST}:{PORT}/{DATA_BASE}'
    engine = create_engine(DB_URL)

    sql = 'create table t_user(id int primary key auto_increment, name varchar(32))'

    conn = engine.connect()
    conn.execute(sql)

# 由于链接了数据库，但是没有断开，本次测试用于测试断开
def conn_db3():
    HOST = 'localhost'
    PORT = 3306
    DATA_BASE = 'flask_db'
    USER = "root"
    PWD = 'root'
    DB_URL = f'mysql+pymysql://{USER}:{PWD}@{HOST}:{PORT}/{DATA_BASE}'
    engine = create_engine(DB_URL)

    sql = 'create table t_user1(id int primary key auto_increment, name varchar(32))'

    with engine.connect() as conn:
        conn.execute(sql)


if __name__=="__main__":
    conn_db3()