python 数据库操作框架_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预期返回的字段')