Update app.py

app.py

@@ -696,69 +696,64 @@ def nutri_call():
 
 from tabulate import tabulate
 
-# Глобальные параметры
-TOTAL_NITROGEN = 120.0 # Общее количество азота
-NO3_RATIO = 8.0         # Соотношение NO3:NH4
-NH4_RATIO = 1.00          # Соотношение NH4:NO3
-VOLUME_LITERS = 100       # Объем раствора
+# Константы
+TOTAL_NITROGEN = 125.000
+NO3_RATIO = 8.25
+NH4_RATIO = 1.00
+VOLUME_LITERS = 100
 
+# Коэффициенты электропроводности
+EC_COEFFICIENTS = {
+    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015, 
+    'Ca': 0.0016, 'S': 0.0014, 
+    'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
+}
+
+# Базовый профиль питательного раствора
 BASE_PROFILE = {
-    "P": 50,      # Фосфор
-    "K": 300,     # Калий
-    "Mg": 120,    # Магний (высокий уровень)
-    "Ca": 150,    # Кальций
-    "S": 100,     # Сера
-    "N (NO3-)": 0,  # Рассчитывается автоматически
-    "N (NH4+)": 0   # Рассчитывается автоматически
+    'P': 31.000, 'K': 210.000, 'Mg': 24.000,
+    'Ca': 84.000, 'S': 56.439,
+    'N (NO3-)': 0, 'N (NH4+)': 0
 }
 
+# Содержание элементов в удобрениях
 NUTRIENT_CONTENT_IN_FERTILIZERS = {
     "Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca": 0.16972},
     "Калий азотнокислый": {"N (NO3-)": 0.136, "K": 0.382},
-    "Калий сернокислый": {"K": 0.44874, "S": 0.18401},
     "Аммоний азотнокислый": {"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}
-}
-
-EC_COEFFICIENTS = {
-    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015, 
-    'Ca': 0.0016, 'S': 0.0014, 
-    'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
+    "Калий сернокислый": {"K": 0.44874, "S": 0.18401}
 }
 
 class NutrientCalculator:
-    def __init__(self, volume_liters=1.0, profile=BASE_PROFILE):
+    def __init__(self, volume_liters=1.0):
+        # Общие параметры
         self.volume = volume_liters
         self.results = {}
-        self.target_profile = profile.copy()
-        self.actual_profile = {k: 0.0 for k in self.target_profile}
-        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
+        self.target_profile = BASE_PROFILE.copy()
+        self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
+        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS
         self.total_ec = 0.0
 
