Update app.py

app.py

@@ -18,7 +18,7 @@ import numpy as np
 from tabulate import tabulate
 
 from typing import Dict, Any, Union, List
-
+import itertools
 
 import traceback
 import logging
@@ -717,7 +717,8 @@ NUTRIENT_CONTENT_IN_FERTILIZERS = {
     "Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499},
     "Сульфат магния": {"Mg": 0.09861, "S": 0.13010},
     "Монофосфат калия": {"P": 0.218, "K": 0.275},
-    "Сульфат кальция": {"Ca": 0.23, "S": 0.186}
+    "Сульфат кальция": {"Ca": 0.23, "S": 0.186},
+    "Кольцевая селитра": {"N (NO3-)": 0.15, "Ca": 0.20}  # Новое удобрение
 }
 
 EC_COEFFICIENTS = {
@@ -726,6 +727,8 @@ EC_COEFFICIENTS = {
     'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
 }
 
+
+
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
@@ -746,16 +749,8 @@ class NutrientCalculator:
             "NH4+": self.target_profile['N (NH4+)']
         }
 
-        # Веса компенсации
-        self.compensation_weights = {
-            "Ca": {"weight": 0.3, "fert": "Сульфат кальция", "main_element": "Ca"},
-            "K": {"weight": 0.2, "fert": "Калий азотнокислый", "main_element": "K"},
-            "Mg": {"weight": 0.2, "fert": "Сульфат магния", "main_element": "Mg"},
-            "P": {"weight": 0.1, "fert": "Монофосфат калия", "main_element": "P"},
-            "S": {"weight": 0.1, "fert": "Калий сернокислый", "main_element": "S"},
-            "N (NO3-)": {"weight": 0.05, "fert": "Калий азотнокислый", "main_element": "N (NO3-)"},
-            "N (NH4+)": {"weight": 0.05, "fert": "Аммоний азотнокислый", "main_element": "N (NH4+)"}
-        }
+        # Список всех удобрений
+        self.fertilizer_list = list(self.fertilizers.keys())
 
     def _label(self, element):
         """Форматирование названий элементов для вывода"""
@@ -767,103 +762,71 @@ class NutrientCalculator:
 
     def calculate(self):
         try:
-            max_iterations = 100  # Максимальное количество итераций
-            iteration = 0
-            while iteration < max_iterations:
-                iteration += 1
-                print(f"Итерация {iteration} из {max_iterations}")
+            max_attempts = 100  # Максимальное количество попыток
+            attempt = 0
+            while attempt < max_attempts:
+                attempt += 1
+                print(f"Попытка {attempt} из {max_attempts}")
+
+                # Генерация случайного порядка удобрений
+                fertilizer_order = self._shuffle_fertilizers()
 
                 # Сброс фактического профиля
                 self._reset_actual_profile()
 
-                # Применение удобрений
-                self._apply_fertilizers()
-
-                # Проверка результатов
-                if self.validate_results():
+                # Пробуем рассчитать с текущим порядком удобрений
+                if self._try_combination(fertilizer_order):
                     print("Успешная комбинация найдена!")
                     return self.results
 
-                # Корректировка перебора
-                self._adjust_overages()
-
-            raise ValueError("Не удалось найти подходящую комбинацию удобрений после максимального количества итераций.")
+            raise ValueError("Не удалось найти подходящую комбинацию удобрений после максимального количества попыток.")
 
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
+    def _shuffle_fertilizers(self):
+        """Генерация случайного порядка удобрений"""
+        return list(itertools.permutations(self.fertilizer_list))[hash(str(self.target_profile)) % len(self.fertilizer_list)]
+
     def _reset_actual_profile(self):
-        """Сброс фактического профиля перед новой итерацией"""
+        """Сброс фактического профиля перед новой попыткой"""
         self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
         self.total_ec = 0.0
         self.results = {}
 
-    def _apply_fertilizers(self):
-        """Применение удобрений"""
-        for fert_name, fert_data in self.fertilizers.items():
-            for element, content in fert_data.items():
+    def _try_combination(self, fertilizer_order):
+        """Попытка расчета с заданным порядко�� удобрений"""
+        for fert_name in fertilizer_order:
+            for element, content in self.fertilizers[fert_name].items():
                 required_ppm = self.target_profile[element] - self.actual_profile[element]
                 if required_ppm > 0.1:
