问题 如何在SQLAlchemy中实现空合并运算符?


或者我该如何使这个东西工作?

我有一个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... 线。

所以我在看 这部分 的文件和理论认为,可能有某种方式,我可以写一些能做我想做的事情。 这部分 也看起来它可能有用,我会在等待答案时看着它;)


8834
2017-08-07 11:23


起源



答案:


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

6
2017-08-07 12:01





有关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()

7
2018-06-06 14:37





有一个完整的例子可以使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

希望有所帮助。


0
2017-08-02 17:29