0

我想通过工厂方法在 Pony ORM 中创建数据库实体，以避免类似表的代码重复。

这是我不完全工作的最小示例：

from pony.orm import *


def factory(db, tablename):
    class TableTemplate(db.Entity):
        _table_ = tablename
        first_name = Required(str)
        last_name = Required(str)
        composite_index(first_name, last_name)
    return TableTemplate


db = Database(provider='sqlite', filename=':memory:')
Table1 = factory(db, "TABLE_1")

# the following line produces the exception:
#    pony.orm.core.ERDiagramError: Entity TableTemplate already exists
Table2 = factory(db, "TABLE_2")

db.generate_mapping(create_tables=True)
with db_session:
    Table1(first_name="foo", last_name="bar")

可以通过使用创建具有动态名称的类来规避异常type，但这不适用于composite_index...

有没有使用 Pony ORM 的表工厂的好方法？

4

1 回答 1

2

这是我对你的类工厂的看法：

def factory(db, tablename):
    fields = {
        '_table': tablename,
        'first_name': Required(str)
        # rest of the fields
    }
    table_template = type(tablename.capitalize(),(db.Entity,),fields)
    return table_template

这将通过将名称大写tablename并设置描述符来创建一个类。我不确定元类虽然

composite_index问题更新

composite_index通过调用此方法使用一些非常晦涩的功能：

def _define_index(func_name, attrs, is_unique=False):
    if len(attrs) < 2: throw(TypeError,
        '%s() must receive at least two attributes as arguments' % func_name)
    cls_dict = sys._getframe(2).f_locals
    indexes = cls_dict.setdefault('_indexes_', [])
    indexes.append(Index(*attrs, is_pk=False, is_unique=is_unique))

一个小实验告诉我，你也许可以通过自己添加字段来执行相同的操作。所以这将使我们的工厂fields变量看起来像这样：

fields = {
        '_table': tablename,
        'first_name': Required(str),
        '_indexes_':[Index(('first_name','last_name'),is_pk=False,is_unique=False)]
        # rest of the fields
    }

试一试，让我知道。

OP 实验更新

最终的代码将是这样的：

from pony.orm import *
from pony.orm.core import Index
def factory(db, tablename):
    fields = {
        '_table': tablename,
        'first_name': Required(str)
        # rest of the fields
    }
    fields['_indexes_'] = [Index(fields['first_name'],fields['last_name'],is_pk=False,is_unique=False)]
    table_template = type(tablename.capitalize(),(db.Entity,),fields)
    return table_template
于 2019-02-12T17:26:16.560 回答