From b257fa1884eab33f80764e234031331625d5d6e0 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Fri, 25 Oct 2024 15:22:48 -0700 Subject: [PATCH] Add a None check before doing a DT comparison when getting subscription type --- src/khoj/database/adapters/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index b77eccdd..6de89f9f 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -301,7 +301,7 @@ def subscription_to_state(subscription: Subscription) -> str: return SubscriptionState.INVALID.value elif subscription.type == Subscription.Type.TRIAL: # Check if the trial has expired - if datetime.now(tz=timezone.utc) > subscription.renewal_date: + if subscription.renewal_date and datetime.now(tz=timezone.utc) > subscription.renewal_date: return SubscriptionState.EXPIRED.value return SubscriptionState.TRIAL.value elif subscription.is_recurring and subscription.renewal_date > datetime.now(tz=timezone.utc):