在日常工作时，做为测试很少需要新建表，一般都是来操作已有的表。python提供了很多模块来直接操作数据库，例如：pymysql；但是需要写该死的sql，刚学习python时，感觉使用pymysql来操作数据库来可以哈，挺简单，但是随着经常用，就感觉比较麻烦。后来随着对python的不断学习，了解到了ORM框架(关系映射对象) SQLAlchemy，详细使用请参考这个。该框架建立在数据库API智商，使用关系对象映射进行数据库操作，换句话说就是：将对象转换成SQL，然后使用数据API执行SQL语句并获取到执行结果。
废话不多说，直接说怎么去操作已有的数据库。
1. 首先需要安装sqlacodegen
pip install sqlacodegen
2. 执行以下命令 需要安装pymysql
# sqlacodegen mysql+pymysql://root:[email protected]: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预期返回的字段')