python 之路 13 ORM SQLAlchemy
1. 安装
# pip install sqlalchemy
# pip install pymysql
2. 创建引擎
>>> from sqlalchemy import create_engine
>>> engine = create_engine('mysql+pymysql://USERNAME:PASSWORD@DB_HOST:DB_PORT/DB_NAME', pool_recycle=3600)
pool_recycle 指定连接池收回时间。
引擎初始化后，就可以连接数据库了。
3. 连接数据库
>>> connection = engine.connect()
4. metadata
>>> from sqlalchemy import MetaData
>>> metadata = MetaData()
5. 定义表
>>> from sqlalchemy import Table, Column, Integer, Numeric, String, ForeignKey, DateTime
>>> from datetime import datetime
>>> users = Table('users', metadata,
... Column('user_id', Integer(), primary_key=True),
... Column('username', String(15), nullable=False, unique=True),
... Column('email_address', String(255), nullable=False),
... Column('phone', String(20), nullable=False),
... Column('password', String(25), nullable=False),
... Column('created_on', DateTime(), default=datetime.now),
... Column('updated_on', DateTime(), default=datetime.now, onupdate=datetime.now)
... )
6. 创建表
>>> metadata.create_all(engine)
7. 插入数据
>>> ins = users.insert().values(
... user_id=8888,
... username='111cn.net ',
... email_address = 'support@111cn.net ',
... phone = 12345678901,
... password = 'www.111cn.net '
... )
>>> print ins
INSERT INTO users (user_id, username, email_address, phone, password, created_on, updated_on) VALUES (:user_id, :username, :email_address, :phone, :password, :created_on, :updated_on)
>>> ins.compile().params
{'username': '111cn.net ', 'user_id': 8888, 'phone': 12345678901, 'created_on': None, 'updated_on': None, 'password': 'www.111cn.net ', 'email_address': 'support@111cn.net '}
8. 执行
>>> result = connection.execute(ins)
>>> result.inserted_primary_key
[8888]
9. 插入多条
>>> ins = users.insert()
>>> multi_data = [
... {
... 'user_id':1,
... 'username':'u1',
... 'email_address':'u1@111cn.net ',
... 'phone' : 12345678901,
... 'password': 'www.111cn.net '
... },
... {
... 'user_id':2,
... 'username':'u2',
... 'email_address':'u2@111cn.net ',
... 'phone' : 12345678901,
... 'password': 'www.111cn.net '
... }
... ]
>>> result = connection.execute(ins, multi_data)
>>> result.rowcount
2
10. 查询
>>> from sqlalchemy import select
>>> s = select([users])
>>> print str(s)
SELECT users.user_id, users.username, users.email_address, users.phone, users.password, users.created_on, users.updated_on
FROM users
>>> rp = connection.execute(s)
>>> results = rp.fetchall()
>>> print results
[(1, 'u1', 'u1@111cn.net ', '12345678901', 'www.111cn.net ', datetime.datetime(2016, 6, 22, 11, 26, 5), datetime.datetime(2016, 6, 22, 11, 26, 5)), (2, 'u2', 'u2@111cn.net ', '12345678901', 'www.111cn.net ', datetime.datetime(2016, 6, 22, 11, 26, 5), datetime.datetime(2016, 6, 22, 11, 26, 5)), (8888, '111cn.net ', 'support@111cn.net ', '12345678901', 'www.111cn.net ', datetime.datetime(2016, 6, 22, 11, 13, 32), datetime.datetime(2016, 6, 22, 11, 13, 32))]
>>> first_row = results[0]
>>> print first_row
(1, 'u1', 'u1@111cn.net ', '12345678901', 'www.111cn.net ', datetime.datetime(2016, 6, 22, 11, 26, 5), datetime.datetime(2016, 6, 22, 11, 26, 5))
>>> first_row[1]
'u1'
>>> first_row.phone
'12345678901'
>>> first_row[users.c.user_id]
>>> rp = connection.execute(s)
>>> for record in rp:
... print record.username
...
u1
u2
111cn.net
>>> s = select([users.c.user_id, users.c.username])
>>> rp = connection.execute(s)
>>> print rp.keys()
['user_id', 'username']
>>> print rp.fetchone()
(8888, '111cn.net ')
>>> print rp.fetchone()
(1, 'u1')
>>> print rp.fetchone()
(2, 'u2')
>>> print rp.fetchone()
None
11. 排序
>>> s = select([users.c.user_id, users.c.username])
>>> s = s.order_by(users.c.user_id)
>>> print str(s)
SELECT users.user_id, users.username
FROM users ORDER BY users.user_id
>>> rp = connection.execute(s)
>>> for i in rp:
... print ('{} - {}'.format(i.user_id, i.username))
...
1 - u1
2 - u2
8888 - 111cn.net
>>> s = select([users.c.user_id, users.c.username])
>>> s = s.order_by(desc(users.c.user_id))
>>> rp = connection.execute(s)
>>> print (['{} - {}'.format(i.user_id, i.username) for i in rp])
['8888 - 111cn.net ', '2 - u2', '1 - u1']
>>> s = select([users.c.user_id, users.c.username]).where(users.c.user_id.in_([1,2]))
>>> rp = connection.execute(s)
>>> print (['{} - {}'.format(i.user_id, i.username) for i in rp])
['1 - u1', '2 - u2']
12. 更新
>>> from sqlalchemy import update
>>> u = update(users).where(users.c.username == 'u1')
>>> u = u.values(phone=00000000000)
>>> print str(u)
UPDATE users SET phone=:phone, updated_on=:updated_on WHERE users.username = :username_1
>>> result = connection.execute(u)
>>> print result
<sqlalchemy.engine.result.ResultProxy object at 0x7f6839f1a290>
>>> print result.rowcount
1
>>> s = select([users]).where(users.c.username == 'u1')
>>> result = connection.execute(s).first()
>>> print result
(1, 'u1', 'u1@111cn.net ', '0', 'www.111cn.net ', datetime.datetime(2016, 6, 22, 11, 26, 5), datetime.datetime(2016, 6, 22, 13, 51, 16))
>>> print result.keys()
['user_id', 'username', 'email_address', 'phone', 'password', 'created_on', 'updated_on']
>>> for key in result.keys():
... print('{:>20}: {}'.format(key, result[key]))
...
user_id: 1
username: u1
email_address: u1@111cn.net
phone: 0
password: www.111cn.net
created_on: 2016-06-22 11:26:05
updated_on: 2016-06-22 13:51:16
13. 删除
>>> print str(u)
DELETE FROM users WHERE users.username = :username_1
>>> s = select([users]).where(users.c.username == 'u1')
>>> result = connection.execute(s).fetchall()
>>> print result
[]
