2

我在 sqlalchemy-0.7 中有一个带有日期列的类。我可以有一个column_property或任何类似的东西来让我获得一年并让我轻松过滤它吗？我该怎么写？

IE，我想要（声明式语法）：

class Foo(Base):
    id   = Column(Integer, primary_key=True)
    date = Column(Date(), nullable=False)
    year = column_property(something here)

# later on
q = session().query(Foo).filter_by(year=2011)
4

1 回答 1

5

当然你可以定义这样的属性：

year = column_property(extract('year', date))

但是你真的需要吗？您可以按年份过滤，而无需通过重写过滤条件定义此类属性：

query(Foo).filter(extract('year', Foo.date)==2011)

更新

虽然这个解决方案看起来很简单，但它也有一个缺点：WHERE 子句中的这种条件永远不会在日期字段上使用索引。有很多行和条件的高选择性，这将对性能产生很大影响。因此，您可能想要重写将导致 RANGE INDEX SCAN 而不是 FULL TABLE SCAN 的条件（如西蒙的评论中所建议的那样）：

start = datetime.date(year, 1, 1)
end = datetime.date(year, 12, 31)
query(Foo).filter(Foo.date.between(start, end))

也可以定义具有这种行为的属性，您只需重新定义比较器：

class YearComparator(ColumnProperty.Comparator):
    def __eq__(self, year):
        if isinstance(year, int):
            column = self.prop.columns[0].get_children()[0].expr
            start = datetime.date(year, 1, 1)
            end = datetime.date(year, 12, 31)
            return column.between(start, end)
        else:
            # It can be a column or exression which we can't handle such way
            return ColumnProperty.Comparator.__eq__(self, year)
    # __lt__, __gt__ etc. are very similar to __eq__

def year_property(date_column, **kwargs):
    kwargs.setdefault('comparator_factory', YearComparator)
    return column_property(extract('year', date_column), **kwargs)

class Foo(Base):
    __tablename__ = 'Foo'
    id = Column(Integer, primary_key=True)
    date = Column(Date, index=True)
    year = year_property(date)
于 2011-04-05T07:26:14.540 回答