Flask 源代码阅读笔记

我认为我已经养成了一个坏习惯。在使用一个框架过程中对它的内部原理非常感兴趣，有时候须要花不少精力才

明确，这也导致了学习的缓慢，但换来的是对框架的内部机理的熟悉，正如侯捷所说，源代码面前，了无秘密。这也是

本文产生的直接原因。

一.flask session原理

flask的session是通过client的cookie实现的。不同于diango的server端实现，flask通过itsdangerous这个苦将session的内容序列化到浏览器的cookie,当浏览器再次请求时将反序列化cookie内容。也就得到我们的session内容。比方说session['name']='kehan',clientsession例如以下。



我们来解密这个cookie存储了什么值该cookie通过.切割，分成了三部分：内容序列化+时间+防篡改值



通过第一部分我们就获得了session['name']的值，我们看看第二部分



第二部分保存的是时间。itsdangerous库为了降低时间戳的值。之前减掉了118897263，所以我们要加上。这个时间flask是用来推断session是否过期使用的。第三部分是session值和时间戳以及我们SECRET_KEY的防篡改值，通过HMAC算法签名。也就是说即使你改动了前面的值。因为签名值有误，flask不会使用该session。所以一定要保存好SECRET_KEY。以及确保它的复查度。不然一旦让别人知道了SECRET_KEY，就能够通过构造cookie伪造session值，这是非常恐怖的一件事。
我们知道一般为了保护session，所以session的生成还会包括clientuser_agent,remete_addr等，假设你认为使用flask提供的保护力度不够，能够使用flask_login这个扩展，一帮在flask使用认证时都会使用这个扩展，简单易用，还提供了更加强度的session保护。

二. flask扩展import 原理

我喜欢flask的一个理由就是导入简单。非扩展的都能够通过from flask导入，扩展的都是通过from flask.ext.导入。非常简洁。用django的过程中，常常不记得该从哪里导入。在flask的世界里，你无需烦恼。那么flask的扩展导入原理是什么呢？主要通过sys.meta_path实现的
当导入 from falsk.ext.example import E是将会运行flask/ext/__init__.py

def setup():
from ..exthook import ExtensionImporter
importer = ExtensionImporter(['flask_%s', 'flaskext.%s'], __name__)
importer.install()

install将会向sys.meta_path加入模块装载类，当import时会调用其find_module，假设返回非None,会调用load_module载入
比方当我们 from flask.ext.script import Manager时
会调用find_module('flask.ext.script')。prefinx是flask.ext所以将会调用load_module()
此时将会尝试import flask_script模块或flaskext.script

def install(self):
sys.meta_path[:] = [x for x in sys.meta_path if self != x] + [self]

def find_module(self, fullname, path=None):
if fullname.startswith(self.prefix):
return self

def load_module(self, fullname):
modname = fullname.split('.', self.prefix_cutoff)[self.prefix_cutoff]
for path in self.module_choices:
realname = path % modname
__import__(realname)

三. flask sqlalchemy原理

sqlalchemy是python中最强大的orm框架。无疑sqlalchemy的使用比django自带的orm要复杂的多。使用flask sqlalchemy扩展将拉近和django的简单易用距离。先来说两个比較重要的配置
app.config['SQLALCHEMY_ECHO'] = True =》配置输出sql语句
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True =》每次request自己主动提交db.session.commit(),
假设有一天你发现别的写的视图中有db.session.add，但没有db.session.commit。不要疑惑。他肯定配置了上面的选项。这是通过app.teardown_appcontext注冊实现

@teardown
def shutdown_session(response_or_exc):
if app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN']:
if response_or_exc is None:
self.session.commit()
self.session.remove()
return response_or_exc

response_or_exc为异常值。默认为sys.exc_info()[1]
上面self.session.remove()表示每次请求后都会销毁self.session，为什么要这么做呢？
这就要说说sqlalchemy的session对象了。
from sqlalchemy.orm import sessionmaker
session = sessionmaker()
一帮我们会通过sessionmaker()这个工厂函数创建session，但这个session并不能用在多线程中，为了支持多线程操作，sqlalchemy提供了scoped_session，通过名字反映出scoped_session是通过某个作用域实现的
所以在多线程中一帮都是例如以下使用session
from sqlalchemy.orm import scoped_session, sessionmaker
session = scoped_session(sessionmaker())

我们来看看scoped_session是怎样提供多线程环境支持的

class scoped_session(object):
def __init__(self, session_factory, scopefunc=None):

self.session_factory = session_factory
if scopefunc:
self.registry = ScopedRegistry(session_factory, scopefunc)
else:
self.registry = ThreadLocalRegistry(session_factory)

__init__中，session_factory是创建session的工厂函数，而sessionmaker就是一工厂函数(事实上是定义了__call__的函数)而scopefunc就是能产生某个作用域的函数，假设不提供将使用ThreadLocalRegistry

class ThreadLocalRegistry(ScopedRegistry):
def __init__(self, createfunc):
self.createfunc = createfunc
self.registry = threading.local()

def __call__(self):
try:
return self.registry.value
except AttributeError:
val = self.registry.v

从上面__call__能够看出。每次都会创建新的session，并发在线程本地变量中。你可能会好奇__call__是在哪里调用的？

def instrument(name):
def do(self, *args, **kwargs):
return getattr(self.registry(), name)(*args, **kwargs)
return do

for meth in Session.public_methods:
setattr(scoped_session, meth, instrument(meth))

