或者我该如何使这个东西工作?
我有一个Interval对象:
class Interval(Base):
__tablename__ = 'intervals'
id = Column(Integer, primary_key=True)
start = Column(DateTime)
end = Column(DateTime, nullable=True)
task_id = Column(Integer, ForeignKey('tasks.id'))
@hybrid_property #used to just be @property
def hours_spent(self):
end = self.end or datetime.datetime.now()
return (end-start).total_seconds()/60/60
一个任务:
class Task(Base):
__tablename__ = 'tasks'
id = Column(Integer, primary_key=True)
title = Column(String)
intervals = relationship("Interval", backref="task")
@hybrid_property # Also used to be just @property
def hours_spent(self):
return sum(i.hours_spent for i in self.intervals)
当然,添加所有典型的设置代码。
现在,当我尝试做 session.query(Task).filter(Task.hours_spent > 3).all()
我明白了 NotImplementedError: <built-in function getitem>
来自 sum(i.hours_spent...
线。
所以我在看 这部分 的文件和理论认为,可能有某种方式,我可以写一些能做我想做的事情。 这部分 也看起来它可能有用,我会在等待答案时看着它;)
SQLAlchemy不够智能从这些操作数构建SQL表达式树,您必须使用显式 propname.expression
装饰者提供它。但接下来又出现了另一个问题:没有可移植的方法将间隔转换为数据库中的小时数。你用的 TIMEDIFF
在MySQL中, EXTRACT(EPOCH FROM ... ) / 3600
在PostgreSQL等我建议更改属性以返回 timedelta
相反,并将苹果与苹果进行比较。
from sqlalchemy import select, func
class Interval(Base):
...
@hybrid_property
def time_spent(self):
return (self.end or datetime.now()) - self.start
@time_spent.expression
def time_spent(cls):
return func.coalesce(cls.end, func.current_timestamp()) - cls.start
class Task(Base):
...
@hybrid_property
def time_spent(self):
return sum((i.time_spent for i in self.intervals), timedelta(0))
@time_spent.expression
def hours_spent(cls):
return (select([func.sum(Interval.time_spent)])
.where(cls.id==Interval.task_id)
.label('time_spent'))
最后的查询是:
session.query(Task).filter(Task.time_spent > timedelta(hours=3)).all()
转换为(在PostgreSQL后端):
SELECT task.id AS task_id, task.title AS task_title
FROM task
WHERE (SELECT sum(coalesce(interval."end", CURRENT_TIMESTAMP) - interval.start) AS sum_1
FROM interval
WHERE task.id = interval.task_id) > %(param_1)s
有关SQLAlchemy的合并函数的简单示例,这可能有所帮助: 处理SQLAlchemy查询中的空值 - 相当于isnull,nullif或coalesce。
以下是该帖子的几行代码:
from sqlalchemy.sql.functions import coalesce
my_config = session.query(Config).order_by(coalesce(Config.last_processed_at, datetime.date.min)).first()
有一个完整的例子可以使func动作类似于 coalesc
要么 nvl
。
注意它是如何引用争论的,并呈现一个表达式......在这种情况下,NVL(a,b)与Oracle一起使用时。
http://docs.sqlalchemy.org/en/latest/core/compiler.html#subclassing-guidelines
from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import FunctionElement
class coalesce(FunctionElement):
name = 'coalesce'
@compiles(coalesce)
def compile(element, compiler, **kw):
return "coalesce(%s)" % compiler.process(element.clauses)
@compiles(coalesce, 'oracle')
def compile(element, compiler, **kw):
if len(element.clauses) > 2:
raise TypeError("coalesce only supports two arguments on Oracle")
return "nvl(%s)" % compiler.process(element.clauses)
然后当你想要使用它时......
from my_oracle_functions_sqla import coalesce
select([coalesce(A.value, '---')]) # etc
希望有所帮助。