python使用SQLAlchemy模块连接MySQL

ORM技术：Object-Relational Mapping，负责把关系数据库的表结构映射到对象上。

1.安装SQLAlchemy，MySQLdb模块

MySQLdb安装教程：http://www.cnblogs.com/jfl-xx/p/7299221.html

SQLAlchemy模块：

pip install sqlalchemy

2.初始化连接

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from sqlalchemy import Column, create_engine
from sqlalchemy.types import *
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 连接数据库
sql_connect = 'mysql://user:password@hostname:port/database?charset=xx'
engine = create_engine(sql_connect)
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)

注：红色部分：数据库类型://用户名:口令@机器地址:端口号/数据库名?字符编码.

完整的则为：数据库类型+数据库驱动://用户名:口令@机器地址:端口号/数据库名?字符编码.

例子：sql_connect = 'mysql://root:123456@192.168.1.110:3306/mytest?charset=utf8'

3.自动建表

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from sqlalchemy import Column
from sqlalchemy.types import *
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 连接数据库
sql_connect = 'mysql://user:password@hostname:port/datbase?charset=utf8'
engine = create_engine(sql_connect)
DBSession = sessionmaker(bind=engine)
# 创建对象的基类:
BaseModel = declarative_base()

#定义对象
class User(BaseModel):
    # 表名
    __tablename__ = 'user'
    # 表结构
    id = Column(String(20), primary_key=True)
    name = Column(String(20))
    age = Column(Integer)

#创建映射的数据库表
def init_db():
    BaseModel.metadata.create_all(engine)

#删除映射的数据库表
def drop_db():
    BaseModel.metadata.drop_all(engine)

4.添加数据

# 创建session对象,相当于MySQLdb里面的游标
session = DBSession()
# 创建新User对象:
new_user = User(id='2', name='John', age=13)
# 添加到session:
session.add(new_user)
# 提交即保存到数据库
session.commit()
# 关闭session
session.close()

5.查询数据

在查询语句中的where子句用filter()替代，不过匹配的值需要用"=="；如果使用filter_by()，则使用"="即可。

查询一条：

# 创建Session:
session = DBSession()
# 创建Query查询，filter是where条件，最后调用one()返回唯一行，如果调用all()则返回所有行:
user = session.query(User).filter(User.name == 'John', User.age > 12).one()
# 打印类型和对象的name属性:
print 'type:', type(user)
print 'age:', user.age
# 关闭Session:
session.close()

查询多条：

 1 # 创建Session:
 2 session = DBSession()
 3 # 创建Query查询，filter是where条件，最后调用one()返回唯一行，如果调用all()则返回所有行:
 4 users = session.query(User).filter(User.name == 'John').all()
 5 # 打印类型和对象的name属性:
 6 print 'type:', type(users)
 7 for u in users:
 8     print u.id
 9 # 关闭Session:
10 session.close()

6.更新数据

方式一：

# 创建Session:
session = DBSession()
# 可以进行多条数据更新
user = session.query(User).filter(User.id == '5')
user.update({User.age: 14})
# 提交数据
session.commit()
# 关闭Session
session.close()

方式二：

# 创建Session:
session = DBSession()
# 可以进行多条数据更新
user = session.query(User).filter_by(id='5').one()
user.age = 16
# 提交数据
session.commit()
# 关闭Session
session.close()

7.删除数据

# 创建Session
session = DBSession()
# 删除哪些数据
user = session.query(User).filter(User.id == '6').one()
session.delete(user)
# 提交数据
session.commit()
# 关闭session
session.close()