Update app.py

app.py

@@ -685,7 +685,6 @@ def nutri_call():
 
 
 
-
 from tabulate import tabulate
 
 # Константы
@@ -696,8 +695,8 @@ VOLUME_LITERS = 100
 
 # Коэффициенты электропроводности
 EC_COEFFICIENTS = {
-    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015, 
-    'Ca': 0.0016, 'S': 0.0014, 
+    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015,
+    'Ca': 0.0016, 'S': 0.0014,
     'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
 }
 
@@ -723,13 +722,18 @@ class NutrientCalculator:
         self.results = {}
         self.target_profile = BASE_PROFILE.copy()
         self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
-        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
+        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS
         self.total_ec = 0.0
+        
+        # Добавляем константы как атрибуты
+        self.NO3_RATIO = NO3_RATIO
+        self.NH4_RATIO = NH4_RATIO
+        self.TOTAL_NITROGEN = TOTAL_NITROGEN
 
         # Расчёт азота
-        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)
+        total_parts = self.NO3_RATIO + self.NH4_RATIO
+        self.target_profile['N (NO3-)'] = self.TOTAL_NITROGEN * (self.NO3_RATIO / total_parts)
+        self.target_profile['N (NH4+)'] = self.TOTAL_NITROGEN * (self.NH4_RATIO / total_parts)
         self.initial_n_profile = {
             "NO3-": self.target_profile['N (NO3-)'],
             "NH4+": self.target_profile['N (NH4+)']
@@ -743,56 +747,35 @@ class NutrientCalculator:
         }
         return labels.get(element, element)
 
-
     def _balance_nitrogen(self):
-        # Целевые значения NH₄⁺ и NO₃⁻
+        """Балансировка азотных компонентов"""
         total_parts = self.NO3_RATIO + self.NH4_RATIO
         target_nh4 = (self.TOTAL_NITROGEN * self.NH4_RATIO) / total_parts
         target_no3 = self.TOTAL_NITROGEN - target_nh4
 
-        # Вклад NH₄NO₃
-        nh4no3_nh4 = target_nh4
-        nh4no3_no3 = nh4no3_nh4  # NH₄NO₃ вносит оба вида азота
-        self._apply("Аммоний азотнокислый", "N (NH4+)", nh4no3_nh4)
-
-        # Оставшийся NO₃⁻ (цель - вклад NH₄NO₃)
-        remaining_no3 = target_no3 - nh4no3_no3
-        if remaining_no3 > 0.1:
-            # Распределить между Ca(NO₃)₂ и KNO₃ (например, 70/30)
-            ca_no3_part = remaining_no3 * 0.7
-            k_no3_part = remaining_no3 * 0.3
-            self._apply("Кальциевая селитра", "N (NO3-)", ca_no3_part)
-            self._apply("Калий азотнокислый", "N (NO3-)", k_no3_part)
-
-
-    def calculate(self):
-        try:
-            # 1. Всё, что не связано с азотом (например, Mg)
-            self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
+        # Вносим NH₄⁺ через NH₄NO₃
+        self._apply("Аммоний азотнокислый", "N (NH4+)", target_nh4)
         
-            # 2. Балансировка азота (NH₄⁺/NO₃⁻)
-            self._balance_nitrogen()  # <-- Ваш новый блок
+        # Оставшийся NO₃⁻ распределяем
+        remaining_no3 = target_no3 - self.fertilizers["Аммоний азотнокислый"]["N (NO3-)"] * \
+                       (target_nh4 / self.fertilizers["Аммоний азотнокислый"]["N (NH4+)"])
         
-            # 3. Кальций и фосфор (после азота!)
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
-            self._apply("Монофосфат калия", "P", self.target_profile['P'])
-        
-            # 4. Калий и сера (коррекция)
-            self._apply_k_sulfate()
-        
-            # 5. Последние проверки (например, дефицит калия)
-            k_deficit = self.target_profile['K'] - self.actual_profile['K']
-            if k_deficit > 0.1:
-                self._apply("Калий азотнокислый", "K", k_deficit)
+        if remaining_no3 > 0.1:
+            # Сначала вносим через Ca(NO₃)₂ (для кальция)
+            ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
+            max_ca_no3 = ca_needed / self.fertilizers["Кальциевая селитра"]["Ca"] * \
+                        self.fertilizers["Кальциевая селитра"]["N (NO3-)"]
             
