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
[]