SQLAlchemy介绍和基本使用

数据库是一个网站的基础,在Flask中可以自由的使用MySQLPostgreSQLSQLiteRedisMongoDB来写原生的语句实现功能,也可以使用更高级别的数据库抽象方式,如SQLAlchemyMongoEngine这样的OR(D)M。本教程以MySQL+SQLAlchemy的组合来进行讲解。
在讲解Flask中的数据库操作之前,先确保你已经安装了以下软件:

  • mysql:如果是在windows上,到官网下载。如果是ubuntu,通过命令sudo apt-get install mysql-server libmysqlclient-dev -yq进行下载安装。
  • MySQLdbMySQLdb是用Python来操作mysql的包,因此通过pip来安装,命令如下:pip install mysql-python
  • pymysqlpymysql是用Python来操作mysql的包,因此通过pip来安装,命令如下:pip3 install pymysql
  • SQLAlchemySQLAlchemy是一个数据库的ORM框架,我们在后面会用到。安装命令为:pip3 install SQLAlchemy

    通过SQLAlchemy连接数据库:

    首先来看一段代码:
    1. from sqlalchemy import create_engine
    2. # 数据库的配置变量
    3. HOSTNAME = '127.0.0.1'
    4. PORT = '3306'
    5. DATABASE = 'xt_flask'
    6. USERNAME = 'root'
    7. PASSWORD = 'root'
    8. DB_URI = 'mysql+mysqldb://{}:{}@{}:{}/{}'.format(USERNAME,PASSWORD,HOSTNAME,PORT,DATABASE)
    9. # 创建数据库引擎
    10. engine = create_engine(DB_URI)
    11. #创建连接
    12. with engine.connect() as con:
    13. rs = con.execute('SELECT 1')
    14. print rs.fetchone()
    首先从sqlalchemy中导入create_engine,用这个函数来创建引擎,然后用engine.connect()来连接数据库。其中一个比较重要的一点是,通过create_engine函数的时候,需要传递一个满足某种格式的字符串,对这个字符串的格式来进行解释:
    1. dialect+driver://username:password@host:port/database
    dialect是数据库的实现,比如MySQLPostgreSQLSQLite,并且转换成小写。driverPython对应的驱动,如果不指定,会选择默认的驱动,比如MySQL的默认驱动是MySQLdbusername是连接数据库的用户名,password是连接数据库的密码,host是连接数据库的域名,port是数据库监听的端口号,database是连接哪个数据库的名字。
    如果以上输出了1,说明SQLAlchemy能成功连接到数据库。

    用SQLAlchemy执行原生SQL:

    我们将上一个例子中的数据库配置选项单独放在一个constants.py的文件中,看以下例子:
    1. from sqlalchemy import create_engine
    2. from constants import DB_URI
    3. #连接数据库
    4. engine = create_engine(DB_URI,echo=True)
    5. # 使用with语句连接数据库,如果发生异常会被捕获
    6. with engine.connect() as con:
    7. # 先删除users表
    8. con.execute('drop table if exists users')
    9. # 创建一个users表,有自增长的id和name
    10. con.execute('create table users(id int primary key auto_increment,'
    11. 'name varchar(25))')
    12. # 插入两条数据到表中
    13. con.execute('insert into users(name) values("xiaoming")')
    14. con.execute('insert into users(name) values("xiaotuo")')
    15. # 执行查询操作
    16. rs = con.execute('select * from users')
    17. # 从查找的结果中遍历
    18. for row in rs:
    19. print row