From fd93b08ce645dfd2547615257945f528d178e60a Mon Sep 17 00:00:00 2001 From: =?utf8?q?Brigitta=20Sip=C5=91cz?= Date: Wed, 4 Sep 2024 17:41:43 -0400 Subject: [PATCH] BUG: using partial to have functions --- examples/extending_query/temporal_range.py | 6 +++--- test/orm/test_relationship_criteria.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/extending_query/temporal_range.py b/examples/extending_query/temporal_range.py index e7bde15af2..92f182780f 100644 --- a/examples/extending_query/temporal_range.py +++ b/examples/extending_query/temporal_range.py @@ -5,6 +5,7 @@ to selected entities. """ import datetime +from functools import partial from sqlalchemy import Column from sqlalchemy import create_engine @@ -22,9 +23,8 @@ from sqlalchemy.orm import sessionmaker class HasTemporal: """Mixin that identifies a class as having a timestamp column""" - timestamp = Column( - DateTime, default=datetime.datetime.now(datetime.UTC), nullable=False - ) + utc = partial(datetime.datetime.now, datetime.UTC) + timestamp = Column(DateTime, default=utc, nullable=False) def temporal_range(range_lower, range_upper): diff --git a/test/orm/test_relationship_criteria.py b/test/orm/test_relationship_criteria.py index a603997e3c..d4c532a0b7 100644 --- a/test/orm/test_relationship_criteria.py +++ b/test/orm/test_relationship_criteria.py @@ -1,6 +1,7 @@ from __future__ import annotations import datetime +from functools import partial import random from typing import List @@ -1660,9 +1661,8 @@ class TemporalFixtureTest(testing.fixtures.DeclarativeMappedTest): class HasTemporal: """Mixin that identifies a class as having a timestamp column""" - timestamp = Column( - DateTime, default=datetime.datetime.now(datetime.UTC), nullable=False - ) + utc = partial(datetime.datetime.now, datetime.UTC) + timestamp = Column(DateTime, default=utc, nullable=False) cls.HasTemporal = HasTemporal -- 2.47.2