python-orm框架之操作已有数据库

  在日常工作时，做为测试很少需要新建表，一般都是来操作已有的表。python提供了很多模块来直接操作数据库，例如：pymysql；但是需要写该死的sql，刚学习python时，感觉使用pymysql来操作数据库来可以哈，挺简单，但是随着经常用，就感觉比较麻烦。后来随着对python的不断学习，了解到了ORM框架(关系映射对象) SQLAlchemy，详细使用请参考这个。该框架建立在数据库API智商，使用关系对象映射进行数据库操作，换句话说就是：将对象转换成SQL，然后使用数据API执行SQL语句并获取到执行结果。
  废话不多说，直接说怎么去操作已有的数据库。

1. 首先需要安装sqlacodegen
pip install sqlacodegen
2. 执行以下命令 需要安装pymysql
# sqlacodegen mysql+pymysql://root:qwer1234@127.0.0.1:3306/test > models.py

下面就是我生成的models.py

# coding: utf-8
from sqlalchemy import Column, String, text
from sqlalchemy.dialects.mysql import BIGINT, INTEGER
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
metadata = Base.metadata


class TbUser(Base):
    __tablename__ = 'tb_user'

    id = Column(BIGINT(20), primary_key=True)
    user_id = Column(INTEGER(20), nullable=False, comment='主键')
    accountNumber = Column(String(200), nullable=False, unique=True, server_default=text("''"), comment='账号')
    password = Column(String(50), nullable=False, unique=True, server_default=text("''"), comment='密码')
    name = Column(String(20), nullable=False, unique=True, server_default=text("'POST'"))


class TestCase(Base):
    __tablename__ = 'test_case'

    id = Column(BIGINT(20), primary_key=True, comment='主键')
    case_name = Column(String(500), nullable=False, server_default=text("''"), comment='用例名称')
    url = Column(String(100), nullable=False, server_default=text("''"), comment='url')
    method = Column(String(10), nullable=False, server_default=text("'POST'"), comment='请求方式')
    request_data = Column(String(1000), nullable=False, server_default=text("''"), comment='请求body')
    case_dependency = Column(INTEGER(4), nullable=False, server_default=text("'0'"), comment='依赖的case,0没有依赖')
    dependency_response = Column(String(100), nullable=False, server_default=text("''"), comment='依赖的case返回的字段')
    check_data = Column(String(100), nullable=False, server_default=text("''"), comment='case返回的字段')
    expect_data = Column(String(100), nullable=False, server_default=text("''"), comment='case预期返回的字段')