本文目的是为了完成一个项目用到的flask基本知识,例子会逐渐加深。最好对着源码,一步一步走。
下载源码,运行
pip install -r requirements.txt 建立环境
python db_create.py 创建自己数据库
python db_migrate 迁移数据库
————————————————————————————–
flask 不仅简介小巧,同时运用的时候十分灵活。下面简单介绍一下如何编写一个 flask项目。
涉及调用开发服务器,数据库连接以及 ORM 映射,还有数据库的迁移,模板使用。后期再完善会话,管理后台,缓存等。
一 、安装环境
我们使用 flask web框架,并用 sqlalchemy来做数据库映射,并使用 migrate做数据迁移。
$ pip install flask
$ pip install SQLAlchemy==0.7.
$ pip install flask-sqlalchemy
$ pip install flask-migrate
$ pip install sqlalchemy-migrate
二、建立项目
flask 没有 django 那样原生的 manage管理工具(flask-admin可以实现,日后再说)。因此我们需要手动建立目录。新建一个 myproject目录,在里面建 app tmp两个文件夹,然后在 app文件夹里面建立 static, templates 两个文件夹,用来存储静态文件和模板。最后目录结构如下:
├── app
│ ├── static
│ ├── templates
└── tmp
三、 初始化文件
建立一些文件,在 app文件里建立 __init__.py models.py views.py 然后在与app 同目录下建立 config.py 和 run.py 此时的目录结构如下:
(env)admin@admindeMacBook-Air:~/project/python/flask/project$ tree ├── app
│ ├── static
│ ├── templates
│ ├── __init__.py
│ ├── models.py
│ ├── views.py
├── config.py
├── run.py
└── tmp
四 、开始项目
1 hello wrod
打开 (/app/__init.py) 文件,写入
# -*- coding: utf-8 -*-
from flask import Flask # 引入 flask
app = Flask(__name__) # 实例化一个flask 对象
import views # 导入 views 模块
# from app import views
注意,我们的 app 文件夹其实是一个python包,from app import views 这句话也就是从 包app里面导入 views模块,所以写成注释的那句话也没错,其他代码实践喜欢写成后面那种
现在我们来开始写我们的视图,打开 (/app/views.py)
# -*- coding: utf-8 -*-
from app import app
from models import User, Post, ROLE_USER, ROLE_ADMIN
@app.route(‘/’)
def index():
return ‘hello world, hello flask’
下一步我们将要启动我们的开发服务器,打开 (/run.py)
# -*- coding: utf-8 -*-
from app import app
if __name__ == ‘__main__’:
app.debug = True # 设置调试模式,生产模式的时候要关掉debug
app.run() # 启动服务器
这段代码需要注意第一句 from app import app。也许会有疑问,我们的 app 包里,貌似没有 app.py 这样的模块。其实这是 flask 方式和python的导入方式。from app 指导入 app 包里的 __iniy__.py 所以这句话的含义是导入 __.init__.py 里面的 app实例。
打开shell 运行
$ python run.py
(env)admin@admindeMacBook-Air:~/project/python/flask/project$ python run.py
* Running on http://127.0.0.1:5000/
* Restarting with reloader
用浏览器打开 http://127.0.0.1:5000/ 将会看到一个输入 hello world 的页面
2 连接数据库
下面开始连接数据库引擎 先要做一个简单的配置 打开 (/config.py)
# -*- coding: utf-8 -*- import os
basedir = os.path.abspath(os.path.dirname(__file__))
SQLALCHEMY_DATABASE_URI = ‘sqlite:///%s’ % os.path.join(basedir, ‘app.db’)
SQLALCHEMY_MIGRATE_REPO = os.path.join(basedir, ‘db_repository’)
CSRF_ENABLED = True
SECRET_KEY = ‘you-will-never-guess’
SQLALCHEMY_DATABASE_URI是the Flask-SQLAlchemy必需的扩展。这是我们的数据库文件的路径。
SQLALCHEMY_MIGRATE_REPO 是用来存储SQLAlchemy-migrate数据库文件的文件夹。
然后打开 (/app/__init__.py)
# -*- coding: utf-8 -*-
import os
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from config import basedir app = Flask(__name__)
app.config.from_object(‘config’) # 载入配置文件
db = SQLAlchemy(app) # 初始化 db 对象
# from app import views, models # 引用视图和模型
import views, models
打开 (/app/models.py)
# -*- coding: utf-8 -*- from app import db
ROLE_USER = 0
ROLE_ADMIN = 1
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
nickname=db.Column(db.String(60), index=True, unique=True)
email = db.Column(db.String(120), index=True, unique=True)
role = db.Column(db.SmallInteger, default=ROLE_USER)
def __repr__(self):
return ‘<User %r>’ % self.nickname
我们的模型建立了一个 user类,正好是 数据库里面的 user 表,表有四个字段
现在我们的数据库配置已经好了,可是还没有建立数据库。新建一个文件 (/db_create.py)
# -*- coding: utf-8 -*- from migrate.versioning import api
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO
from app import db
import os.path db.create_all() if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
else:
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, api.version(SQLALCHEMY_MIGRATE_REPO))
这个脚本是完全通用的,所有的应用路径名都是从配置文件读取的。当你用在自己的项目时,你可以把脚本拷贝到你的项目目录下就能正常使用了。
打开shell 运行
$ python db_create.py
运行这条命令之后,你就创建了一个新的app.db文件。这是个支持迁移的空sqlite数据库,同时也会生成一个带有几个文件的db_repository目录,这是SQLAlchemy-migrate存储数据库文件的地方,注意如果数据库已存在它就不会再重新生成了。这将帮助我们在丢失了现有的数据库后,再次自动创建出来。
运行
slqite3 app.db
>>> .tables
>>> user
或者使用 sqlite 图形化客户端,没有需要安装 (windows 下直接下载安装包即可)
$ sudo apt-get install sqlitebrowser
3 数据库迁移
每当我们更改了 models 。等价于更改了数据库的表结构,这个时候,需要数据库同步。新建 (/db_migrate.py)
# -*- coding: utf-8 -*- import imp
from migrate.versioning import api
from app import db
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO migration = SQLALCHEMY_MIGRATE_REPO + '/versions/%03d_migration.py' % (api.db_version(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO) + 1)
tmp_module = imp.new_module('old_model')
old_model = api.create_model(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO) exec old_model in tmp_module.__dict__ script = api.make_update_script_for_model(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, tmp_module.meta, db.metadata)
open(migration, 'wt').write(script) api.upgrade(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO) print 'New migration saved as' + migration
print 'Current database version:' + str(api.db_version(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO))
这个脚本看起来很复杂,SQLAlchemy-migrate通过对比数据库的结构(从app.db文件读取)和models结构(从app/models.py文件读取)的方式来创建迁移任务,两者之间的差异将作为一个迁移脚本记录在迁移库中,迁移脚本知道如何应用或者撤销一次迁移,所以它可以方便的升级或者降级一个数据库的格式。
开始数据库迁移
$ python db_mirgate.py
New migration saved as db_repository/versions/001_migration.py Current database version: 1
相应的,我们继续创建数据库 升级和回退的脚本
(/db_upgrade.py)
# -*- coding: utf-8 -*- from migrate.versioning import api
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO
api.upgrade(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
print 'Current database version:' + str(api.db_version(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO))
(/db_downgrade.py)
# -*- coding: utf-8 -*- from migrate.versioning import api
from config import SQLALCHEMY_DATABASE_URI
from config import SQLALCHEMY_MIGRATE_REPO v = api.db_version(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
api.downgrade(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, v – 1) print 'Current database version:' + str(api.db_version(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO))
4 操作数据库
接下来,我们定义一个新的models class 并做第二次迁移归并
打开(/app/models.py)
# -*- coding: utf-8 -*- from app import db
ROLE_USER = 0
ROLE_ADMIN = 1 class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
nickname=db.Column(db.String(60), index=True, unique=True)
email = db.Column(db.String(120), index=True, unique=True)
role = db.Column(db.SmallInteger, default=ROLE_USER)
posts = db.relationship(‘Post’, db.backref = ‘author’, db.lazyload = ‘dynamic’) def __repr__(self):
return ‘<User %r>’ % self.nickname class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.String(140))
timestamp = db.Column(db.DateTime)
user_id = db.Column(db.Integer, db.ForeignKey(‘user.id’)) def __repr__(self):
return ‘<Post %r>’ % (self.body)
这里我们进行了两个class 的关联。
$ python db_mirgrate.py
New migration saved as db_repository/versions/002_migration.py Current database version: 2
打开 (/app/views.py)
# -*- coding: utf-8 -*- from flask import render_template, flash, redirect, session, url_for, request, g
from app import app, db
from models import User, Post, ROLE_USER, ROLE_ADMIN @app.route(‘/’)
def index():
return render_template(‘index.html’) @app.route(‘/adduser/<nickname>/<email>’)
def adduser(nickname, email):
u = User(nickname=nickname, email=email)
try:
db.session.add(u)
db.session.commit()
return ‘add successful’
except Exception, e:
return ‘something go wrong’ @app.route(‘/getuser/<nickname>’)
def getuser(nickname):
user = User.query.filter_by(nickname=nickname).first()
return render_template(‘user.html’, user=user) @app.errorhandler(404)
def internal_error(error):
return render_template(‘404.html’), 404 @app.errorhandler(500)
def internal_error(error):
db.session.rollback()
return render_template(‘500.html’), 500
这次我们使用了模板, 新建(/app/templates/user.html)
<html>
<head>
<title>user</title>
</head>
<body>
<h1>user</h1>
<ul>
<li>user: {{ user.nickname }}</li>
<li>email: {{ user.email }}</li>
</ul>
</body>
</html>
最后运行
$ python run.py
用浏览器访问 http://127.0.0.1:5000/adduser/username/useremail
最后还可以用 sqlite 客户端打开查看我们的数据库变换。关于更多的数据库操作方法,请阅读 sqlalchemy文档。
最后我们的代码目录结构如下:
(env)admin@admindeMacBook-Air:~/project/python/flask/project$ tree
.
├── app
│ ├── __init__.py
│ ├── models.py
│ ├── static
│ ├── templates
│ │ ├── .html
│ │ ├── .html
│ │ ├── index.html
│ │ └── user.html
│ ├── views.py
├── app.db
├── config.py
├── db_create.py
├── db_downgrade.py
├── db_migrate.py
├── db_repository
│ ├── __init__.py
│ ├── manage.py
│ ├── migrate.cfg
│ ├── README
│ └── versions
│ ├── 001_migration.py
│ ├── 002_migration.py
│ ├── __init__.py
├── db_upgrade.py
├── requirements.txt
├── run.py
├── tmp
5 表单
接下来,我们将要接触表单方面的内容。flask原生提供的表单处理,也比较简单。当然,有了*,我们就直接用好了。需要安装一个
Flask-WTF==0.9.4。
安装完之后,打开 (/app/forms.py)
# -*- coding: utf-8 -*- import re
from flask_wtf import Form
from flask_wtf.html5 import EmailField
from wtforms import TextField, PasswordField, BooleanField
from wtforms.validators import DataRequired, ValidationError
from models import User
from hashlib import md5 class RegisterFrom(Form):
nickname = TextField(‘nickname’, validators=[DataRequired()])
email = EmailField(’email’, validators=[DataRequired()])
password = PasswordField(‘password’, validators=[DataRequired()])
conform = PasswordField(‘conform’, validators=[DataRequired()]) def validate_nickname(self, field):
nickname = field.data.strip()
if len(nickname) < 3 or len(nickname) > 20:
raise ValidationError(‘nickname must be 3 letter at least’)
elif not re.search(r’^\w+$’, nickname):
raise ValidationError(‘User names can contain only alphanumeric characters and underscores.’)
else:
# 验证是否已经注册
u = User.query.filter_by(nickname=nickname).first()
if u:
raise ValidationError(‘The nickname already exists’) def validate_email(self, field):
email = field.data.strip()
email = User.query.filter_by(email=email).first()
if email:
raise ValidationError(‘The email already exists’) def validate_password(self, field):
password = field.data.strip()
if len(password) < 6:
raise ValidationError(‘password must be 3 letter at least’) def validate_conform(self, field):
conform = field.data.strip()
if self.data[‘password’] != conform:
raise ValidationError(‘the password and conform are different’)
上述代码定义了一个 RegisterFrom class,正好是一个表单,class的字段映射为表单的域。其中 wtforms 中有 TextField, PasswordField, BooleanField这些表单类型,flask_wtf.html5 中 EmailField 则是 html5 规范的。
表单有 form.validate_on_submit() 方法,这个方法执行之前会验证表单域。验证方法和django类似,都是 validate_field 的方法。其中一个参数 field正好是表单的value值 需要注意验证两次密码是否相同的时候,需要用到 self.data 。这是一个字典,包含了表单域的name和value
6 用户注册
打开(/app/views.py)添加注册方法。
@app.route(‘/account/signup’, methods=[‘GET’, ‘POST’])
def signup():
form = RegisterFrom()
if request.method == ‘POST’:
if form.validate_on_submit():
psdmd5 = md5(form.data[‘password’])
password = psdmd5.hexdigest()
u = User(nickname=form.data[‘nickname’], email=form.data[’email’], password=password)
try:
db.session.add(u)
db.session.commit()
flash(‘signup successful’)
except Exception, e:
return ‘something goes wrong’
return redirect(url_for(‘signin’)) return render_template(‘signup.html’, form=form)
模版 新建 (/app/tempaltes/signup.html)
<!– extend base layout –>
{% extends “base.html” %} {% block content %}
<form method=”POST” action=”>
{{ form.hidden_tag() }}
<p>{{ form.nickname.label }} {{ form.nickname(size=20) }}</p>
<p>{{ form.email.label }} {{ form.email(size=20) }}</p>
<p>{{ form.password.label }} {{ form.password(size=20) }}</p>
<p>{{ form.conform.label }} {{ form.conform(size=20) }}</p>
<input type=”submit” value=”Sign Up”>
<input type=”reset” value=”Reset”>
</form> {{ form.errors }}
{% endblock %}
其中,主要是用到 form 对象的一些属性。form.nickname.label 是指表单类定义的名字,form.nickname 会转变成 表单域的 html 代码 即 <input type=”text” id=”nickname” name=”nickaname” value=””/>
7 用户登录
先添加一个用户登录的表单,打开 (/app/forms.py)
class LoginForm(Form):
nickname = TextField(‘nickname’, validators=[DataRequired()])
password = PasswordField(‘password’, validators=[DataRequired()])
remember_me = BooleanField(‘remember_me’, default=False) def validate_nickname(self, field):
nickname = field.data.strip()
if len(nickname) < 3 or len(nickname) > 20:
raise ValidationError(‘nickname must be 3 letter at least’)
elif not re.search(r’^\w+$’, nickname):
raise ValidationError(‘User names can contain only alphanumeric characters and underscores.’)
else:
return nickname
用户登录,最简单的方法就是 验证用户名,如果通过,则添加 session,登出的时候,清除session即可,模版里面可以直接使用 request.session用来判断用户登录状态
打开(/app/views.py) 添加登录登出方法
@app.route(‘/account/signin’, methods=[‘GET’, ‘POST’])
def signin():
form = LoginForm()
if request.method == ‘POST’:
if form.validate_on_submit():
nickname = form.data[‘nickname’]
psdmd5 = md5(form.data[‘password’])
password = psdmd5.hexdigest()
u = User.query.filter_by(nickname=nickname, password=password).first()
if u:
session[‘signin’] = True
flash(‘signin successful’)
return redirect(url_for(‘index’))
else:
flash(u’用户名或者密码错误’)
return render_template(‘signin.html’, form=form) @app.route(‘/account/signout’)
def signout():
session.pop(‘signin’, None)
flash(‘signout successful’)
return redirect(‘index’)
注意,我们使用 flask 的时候,用了中文,就必须是 unicode
除了原生的登录方式,我们还可以用flask-login。安装flask-login
此时需要初始化一个 login_manager 对象,打开 (/app/__init__.py)
# -*- coding: utf-8 -*- import os
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.login import LoginManager
from config import basedir app = Flask(__name__)
app.config.from_object(‘config’) db = SQLAlchemy(app) # 初始化数据库管理对象
login_manager = LoginManager() # 初始化登录管理对象
login_manager.init_app(app)
login_manager.login_view = “signin” # 登录跳转视图
login_manager.login_message = u”Bonvolu ensaluti por uzi tio paĝo.” # 登录跳转视图前的输出消息 # from app import views, models import views, models
然后打开 (/app/views.py)
添加下面方法, 注释掉之前的 登入登出方法
from flask.ext.login import login_user, logout_user, current_user, login_required
from app import app, db, login_manager @login_manager.user_loader
def load_user(userid):
return User.query.get(userid) @app.route(‘/account/signin’, methods=[‘GET’, ‘POST’])
def signin():
form = LoginForm()
if request.method == ‘POST’:
if form.validate_on_submit():
nickname = form.data[‘nickname’]
psdmd5 = md5(form.data[‘password’])
password = psdmd5.hexdigest()
remember_me = form.data[‘remember_me’]
user = User.query.filter_by(nickname=nickname, password=password).first()
if user:
login_user(user, remember = remember_me)
flash(‘signin successful’)
return redirect(request.args.get(“next”) or url_for(“index”))
else:
flash(u’用户名或者密码错误’)
return render_template(‘signin.html’, form=form) @app.route(‘/account/signout’) @login_required
def signout():
logout_user()
flash(‘signout successful’)
return redirect(‘index’)
模版也会响应的改,具体看源码吧。
base.html
<html>
<head>
<meta http-equiv=”Content-Type” content=”text/html; charset=utf-8″ />
{% if title %}
<title>{{title}} – microblog</title>
{% else %}
<title>microblog</title>
{% endif %}
</head>
<body>
<div class="metanav"> <div>
<a href=”{{ url_for(‘index’) }}”>Home</a>
{% if current_user.is_authenticated() %}
<a href=”{{ url_for(‘signout’) }}”>signout</a>
{% else %}
<a href=”{{ url_for(‘signin’) }}”>signin</a>
{% endif %}
</div> {% for message in get_flashed_messages() %}
<div class="flash">{{ message }}</div>
{% endfor %}
{% block content %}{% endblock %}
</body>
</html>
代码下载: https://github.com/rsj217/flask-extend/tree/master/project