sqlalchemy 使用记录

有些东西 用了就回不去了…
得益于 django 优秀的 orm , 已经完全不想手撸 sql 了…

in flask

自动生成 flask models

安装 flask-sqlacodegen 后 添加 --flask 参数
sqlacodegen --noviews --noconstraints --noindexes --flask --schema api_moniting --tables apis --outfile mod els.py mysql+pymysql://bird:qpzm2745@127.0.0.1:3306/

初始化

1
2
3
4
5
6
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)
app.config.from_pyfile('hello.cfg')
db = SQLAlchemy(app)

hello.cfg

1
2
3
4
SQLALCHEMY_DATABASE_URI = 'sqlite:////tmp/test.db'
SQLALCHEMY_ECHO = False
SECRET_KEY = '\xfb\x12\xdf\xa1@i\xd6>V\xc0\xbb\x8fp\x16#Z\x0b\x81\xeb\x16'
DEBUG = True

也可以使用 app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True 方法配置 app

操作

QuerySet API 可参考 django 虽然并不相同 but 官方文档 寂寞的一逼…
官链 神仙文档… 神仙文档…

1
2
3
new = Models(column1='test1',column2='test2')
db.session.add(new)
db.session.commit()

pass

1
2
3
modify_data = Models.query.filter_by(column1='test1').first()
modify_data.column1 = 'new'
db.session.commit()

1
2
select_data = Models.query.filter_by(column1='test1').all()
print(select_data.column1)

官方 github

todo

  • db.session.close() 方法要研究一下

scripts

QuerySet API 可参考 django 虽然并不相同 but 官方文档 寂寞的一逼…
官链 神仙文档… 神仙文档…

python3 已经放弃MySQLdb, 使用create_engine('mysql://bird:qpzm2745@127.0.0.1/demo?charset=utf8', echo=True)
会报 No module named 'MySQLdb' 的错误, 可怜天真的我还尝试安装这个包…

code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from sqlalchemy import create_engine
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# 创建连接
engine = create_engine('mysql+pymysql://bird:qpzm2745@127.0.0.1/demo?charset=utf8', echo=True)
# 创建元类
Base = declarative_base()
# 连接池
Session = sessionmaker(bind=engine)
# 实例化
session = Session()


class Phone(Base):
# 表名
__tablename__ = 'Phone_test'
id = Column(Integer(), autoincrement=True, primary_key=True, unique=True)
item_id = Column(String(10))
name = Column(String(120))
crawl_time = Column(String(100))

# 查
test1 = session.query(Phone).all()
print('test1 --> {}'.format(test1))
# 增
session.add(Phone(item_id='item_id_2',name='name_2',crawl_time='crawl_time_2'))
session.commit()
session.close()
# 改
session.query(User).filter(User.username == 'abc').update({'name': '123'})
session.commit()

user = session.query(User).filter_by(username='abc').scalar()
user.name = '223'
session.commit()

Q&A

SQLAlchemy Nested rollback error

sqlalchemy.exc.InvalidRequestError: This Session's transaction has been rolled back by a nested rollback() call. To begin a new transaction, issue Session.rollback() first.
A:
stackoverflow

1
2
3
4
5
6
7
8
try:
<use session>
session.commit()
except:
session.rollback()
raise
finally:
session.close() # optional, depends on use case