覆蓋用於測試的SQLAlchemy列

我想覆蓋SQLAlchemy列以在測試數據庫中擁有不同的屬性。基本上，我在我所有的MySQL表中都有一個複合主鍵：一個自動增量BIGINT id和一個DATETIME create_time。對於單元測試，我想使用SQLite，但是我的研究讓我相信在這個用例中autoincrement是不可能的。由於除了在MySQL中創建滾動分區以外，我不使用我的create_time列，所以我想重寫單元測試中的兩列，以便它們與SQLite兼容。注意：我正在使用Base = declarative_base()路線，並且我所有的型號都繼承了Base和PrimaryKeyMixin。覆蓋用於測試的SQLAlchemy列

在我的models.py中，我創建了所有表繼承的mixin PrimaryKeyMixin。所有表格還爲__table_args__組件繼承TABLE_ARGS。

TABLE_ARGS = { 
    'mysql_engine': 'InnoDB', 
    'mysql_charset': 'utf8' 
} 

class PrimaryKeyMixin: 
    id = Column(BIGINT(unsigned=True), primary_key=True, autoincrement=True) 
    create_time = Column(DATETIME, primary_key=True, default=func.now())

在我的單元測試，我做了以下內容：

from myapp.models import TABLE_ARGS, PrimaryKeyMixin, Column, INT, DATETIME, func 
TABLE_ARGS['sqlite_autoincrement'] = True 
PrimaryKeyMixin.id = Column(INT(unsigned=True), primary_key=True, autoincrement=True) 
PrimaryKeyMixin.create_date = Column(DATETIME, default=func.now()) 

TEST_DB = '/tmp/myapp_test.db' 
from myapp import db 
from myapp.models import Base 
db.ENGINE = create_engine(
    'sqlite:///%s' % TEST_DB, 
    encoding='utf-8', 
) 
Base.metadata.drop_all(db.ENGINE) 
Base.metadata.create_all(db.ENGINE)

然而，當我添加爲每個表回聲=真到發動機，但它仍然是創造一個PRIMARY_KEY（ID，CREATE_TIME）。

來源

2013-07-15 Matt Stern

難道這樣的：

Base.metadata = MetaData() 

def copy_table(model): 
    Table(model.__tablename__, Base.metadata, *change_columns(model.__table__.columns), sqlite_autoincrement=True) 

def change_columns(columns): 
    new_cols = [column.copy() for column in columns if column.name != 'id' and column.name !='create_time'] 
    new_cols.append(Column('id', INT(unsigned=True), primary_key=True, autoincrement=True)) 
    new_cols.append(Column('create_time', DATETIME, default=func.now(), autoincrement=True)) 
    return new_cols 

for model in PrimaryKeyMixin.__subclasses__(): 
    copy_table(model)

來源

2013-07-15 20:56:58

覆蓋用於測試的SQLAlchemy列

回答

相關問題