3

执行 Pony ORM 查询并尝试按模型上的三个属性对查询进行排序。首先是歌曲类型，可以是 中列出的五个值之一ssf_type_order_map，然后是持续时间（int）和 uuid（字符串）。

对于歌曲类型，我希望歌曲按以下顺序排序：完整、完整（乐器）、短裤、循环、茎

如果我尝试使用以下.order_by()调用进行排序，它不会返回任何错误，但不会按类型排序，因为我需要按上述顺序进行排序（但持续时间和 UUID 排序工作正常）。

song_source_files = self.song_source_files.select(lambda ssf: True).order_by(lambda ssf: (ssf.type, ssf.duration, ssf.uuid))

这就是我认为理想的查询，将字符串类型映射到对其排序进行排序的映射。

ssf_type_order_map = {
    'Full': 1,
    'Full (Instrumental)': 2,
    'Shorts': 3,
    'Loops': 4,
    'Stems': 5
}

song_source_files = self.song_source_files.select(lambda ssf: True).order_by(lambda ssf: (ssf_type_order_map[ssf.type], ssf.duration, ssf.uuid))

ssf_type_order_map但是当我运行这个说“表达式有不支持的类型'dict'时我得到一个错误。

order_by here上的 Pony ORM 文档对在这种情况下使用 lambdas 非常模糊。

更新 - 9 月 7 日

我还尝试在模型上添加以下 getter 属性，如下所示：

@property
def source_type(self):
  ssf_type_order_map = {
    'Full': 1,
    'Full (Instrumental)': 2,
    'Shorts': 3,
    'Loops': 4,
    'Stems': 5
  }

  return ssy_type_order_map[self.type]

然后我尝试按如下方式对查询进行排序：

song_source_files = self.song_source_files.select(lambda ssf: True).order_by(lambda ssf: (ssf_type_order_map[ssf.type], ssf.duration, ssf.uuid))

但我收到一个错误，基本上说模型没有这个属性。基于 Django 的 ORM 的类似问题，我的假设是您只能访问数据库模型中存在的属性。

如果 Pony 也是这种情况，那么如何完成我想要完成的事情呢？

4

1 回答 1

3

首先，我想说的是，Pony 区分了两种子表达式：外部表达式和关联表达式。外部表达式不依赖于生成器循环变量的值，而相关表达式则依赖。考虑以下示例：

from some_module import f, g

x = 100
query = select(e for e in MyEntity if e.attr > f(x, 200) and g(x, e))

在这个查询中，我们有两个子表达式：第一个是f(x, 200)，第二个是g(x, e)。前者被 Pony 视为外部表达式，因为它不使用任何循环变量。在这种情况下，Pony 假设可以在执行查询之前在 Python 中计算表达式的值，然后将表达式转换为单个参数。对于这样的表达式，Pony 不会对可以在其中使用哪些 Python 函数施加任何限制，因为这样的表达式的结果只是 Python 中计算的单个值。

第二个表达式g(x, e)不能在 Python 中计算，因为它取决于循环变量的值e。对于不同的表行，此类表达式的结果可能不同。因此，Pony 需要将这样的表达式翻译成 SQL。不是每个 Python 表达式都可以翻译成 SQL，并且g需要是 Pony 专门知道如何翻译的函数。Pony 定义了一个可以翻译的 Python 操作子集。该子集包括数值类型的算术运算、字符串方法（如startswith、endswith、in等）以及聚合函数（如sum和max）。

在您的代码中，当您编写

.order_by(lambda ssf: (ssf_type_order_map[ssf.type], ssf.duration, ssf.uuid))

该表达式ssf_type_order_map[ssf.type]引用对象变量ssf，因此每个表行将具有不同的值，因此这是相关表达式，Pony 需要将该表达式转换为 SQL。目前 Pony 不明白如何进行这种特定的翻译，但原则上这是可行的。翻译结果为如下 SQLCASE语句：

ORDER BY CASE ssf.type
    WHEN 'Full' THEN 1
    WHEN 'Full (Instrumental)' THEN 2
    WHEN 'Shorts' THEN 3
    WHEN 'Loops' THEN 4
    WHEN 'Stems' THEN 5
    ELSE 0
END

好消息是你可以使用 Python if-expression 语法在 Pony 中编写这样的表达式：

(1 if ssf.type == 'Full' else
 2 if ssf.type == 'Full (Instrumental)' else
 3 if ssf.type == 'Shorts' else
 4 if ssf.type == 'Loops' else
 5 if ssf.type == 'Stems' else 0)

目前 Pony 还不支持反编译 if 表达式，所以如果你尝试直接编写这样的代码，你会得到一个异常。作为一种解决方法，您需要将 lambda 函数的源作为字符串传递。在这种情况下它会被翻译得恰到好处，因为我们可以直接将字符串解析为 AST 而无需反编译。所以你可以写：

song_source_files = self.song_source_files.select().order_by("""
    lambda ssf: ((1 if ssf.type == 'Full' else
                  2 if ssf.type == 'Full (Instrumental)' else
                  3 if ssf.type == 'Shorts' else
                  4 if ssf.type == 'Loops' else
                  5 if ssf.type == 'Stems' else 0),
                 ssf.duration, ssf.uuid)
""")

这应该可以完美地工作，但我建议以另一种方式解决这个问题：我们可以让SourceFileType实体具有nameandcode属性，然后ssf按值对记录进行排序ssf.type.code：

class SongSourceFile(db.Entity):
    name = Required(str)
    type = Required(lambda: SourceFileType)
    duration = Required(timedelta)
    uuid = Required(uuid.UUID, unique=True, default=uuid.uuid4)

class SourceFileType(db.Entity):
    name = Required(str)
    code = Required(int)
    files = Set(lambda: SongSourceFile)

然后可以通过以下方式编写查询：

song_source_files = self.song_source_files.select().order_by(
    lambda ssf: (ssf.type.code, ssf.duration, ssf.uuid)
)

我认为这种方法更通用，因为现在您可以添加其他有用的属性，SourceFileType并且name也code可以在查询中使用它们。

于 2016-09-07T13:13:29.713 回答