与django contenttypes中的sqlalchemy的通用关系

时间:2011-03-22 02:13:14

标签: python django sqlalchemy flask

我正在尝试使用Sql Alchemy创建一些通用应用程序,例如任何模型的标签或评级。但我在文档中找不到任何帮助。我真的很喜欢用django contenttypes框架做什么? Sql Alchemy中是否有类似的功能?

1 个答案:

答案 0 :(得分:3)

我曾经写过一些关于类似内容的示例代码(请参阅http://taketwoprogramming.blogspot.com/2009/08/reusable-sqlalchemy-models.html)。

基本思想是你可以创建一个这样的模型:

@commentable
class Post(Base):
    __tablename__ = 'posts'
    id = sa.Column(sa.Integer, primary_key=True)
    text = sa.Column(sa.String)

...其中可评论的定义如下......

class BaseComment(object):
    pass

def build_comment_model(clazz):
    class_table_name = str(class_mapper(clazz).local_table)
    metadata = clazz.metadata

    comment_class_name = clazz.__name__ + 'Comment'
    comment_class = type(comment_class_name, (BaseComment,), {}) 

    comment_table_name = class_table_name + '_comments'
    comment_table = sa.Table(comment_table_name, metadata,
        sa.Column('id', sa.Integer, primary_key=True),
        sa.Column(class_table_name + '_id',
            sa.Integer,
            sa.ForeignKey(class_table_name + '.id')),

        sa.Column('text', sa.String),
        sa.Column('name', sa.String(100)),
        sa.Column('url', sa.String(255)),
    )   

    mapper(comment_class, comment_table)

    return comment_class, comment_table

def commentable(clazz):
    comment_class, comment_table = build_comment_model(clazz)

    clazz.Comment = comment_class
    setattr(clazz, 'comments', relation(comment_class))

    def add_comment(self, comment):
        self.comments.append(comment)

    setattr(clazz, 'add_comment', add_comment)

    return clazz

基本上,可注释装饰器动态地创建一个新类型和表,以及一些辅助方法到装饰类。这是我用来测试代码工作的测试,它显示了一些如何工作的例子......

class TestModels(SATestCase):
    def test_make_comment(self):
        p = Post()
        p.text = 'SQLAlchemy is amazing!'

        text = 'I agree!'
        name = 'Mark'
        url = 'http://www.sqlalchemy.org/'

        c = Post.Comment()
        c.text = text
        c.name = name
        c.url = url 
        p.add_comment(c)
        Session.add(p)

        # This is a method I use to force the reload of the objects from
        # the database to make sure that when I test them, I'm actually
        # pulling from the database rather than just getting the data
        # of the object still in the session.
        p = self.reload(p)

        self.assertEquals(len(p.comments), 1)
        c = p.comments[0]
        self.assertEquals(c.text, text)
        self.assertEquals(c.name, name)
        self.assertEquals(c.url, url)

我之前写过这篇文章,但我认为SQLA中没有任何东西可以为你做这种事情,但你可以创建类似的东西而不会有太多麻烦。在我的例子中,我创建了新的映射类和方法,以便在类装饰器中即时使用它。

我从来没有真正使用它,但它可能会给你一些想法。