From dcb101d19df077b5b9f8e932b599343a41e22c14 Mon Sep 17 00:00:00 2001
From: Capybara-z <stas89091000@yandex.ru>
Date: Tue, 27 May 2025 00:15:36 +0300
Subject: [PATCH] tariff lookup for country model
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Определение тарифа для режима стран при продлении ключей.
---
 database/tariffs.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/database/tariffs.py b/database/tariffs.py
index dd81fd5a..3d4aae07 100644
--- a/database/tariffs.py
+++ b/database/tariffs.py
@@ -45,6 +45,15 @@ async def get_tariffs_for_cluster(session: AsyncSession, cluster_name: str):
             .limit(1)
         )
         row = server_row.first()
+        
+        if not row:
+            server_row = await session.execute(
+                select(Server.tariff_group)
+                .where(Server.server_name == cluster_name)
+                .limit(1)
+            )
+            row = server_row.first()
+            
         if not row or not row[0]:
             return []