Tornado Web 开发 框架搭建 (4)

数据库设置

使用 mysqlconnector，python3 环境下比较好用的mysql连接库

DB_CONNECT_STRING = 'mysql+mysqlconnector://root:@localhost/demo?charset=utf8'

from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base

engine = create_engine(DB_CONNECT_STRING)
db_session = scoped_session(sessionmaker(autocommit=False,
autoflush=False,
bind=engine))
Base = declarative_base(metadata=MetaData(), metaclass=DeclarativeMeta)
Base.query = db_session.query_property()

这样 Base 可以作为 sqlalchemy的 Model 的 基类使用

class User(Base):
__tablename__ = 'user_tab'
id = Column(Integer, primary_key=True)
name = Column(String(50), unique=True, index=True, nullable=False)
pwd = Column(String(50), nullable=False)
real_name = Column(String(50), nullable=False)

如上，定义model: User

BaseHandler 中需要 重写俩方法

class BaseHandler(tornado.web.RequestHandler):
def initialize(self):
self.session = db_session

def on_finish(self):
self.session.close()

初始化时 加载 session,

请求结束关闭session

Handler中就可以正常使用sqlalchemy的语法了

user = self.session.query(User).filter_by(name=name).first()

表初始化

在环境搭建好后需要初始化数据库

如下 :

import models.userModel

def init_db():
# 在这里导入定义模型所需要的所有模块，这样它们就会正确的注册在
# 元数据上。否则你就必须在调用 init_db() 之前导入它们。
#import yourapplication.models
Base.metadata.create_all(bind=engine)

记住，在 create_all 之前需要导入 准备好的model

删除所有表可以

def drop_db():
#from models.userModel import User
"""

"""
Base.metadata.drop_all(engine)

使用alembic

正常开发中，数据库发生变动的情况还是蛮多的，版本管理就是个问题。

python3 的环境的话，推荐使用 alembic,是 sqlalchemy 作者写的

安装好 alembic后 cd 到你的目录文件夹,运行

alembic init alembic

会在同目录下生成 alembic.ini 和alembic 目录

打开alembic.ini 配置一下 数据库连接

sqlalchemy.url = mysql+mysqlconnector://root:@localhost/demo?charset=utf8

然后 打开alembic 目录下的 env.py

在这 需要把

def run_migrations_online():
"""Run migrations in 'online' mode.

In this scenario we need to create an Engine
and associate a connection with the context.

"""
engine = engine_from_config(
config.get_section(config.config_ini_section),
prefix='sqlalchemy.',
poolclass=pool.NullPool)

connection = engine.connect()
context.configure(
connection=connection,
target_metadata=target_metadata
)

try:
with context.begin_transaction():
context.run_migrations()
finally:
connection.close()

这里面的 target_metadata 取成 自己的 sqlalchemy 的 metadata

所以在之前加上

sys.path.append(os.getcwd())
from lib.db import Base
target_metadata = Base.metadata

这里要求就是

包含Base 所在包的路径

Base 在import进来的时候,所有model 必须加载到，也就是在 Base 所在文件里面 import 所有model

在命令行执行

alembic revision --autogenerate -m "start"

配置正确的话,会在alembic/versions 目录下生成 XXXX_start.py ,里面就包含了 更改model后数据库自动升级的信息,大致如下

# revision identifiers, used by Alembic.
revision = '12743faa412'
down_revision = None

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql

def upgrade():
### commands auto generated by Alembic - please adjust! ###
op.add_column('user_info_tab', sa.Column('mobile', sa.String(length=50), nullable=True))
op.alter_column('user_roleinfo_tab', 'id_no',
existing_type=mysql.VARCHAR(length=50),
nullable=False)
### end Alembic commands ###

def downgrade():
### commands auto generated by Alembic - please adjust! ###
op.alter_column('user_roleinfo_tab', 'id_no',
existing_type=mysql.VARCHAR(length=50),
nullable=True)
op.drop_column('user_info_tab', 'mobile')
### end Alembic commands ###

然后在命令行执行

alembic upgrade head

就一步更新到位了。
也可以使用

alembic revision -m "Add a column"

生成空白的版本文件，自己填充升级降级信息。

同样，也可以自由控制版本升降

alembic upgrade +2
alembic downgrade -1
alembic upgrade ae1(版本号)