-            return self.results
-        except Exception as e:
-            print(f"Ошибка: {str(e)}")
-            raise        
-
-
+            ca_no3_part = min(remaining_no3, max_ca_no3)
+            self._apply("Кальциевая селитра", "N (NO3-)", ca_no3_part)
+            
+            # Остаток через KNO₃
+            remaining_no3 -= ca_no3_part
+            if remaining_no3 > 0.1:
+                self._apply("Калий азотнокислый", "N (NO3-)", remaining_no3)
 
     def _apply(self, fert_name, main_element, required_ppm):
+        """Применение удобрения с расчётом вносимых элементов"""
         if required_ppm <= 0:
             return
             
@@ -827,6 +810,7 @@ class NutrientCalculator:
             raise
 
     def _apply_k_sulfate(self):
+        """Коррекция калия и серы через сульфат калия"""
         fert = "Калий сернокислый"
         k_def = self.target_profile['K'] - self.actual_profile['K']
         s_def = self.target_profile['S'] - self.actual_profile['S']
@@ -835,27 +819,43 @@ class NutrientCalculator:
             return
             
         try:
+            # Всегда применяем по сере, если есть дефицит
             if s_def > 0.1:
-                s_content = self.fertilizers[fert]["S"]
-                grams_s = (s_def * self.volume) / (s_content * 1000)
-                
-                k_content = self.fertilizers[fert]["K"]
-                k_from_s = (grams_s * k_content * 1000) / self.volume
-                
-                if k_from_s > k_def and k_def > 0.1:
-                    grams = (k_def * self.volume) / (k_content * 1000)
-                else:
-                    grams = grams_s
-                    
                 self._apply(fert, "S", s_def)
+            # Отдельно применяем по калию, если остался дефицит
+            if k_def > 0.1:
+                self._apply(fert, "K", k_def)
         except Exception as e:
             print(f"Ошибка при расчёте сульфата калия: {str(e)}")
             raise
 
+    def calculate(self):
+        """Основной метод расчёта питательного раствора"""
+        try:
+            # 1. Всё, что не связано с азотом (например, Mg)
+            self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
+        
+            # 2. Балансировка азота (NH₄⁺/NO₃⁻)
+            self._balance_nitrogen()
+        
+            # 3. Кальций и фосфор (после азота!)
+            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+            self._apply("Монофосфат калия", "P", self.target_profile['P'])
+        
+            # 4. Калий и сера (коррекция)
+            self._apply_k_sulfate()
+            
+            return self.results
+        except Exception as e:
+            print(f"Ошибка: {str(e)}")
+            raise
+
     def calculate_ec(self):
+        """Расчёт общей электропроводности"""
         return round(self.total_ec, 2)
 
     def print_report(self):
+        """Вывод отчёта по расчёту"""
         try:
             print("\n" + "="*60)
             print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
@@ -884,7 +884,7 @@ class NutrientCalculator:
                     round(data['вклад в EC'], 3),
                     "\n".join(adds)
                 ])
-            print(tabulate(fert_table, 
+            print(tabulate(fert_table,
                          headers=["Удобрение", "Граммы", "Миллиграммы", "EC (мСм/см)", "Добавит"]))
 
             print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
@@ -906,7 +906,7 @@ if __name__ == "__main__":
     try:
         calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
         calculator.calculate()
-        calculator.print_report()  # Правильный вызов метода класса
+        calculator.print_report()
     except Exception as e:
         print(f"Критическая ошибка: {str(e)}")
 
@@ -916,10 +916,6 @@ if __name__ == "__main__":
 
 
 
-
-
-
-
 from flask import request, jsonify
 
 def round_floats(obj, ndigits=3):