From 29f54a408d4d435ef44e595687ee00f1acc6bd16 Mon Sep 17 00:00:00 2001
From: Vladless <curov2014@yandex.ru>
Date: Sat, 5 Apr 2025 20:48:13 +0300
Subject: [PATCH] add data_state

---
 handlers/keys/key_management.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/handlers/keys/key_management.py b/handlers/keys/key_management.py
index 5baf3a9b..e37aefcf 100644
--- a/handlers/keys/key_management.py
+++ b/handlers/keys/key_management.py
@@ -268,7 +268,11 @@ async def create_key(
             least_loaded_cluster,
             session,
         )
-        data = await state.get_data()
+
+        data = {}
+        if state:
+            data = await state.get_data()
+
         if data.get("is_trial"):
             trial_status = await get_trial(tg_id, session)
             if trial_status in [0, -1]: