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

安装

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