使用外键和SQLAlchemy导入灯具?

时间:2010-06-02 23:20:54

标签: python postgresql sqlalchemy pylons fixture

我一直在尝试使用fixture来将测试数据集加载到我的Pylons / PostgreSQL应用程序中。这很有效,除非它们引用自动增量id字段时无法正确创建外键。

我的夹具看起来像这样:

class AuthorsData(DataSet):
    class frank_herbert:
         first_name = "Frank"
         last_name = "Herbert"

class BooksData(DataSet):
    class dune:
        title = "Dune"
        author_id = AuthorsData.frank_herbert.ref('id')

模特:

t_authors = sa.Table("authors", meta.metadata,
    sa.Column("id", sa.types.Integer, primary_key=True),
    sa.Column("first_name", sa.types.String(100)),
    sa.Column("last_name", sa.types.String(100)),
    )

t_books = sa.Table("books", meta.metadata,
    sa.Column("id", sa.types.Integer, primary_key=True),
    sa.Column("title", sa.types.String(100)),
    sa.Column("author_id", sa.types.Integer, sa.ForeignKey('authors.id')) 
    )

运行“paster setup-app development.ini”时,SQLAlchemey将FK值报告为“无”,因此显然没有找到它:

15:24:27,284 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] 
CREATE TABLE authors (
    id SERIAL NOT NULL, 
    first_name VARCHAR(100), 
    last_name VARCHAR(100), 
    PRIMARY KEY (id)
)

15:24:27,284 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] {}
15:24:27,291 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] COMMIT
15:24:27,292 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] 
CREATE TABLE books (
    id SERIAL NOT NULL, 
    title VARCHAR(100), 
    author_id INTEGER, 
    PRIMARY KEY (id), 
    FOREIGN KEY(author_id) REFERENCES authors (id)
)

15:24:27,293 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] {}
15:24:27,300 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] COMMIT
15:24:27,301 INFO  [blog.websetup] Inserting initial data
15:24:27,302 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] BEGIN
15:24:27,309 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] INSERT INTO authors (first_name, last_name) VALUES (%(first_name)s, %(last_name)s) RETURNING authors.id
15:24:27,309 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] {'first_name': 'Frank', 'last_name': 'Herbert'}
15:24:27,320 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] INSERT INTO books (title, author_id) VALUES (%(title)s, %(author_id)s) RETURNING books.id
15:24:27,320 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] {'author_id': None, 'title': 'Dune'}
15:24:27,322 INFO  [sqlalchemy.engine.base.Engine.0x...9f50] COMMIT
15:24:27,323 INFO  [blog.websetup] Done

夹具文档实际上警告这可能是一个问题:

“但是,在某些情况下,您可能需要引用一个在加载之前没有值的属性,如序列ID列。(请注意,使用会话时,SQLAlchemy数据层不支持此功能。) “

http://farmdev.com/projects/fixture/using-dataset.html#referencing-foreign-dataset-classes

这是否意味着SQLAlchemy不支持这个?或者是否可以在不使用SA“会话”的情况下加载数据?其他人如何处理这个问题?

3 个答案:

答案 0 :(得分:1)

SQLAlchemy中的SQL表达式层不支持引用尚未插入的行,因为它没有意义:所有查询都是显式执行的。 ORM层确实支持对新对象的引用,它甚至使用工作单元模式来正确组织插入顺序。为解决此问题,fixture使用ref()方法。但是请注意,在SQL表达层中你(fixture)执行语句,因此SQLAlchemy没有机会重新排序它们。因此,如果您尝试在其引用的作者之前存储图书,则author_id不可用。你确定你的灯具是否按照正确的顺序组织(传递给data()方法)?

答案 1 :(得分:1)

您不能说author = AuthorsData.frank_herbert而不是直接指定author_id吗?

除此之外,在插入AuthorsData后调用meta.Session.flush()会强制id属性获取值吗?当我需要分配ID时(我不使用夹具),这就是我在测试中使用的。

答案 2 :(得分:0)

我通过定义orm.relation来解决这个问题:

class BooksData(DataSet):
    class dune:
        title = "Dune"
        author = AuthorsData.frank_herbert


mapper(Book, books, properties={
   'author': relation(Author)
 })