Flask-SQLAlchemy管理数据库

发表于 更新于 分类于 阅读次数： Disqus：

关于Flask-SQLAlchemy以前就写过一篇Flask数据库扩展Flask-SQLALchemy，不过当时并没能搞懂具体如何使用。现在阅读了不少相关英文材料后，最终还是选择了它。

关于Flask-SQLAlchemy的介绍及管网教程等可以看上面那一片。这里主要讲一下如何使用他管理数据库。

配置数据库

1
2
3
4
5
6
7
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////root/test/monitorDB.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
  1. 导入Flask主函数及SQLAlchemy扩展
  2. 设置数据库目录，如果用MYsql则采用
1
app.config['SQLALCHEMY_DATABASE_URI'] = 'msql://your-username:your-password@localhost/ufosightings'

设置数据库数据类

创建一个配置文件，如monitorDB.db，这个名字一定要起好，因为后续数据库引用操作都需要用到。

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
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////root/test/monitorDB.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True)
    password = db.Column(db.String(100))
    email = db.Column(db.String(120), unique=True)

    def __init__(self, username, password, email):
        self.username = username
        self.password = password
        self.email = email

    def __repr__(self):
        return '<User %r>' % self.username

class Piclist(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    picname = db.Column(db.String(80))
    time = db.Column(db.String(80))
    status = db.Column(db.Integer)

    def __init__(self, picname, time, status):
        self.picname = picname
        self.time = time
        self.status = status

    def __repr__(self, picname, time, status):
        reurn '<User %r>' % self.picname

我们设置了个类，第一个是User，用于存放用户信息；第二个Piclist用于存放图片信息；

用户信息包括：

  • id：唯一标示，整形。

  • username：用户名，字符串。

  • password：密码，字符串（加密算法暂不考虑）。

  • email：邮箱，字符串。
    照片信息包括：

  • id：唯一标示，整形。

  • picname：照片文件名字，字符串。

  • time：照片拍摄时间，字符串。

  • status：状态（1：正常，2：可疑），整形。

创建数据库

在ssh窗口输入python命令进入python shell，运行下面两行。

1
2
3
4

>>>>from monitorDB import db
>>>>db.create_all()

  1. 先导入数据库主函数db
  2. 创建所有的表

向数据库添加用户

继续在shell中操作，也可以单独写个py文件。

1
2
3
4
5
6
7
8

>>>>from monitorDB import User
>>>>admin = User('admin','password','admin@admin.com')
>>>>demo = User('demo','demo','demo@example.com')
>>>>db.session.add(admin)
>>>>db.session.add(demo)
>>>>db.session.commit()

  1. 从数据库配置中导入User表配置
  2. 创建两个User
  3. 把这两个User写入数据库
  4. 保存更改（如果不进行这句操作，那么上述添加操作就不会保存在数据库上。

查询记录

查询非常简单，只需要进行User.query.all()操作即可

如果需要查询特定用户，则User.query.filter_by(username='admin').first()

1
2
3
4
5
6
7
8
9

>>>>users = User.query.all()
>>>>print usersSql[<User u'admin'>, <User u'demo'>]
>>>>for user in users:
...     print user.id, user.username, user.password, user.email
...
1 admin password admin@admin.com
2 demo demo demo@example.com

SQLAlchemy查询结果为对象，如果需要详细信息，则可以使用上述循环来获取详细信息。

删除记录

删除数据库中记录也异常简单，只需要把添加操作中的add换成delete。

1
2
3
4

>>>>db.session.delete(demo)
>>>> db.session.commit()

更新记录

官网没有直接给出更新记录的方法，差点走了弯路，其实非常简单，比如我要更新demo用户的邮箱。

1
2
3
4

>>>>demo = User.query.filter_by(username='demo').first()
>>>>user.email = 'demo@example2.com'
>>>>db.session.commit()