From fe7f158221d444dda2b79dd0dfcbabef893fc33b Mon Sep 17 00:00:00 2001 From: tim-haselhoff Date: Tue, 26 Nov 2024 21:21:44 +0100 Subject: [PATCH] Fix arrow python type --- sqlalchemy_utils/types/arrow.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/sqlalchemy_utils/types/arrow.py b/sqlalchemy_utils/types/arrow.py index d4c13cc1..6e0958c5 100644 --- a/sqlalchemy_utils/types/arrow.py +++ b/sqlalchemy_utils/types/arrow.py @@ -1,3 +1,4 @@ +from arrow import Arrow from ..exceptions import ImproperlyConfigured from .enriched_datetime import ArrowDateTime from .enriched_datetime.enriched_datetime_type import EnrichedDateTimeType @@ -47,16 +48,15 @@ class Article(Base): # 'an hour ago' """ + cache_ok = True def __init__(self, *args, **kwargs): if not arrow: - raise ImproperlyConfigured( - "'arrow' package is required to use 'ArrowType'" - ) - - super().__init__( - datetime_processor=ArrowDateTime, - *args, - **kwargs - ) + raise ImproperlyConfigured("'arrow' package is required to use 'ArrowType'") + + super().__init__(datetime_processor=ArrowDateTime, *args, **kwargs) + + @property + def python_type(self): + return Arrow