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

 这里要求就是

1.  包含Base 所在包的路径
2.  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(版本号)