当前位置 博文首页 > Python流行ORM框架sqlalchemy的简单使用

    Python流行ORM框架sqlalchemy的简单使用

    作者:吾心依旧 时间:2021-08-04 17:47

    安装

    http://docs.sqlalchemy.org

    1、安装

    #进入虚拟环境
    #执行
    ./python3 -m pip install
    
    import sqlalchemy
    print(sqlalchemy.__version__) # 1.1.15
    我这里使用的版本是1.1.15

    创建连接对象

    http://docs.sqlalchemy.org/en/latest/orm/tutorial.html#connecting

    from sqlalchemy import create_engine
    # 连接本地test数据库
    engine = create_engine("mysql://root:root@localhost/test?charset=utf8")

    运行时会出错,因为需要驱动库,默认会调用MySQLdb。

    ImportError: No module named 'MySQLdb'

    我们前面安装了pymysql,因此完整的要这么写:

    engine = create_engine("mysql+pymysql://root:root@localhost/test?charset=utf8")

    简单使用

    SQL语句查询

    result = engine.execute("select * from news")
    print(result.fetchall())
    #[(1, '本机新闻标题'), (2, '今天的新闻'), (3, '新闻标题1'), (4, '新闻标题2'), (5, '元组新闻1'), (6, '元组新闻2')]

    创建映射

    既然我们用ORM,就是为了少写甚至不写SQL语句。

    ORM是数据表和对象之间的映射。

    http://docs.sqlalchemy.org/en/latest/orm/tutorial.html#declare-a-mapping

    1、创建一个Infos.py文件,这个文件我们来做数据表的映射

    from sqlalchemy.ext.declarative import declarative_base
    Base = declarative_base()
    from sqlalchemy import Column, Integer, String
    class News(Base):
    # 表名称
    __tablename__ = 'news'
    # news表里id字段
    id = Column(Integer, primary_key=True, autoincrement=True)
    # news表里title字段
    title = Column(String(length=255), nullable=False)

    News类就是我们数据表news的映射(字段:id、title)。

    2、使用

    from sqlalchemy import create_engine
    from mappers.Infos import News
    from sqlalchemy.orm import sessionmaker
    # 连接本地test数据库
    engine = create_engine("mysql+pymysql://root:root@localhost/test?charset=utf8")
    # 创建会话
    session = sessionmaker(engine)
    mySession = session()
    # 查询结果集
    result = mySession.query(News).all()
    print(result[0])
    

    我们要注意最后的查询结果,看看结果集中的元素长什么样?^_^

    <mappers.Infos.News object at 0x1050c6e80>

    查询处理的记录都是对象。

    各种查询

    只查询第一条记录

    # 查询第一条
    result = mySession.query(News).first()
    print(result.title) #打印对象属性
    通过id字段查询
    
    # 查询id为2的
    result = mySession.query(News).filter_by(id=2).first()
    print(result.title)
    # 查询id为2的
    result = mySession.query(News).filter(News.id==2).first()
    分页查询
    
    # 分页查询 0,2
    result = mySession.query(News).filter(News.id>1).limit(2).offset(0).all()
    print(result)
    自定义过滤条件
    
    # 自定义过滤条件
    result = mySession.query(News).filter(text("id>:id")).params(id=2).all()

    根据主键查询

    result = mySession.query(News).get(3)
    print(result.title)

    新增和修改

    # 新增
    news = News(title="新增测试标题")
    mySession.add(news)
    mySession.commit()
    #修改
    mySession.query(News).filter(News.id==7).update({"title":"修改之后的标题"})
    mySession.commit()

    Python利用sqlacodegen自动生成ORM实体类示例

    前面方法我们是手动创建了一个名叫Infos.py的文件,然后定义了一个News类,把这个类作为和我们news数据表的映射。

    from sqlalchemy.ext.declarative import declarative_base
    Base = declarative_base()
    from sqlalchemy import Column, Integer, String
    class News(Base):
    # 表名称
    __tablename__ = 'news'
    # news表里id字段
    id = Column(Integer, primary_key=True, autoincrement=True)
    # news表里title字段
    title = Column(String(length=255), nullable=False)
    

    现在我们来看看sqlacodegen这个工具,自动生成像上面那样的类文件。

    1、安装sqlacodegen

    #cd 项目虚拟环境
    #执行
    ./python3 -m pip install sqlacodegen

    2、使用sqlacodegen生成案列

    #注意还是在虚拟环境目录下执行
    ./sqlacodegen --tables fund --outfile ../../mappers/Found.py mysql+pymysql://root:root@localhost/test?charset=utf8
    jsjbwy
    下一篇:没有了