正如我们所示，当我们调用session.query将会调用 getattr(self.registry(), 'query')，self.registry()就是调用__call__的时机。可是在flask_sqlalchemy中并没有使用ThreadLocalRegistry，创建scoped_session步骤例如以下

# Which stack should we use?

_app_ctx_stack is new in 0.9
connection_stack = _app_ctx_stack or _request_ctx_stack

def __init__(self, app=None,
use_native_unicode=True,
session_options=None):
session_options.setdefault(
'scopefunc', connection_stack.__ident_func__
)
self.session = self.create_scoped_session(session_options)

def create_scoped_session(self, options=None):
"""Helper factory method that creates a scoped session."""
if options is None:
options = {}
scopefunc=options.pop('scopefunc', None)
return orm.scoped_session(
partial(_SignallingSession, self, **options), scopefunc=scopefunc
)

我们看到scopefunc被设置为connection_stack.__ident_func__。而connection_stack就是flask中app上下文。
假设你看过前一篇文章你就知道__ident_func__事实上就是在多线程中就是thrading.get_ident。也就是线程id
我们看看ScopedRegistry是怎样通过_操作的

class ScopedRegistry(object):
def __init__(self, createfunc, scopefunc):
self.createfunc = createfunc
self.scopefunc = scopefunc
self.registry = {}

def __call__(self):
key = self.scopefunc()
try:
return self.registry[key]
except KeyError:
return self.registry.setdefault(key, self.createfunc())

代码也非常easy，事实上也就是依据线程id创建相应的session对象，到这里我们基本已经了解了flask_sqlalchemy的魔法了，和flask cookie,g有异曲同工之妙，这里有两个小问题？
1.flask_sqlalchemy是否能使用ThreadLocalRegistry？
大部分情况都是能够的。但假设wsgi对多并发使用的是greenlet的模式就不适用了
2.上面create_scoped_session中partial是干嘛的？
前面我们说过scoped_session的session_factory是可调用对象，但_SignallingSession类并未定义__call__，所以通过partial支持

到这里你就知道为什么每次请求结束要self.session.remove()，不然为导致存放session的字段太大

这里说一下对db.relationship lazy的理解。看例如以下代码

class Role(db.Model):
__tablename__ = 'roles'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
users = db.relationship('User', backref='role', lazy='dynamic')

class User(db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), unique=True, index=True)
role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

假设role是已经获取的一个Role的实例
lazy:dynamic => role.users不会返回User的列表， 返回的是sqlalchemy.orm.dynamic.AppenderBaseQuery对象
当运行role.users.all()是才会真正运行sql，这种优点就是能够继续过滤

lazy:select => role.users直接返回User实例的列表，也就是直接运行sql

注意：db.session.commit仅仅有在对象有变化时才会真的运行update

四. flask moment原理

flask moment简单封装了moment.js，moment.js通过js提供了对时间的支持。感兴趣的童鞋能够关注下，功能非常强大。 flask moment原理非常easy，使用带有时间的格式话字符串在dom载入后，使用moment.js处理一下，
该步操作有moment.include_moment()完毕。假设使用其他语言，如中文，调用moment.lang('zh-cn')
假设使用了flask bootstrap。仅仅须要在最后加入下面代码就可以(须要jquery支持)

{% block scripts %}
{{ super() }}
{{ moment.include_moment() }}
{{ moment.lang('zh-cn') }}
{% endblock %}

flask moment还提供了过了多长时间统计，refresh为True时。每分钟刷新一次，refresh也可为详细的刷新时间，单位为分钟
{{ moment(current_time).fromNow(refresh=True) }}

看上面我们知道。flask moment在模板中导入了moment这个对象。这是怎样实现的呢？

def init_app(self, app):
if not hasattr(app, 'extensions'):
app.extensions = {}
app.extensions['moment'] = _moment
app.context_processor(self.context_processor)

@staticmethod
def context_processor():
return {
'moment': current_app.extensions['moment']
}

通过app.context_processor给模板上下文加入了额为属性
def render_template(template_name_or_list, **context):
ctx.app.update_template_context(context)

在render_template中会把前面注冊的变量加入到context,所以在模板中就能够使用moment了。
而flask bootstrap是通过app.jinja_env.globals['bootstrap_find_resource'] = bootstrap_find_resource实现的

我们知道flask在初始化jinja环境的时候就将request,g,session等注入到全局了

rv.globals.update(
url_for=url_for,
get_flashed_messages=get_flashed_messages,
config=self.config,
# request, session and g are normally added with the
# context processor for efficiency reasons but for imported
# templates we also want the proxies in there.
request=request,
session=session,
g=g
)

但我在看源代码时发现_default_template_ctx_processor也会注入g。request，例如以下

def _default_template_ctx_processor():
"""Default template context processor.  Injects `request`,
`session` and `g`.
"""
reqctx = _request_ctx_stack.top
appctx = _app_ctx_stack.top
rv = {}
if appctx is not None:
rv['g'] = appctx.g
if reqctx is not None:
rv['request'] = reqctx.request
rv['session'] = reqctx.session
return rv

这不是反复嘛，有啥必要呢？
哈哈，认真看上面rv.globals.update的凝视部分能大概明确。
flask模板能够使用宏，须要使用import导入，此时导入的模板不能訪问不能訪问当前模板的本地变量，仅仅能使用全局变量。
这也就是为什么global中有g,request,session的理由，也即是为了支持在宏中使用g对象而本地变量导入g等是为了效率的原因。详细细节须要參考jinja2的文档。