-                    grams = min((required_ppm * self.volume) / (content * 1000), 100)  # Ограничение на граммы
-                    self._apply_with_limit(fert_name, element, grams, content)
-
-    def _apply_with_limit(self, fert_name, element, grams, content):
-        """Применение удобрения с ограничением"""
-        if fert_name not in self.results:
-            self.results[fert_name] = {
-                'граммы': 0.0,
-                'миллиграммы': 0,
-                'вклад в EC': 0.0
-            }
-            for elem in self.fertilizers[fert_name]:
-                self.results[fert_name][f'внесет {self._label(elem)}'] = 0.0
-
-        self.results[fert_name]['граммы'] += grams
-        self.results[fert_name]['миллиграммы'] += int(grams * 1000)
-
-        fert_ec = 0.0
-        for elem, percent in self.fertilizers[fert_name].items():
-            added_ppm = (grams * percent * 1000) / self.volume
-            self.results[fert_name][f'внесет {self._label(elem)}'] += added_ppm
-            self.actual_profile[elem] += added_ppm
-            fert_ec += added_ppm * EC_COEFFICIENTS.get(elem, 0.0015)
-
-        self.results[fert_name]['вклад в EC'] += fert_ec
-        self.total_ec += fert_ec
-
-    def _adjust_overages(self):
-        """Корректировка перебора элементов"""
-        for element in self.actual_profile:
-            if self.actual_profile[element] > self.target_profile[element]:
-                overage = self.actual_profile[element] - self.target_profile[element]
-                print(f"Корректировка перебора: {element} уменьшен на {overage:.2f} ppm")
-
-                # Уменьшаем фактическое значение
-                self.actual_profile[element] -= overage
-
-                # Обновляем результаты удобрений
-                for fert_name, fert_data in self.results.items():
-                    if f'внесет {self._label(element)}' in fert_data:
-                        current_value = fert_data[f'внесет {self._label(element)}']
-                        if current_value > overage:
-                            fert_data[f'внесет {self._label(element)}'] -= overage
-                            overage = 0
-                        else:
-                            overage -= current_value
-                            fert_data[f'внесет {self._label(element)}'] = 0
-
-                        # Пересчитываем граммы удобрения
-                        if element in self.fertilizers[fert_name]:
-                            content = self.fertilizers[fert_name][element]
-                            grams_to_remove = (current_value * self.volume) / (content * 1000)
-                            fert_data['граммы'] -= grams_to_remove
-                            fert_data['миллиграммы'] -= int(grams_to_remove * 1000)
-
-                        # Убеждаемся, что граммы не отрицательны
-                        if fert_data['граммы'] < 0:
-                            fert_data['граммы'] = 0
-                            fert_data['миллиграммы'] = 0
+                    grams = (required_ppm * self.volume) / (content * 1000)
+
+                    if fert_name not in self.results:
+                        self.results[fert_name] = {
+                            'граммы': 0.0,
+                            'миллиграммы': 0,
+                            'вклад в EC': 0.0
+                        }
+                        for elem in self.fertilizers[fert_name]:
+                            self.results[fert_name][f'внесет {self._label(elem)}'] = 0.0
+
+                    self.results[fert_name]['граммы'] += grams
+                    self.results[fert_name]['миллиграммы'] += int(grams * 1000)
+
+                    fert_ec = 0.0
+                    for elem, percent in self.fertilizers[fert_name].items():
+                        added_ppm = (grams * percent * 1000) / self.volume
+                        self.results[fert_name][f'внесет {self._label(elem)}'] += added_ppm
+                        self.actual_profile[elem] += added_ppm
+                        fert_ec += added_ppm * EC_COEFFICIENTS.get(elem, 0.0015)
+
+                    self.results[fert_name]['вклад в EC'] += fert_ec
+                    self.total_ec += fert_ec
+
+        # Проверка результатов
+        return self.validate_results()
 
     def validate_results(self):
         """Проверка соответствия целевым значениям"""