-        # Расчёт азота
-        total_parts = NO3_RATIO + NH4_RATIO
-        self.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
-        self.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
-
-        # Сохраняем исходный профиль азота
-        self.initial_n_profile = {
-            "NO3-": self.target_profile['N (NO3-)'],
-            "NH4+": self.target_profile['N (NH4+)']
+        # Веса компенсации для элементов
+        self.element_compensation_weights = {
+            "POTASSIUM_SULFATE": {"weight": 0.18, "fert": "Калий сернокислый", "main_element": "K"},
+            "MAGNESIUM_SULFATE": {"weight": -0.08, "fert": "Сульфат магния", "main_element": "Mg"},
+            "MONOPOTASSIUM_PHOSPHATE": {"weight": 0.14, "fert": "Монофосфат калия", "main_element": "P"}
         }
 
-        # Веса компенсации
-        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.calcium_nitrate_ratios = {
+            "Ca": 0.7,  # 70% дефицита кальция покрываем кальциевой селитрой
+            "NO3": 0.2  # 30% дефицита NO3 покрываем кальциевой селитрой
         }
 
+        # Расчёт соотношений азотов
+        total_parts = NO3_RATIO + NH4_RATIO
+        self.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
+        self.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
+
     def _label(self, element):
         """Форматирование названий элементов для вывода"""
         labels = {
@@ -768,62 +763,84 @@ class NutrientCalculator:
         return labels.get(element, element)
 
     def calculate(self):
+        """Основной метод расчета"""
         try:
-            # Первый проход: компенсация основных элементов
-            self._compensate_main_elements()
+            # 1. Вносим кальциевую селитру для Ca и частично для NO3
+            self._apply_calcium_nitrate()
 
-            # Второй проход: компенсация азота
-            self._compensate_nitrogen()
+            # 2. Вносим аммонийный азот напрямую
+            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
 
-            # Третий проход: компенсация второстепенных элементов
-            self._compensate_secondary_elements()
+            # 3. Компенсируем остаток NO3 через калийную селитру
+            self._compensate_no3()
 
-            # Четвертый проход: корректировка перебора
-            self._adjust_overages()
+            # 4. Компенсируем остальные элементы через веса компенсации
+            self._compensate_element("Mg")
+            self._compensate_element("P")
+            self._compensate_element("S")
+            self._compensate_element("K")
 
             return self.results
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
-    def _compensate_main_elements(self):
-        """Компенсация основных элементов (Ca, Mg, P)"""
-        for element, weight_data in self.compensation_weights.items():
-            if element in ["Ca", "Mg", "P"]:
-                fert_name = weight_data["fert"]
-                main_element = weight_data["main_element"]
-                required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
-                if required_ppm > 0.1:
-                    self._apply_with_limit(fert_name, main_element, required_ppm)
-
-    def _compensate_nitrogen(self):
-        """Компенсация азота (NO3-, NH4+)"""
-        for nitrogen_type in ["N (NO3-)", "N (NH4+)"]:
-            required_ppm = self.target_profile[nitrogen_type] - self.actual_profile[nitrogen_type]
-            if required_ppm > 0.1:
-                fert_name = self.compensation_weights[nitrogen_type]["fert"]
-                self._apply_with_limit(fert_name, nitrogen_type, required_ppm)
-
-    def _compensate_secondary_elements(self):
-        """Компенсация второстепенных элементов (K, S)"""
-        for element, weight_data in self.compensation_weights.items():
-            if element in ["K", "S"]:
-                fert_name = weight_data["fert"]
-                main_element = weight_data["main_element"]
-                required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
-                if required_ppm > 0.1:
-                    self._apply_with_limit(fert_name, main_element, required_ppm)
-
-    def _apply_with_limit(self, fert_name, main_element, required_ppm):
-        """Применение удобрения с ограничением по перебору"""
+    def _apply_calcium_nitrate(self):
+        """Специальный метод для внесения кальциевой селитры"""
+        # Вносим кальций
+        ca_needed = self.target_profile['Ca'] - self.actual_profile.get('Ca', 0)
+        if ca_needed > 0.1:
+            ca_share = self.calcium_nitrate_ratios['Ca']
+            ca_to_apply = ca_needed * ca_share
+            self._apply("Кальциевая селитра", "Ca", ca_to_apply)
+
+        # Вносим часть NO3 из кальциевой селитры
+        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
+        if no3_needed > 0.1:
+            no3_share = self.calcium_nitrate_ratios['NO3']
+            no3_to_apply = no3_needed * no3_share
+            self._apply("Кальциевая селитра", "N (NO3-)", no3_to_apply)
+
+    def _compensate_no3(self):
+        """Компенсация остатка NO3 после внесения кальциевой селитры"""
+        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
+        if no3_needed > 0.1:
+            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+
+    def _compensate_element(self, element):
+        """Общий метод компенсации для других элементов"""
+        needed = self.target_profile[element] - self.actual_profile.get(element, 0)
+        if abs(needed) < 0.1:
+            return
+
+        candidates = []
+        for weight_key, weight_data in self.element_compensation_weights.items():
+            fert_name = weight_data["fert"]
+            if element in self.fertilizers.get(fert_name, {}):
+                candidates.append({
+                    'name': fert_name,
+                    'weight': weight_data["weight"],
+                    'content': self.fertilizers[fert_name][element]
+                })
+
+        if not candidates:
+            raise ValueError(f"Нет удобрений для элемента {element}")
+
+        total_weight = sum(c['weight'] for c in candidates)
+        for candidate in candidates:
+            share = candidate['weight'] / total_weight
+            ppm_to_apply = needed * share
+            self._apply(candidate['name'], element, ppm_to_apply)
+
+    def _apply(self, fert_name, main_element, required_ppm):
+        """Применение удобрения для конкретного элемента"""
         if required_ppm <= 0:
             return
 
         try:
             content = self.fertilizers[fert_name][main_element]
-            max_allowed_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
-            grams = min((required_ppm * self.volume) / (content * 1000), (max_allowed_ppm * self.volume) / (content * 1000))
-            
+            grams = (required_ppm * self.volume) / (content * 1000)
+
             if fert_name not in self.results:
                 result = {
                     'граммы': 0.0,
@@ -833,43 +850,29 @@ class NutrientCalculator:
                 for element in self.fertilizers[fert_name]:
                     result[f'внесет {self._label(element)}'] = 0.0
                 self.results[fert_name] = result
-                
+
             self.results[fert_name]['граммы'] += grams
             self.results[fert_name]['миллиграммы'] += int(grams * 1000)
-            
+
             fert_ec = 0.0
             for element, percent in self.fertilizers[fert_name].items():
                 added_ppm = (grams * percent * 1000) / self.volume
                 self.results[fert_name][f'внесет {self._label(element)}'] += added_ppm
                 self.actual_profile[element] += added_ppm
                 fert_ec += added_ppm * EC_COEFFICIENTS.get(element, 0.0015)
-                
+
             self.results[fert_name]['вклад в EC'] += fert_ec
             self.total_ec += fert_ec
         except KeyError as e:
             print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
             raise
 
-    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]
-                self.actual_profile[element] -= overage
-                print(f"Корректировка перебора: {element} уменьшен на {overage:.2f} ppm")
-
     def calculate_ec(self):
+        """Расчет общей электропроводности (EC)"""
         return round(self.total_ec, 2)
 
-    def print_initial_nitrogen_report(self):
-        try:
-            print("Исходный расчёт азота:")
-            print(f"  NO3-: {self.initial_n_profile['NO3-']} ppm")
-            print(f"  NH4+: {self.initial_n_profile['NH4+']} ppm")
-        except Exception as e:
-            print(f"Ошибка при выводе отчёта: {str(e)}")
-            raise
     def print_report(self):
+        """Вывод отчета о расчетах"""
         try:
             print("\n" + "="*60)
             print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
@@ -877,10 +880,6 @@ class NutrientCalculator:
             table = [[el, round(self.actual_profile[el], 1)] for el in self.actual_profile]
             print(tabulate(table, headers=["Элемент", "ppm"]))
 
-            print("\nИсходный расчёт азота:")
-            for form, val in self.initial_n_profile.items():
-                print(f"  {form}: {round(val, 1)} ppm")
-
             print("\n" + "="*60)
             print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
             print("="*60)
@@ -916,6 +915,7 @@ class NutrientCalculator:
             print(f"Ошибка при выводе отчёта: {str(e)}")
             raise
 
+
 if __name__ == "__main__":
     try:
         calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)