Update app.py

app.py

@@ -841,6 +841,83 @@ class NutrientCalculator:
             "CALCIUM_NITRATE": {"weight": 0.25, "fert": "Кальциевая селитра", "main_element": "Ca"},
             "CALCIUM_NITRATE_NO3": {"weight": 0.15, "fert": "Кальциевая селитра", "main_element": "N (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 = {
+            'N (NO3-)': 'NO3',
+            'N (NH4+)': 'NH4'
+        }
+        return labels.get(element, element)
+
+    def calculate(self):
+        try:
+            # Вносим магний через компенсацию
+            self._compensate_element("Mg")
+
+            # Вносим кальций напрямую
+            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+
+            # Вносим фосфор через компенсацию
+            self._compensate_element("P")
+
+            # Вносим аммонийный азот напрямую
+            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+
+            # Компенсируем нитратный азот напрямую
+            current_no3 = self.actual_profile['N (NO3-)']
+            no3_needed = self.target_profile['N (NO3-)'] - current_no3
+            if no3_needed > 0.1:
+                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+
+            # Компенсируем серу через компенсацию
+            self._compensate_element("S")
+
+            # Компенсируем калий через компенсацию
+            self._compensate_element("K")
+
+            return self.results
+        except Exception as e:
+            print(f"Ошибка при расчёте: {str(e)}")
+            raise
+
+    def _apply(self, fert_name, main_element, required_ppm):
+        if required_ppm <= 0:
+            return
+
+        try:
+            content = self.fertilizers[fert_name][main_element]
+            grams = (required_ppm * self.volume) / (content * 1000)
+
+            if fert_name not in self.results:
+                result = {
+                    'граммы': 0.0,
+                    'миллиграммы': 0,
+                    'вклад в EC': 0.0
+                }
+                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 _compensate_element(self, element):
         needed = self.target_profile[element] - self.actual_profile.get(element, 0)
@@ -893,37 +970,6 @@ class NutrientCalculator:
             ppm_to_apply = needed * share
             self._apply(candidate['name'], element, ppm_to_apply)
 
-    # ... остальной код ...
-
-    def calculate(self):
-        try:
-            # Вносим магний через компенсацию
-            self._compensate_element("Mg")
-
-            # Теперь вносим кальций через компенсацию, а не напрямую
-            self._compensate_element("Ca")
-
-            # Вносим фосфор через компенсацию
-            self._compensate_element("P")
-
-            # Вносим аммонийный азот напрямую
-            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
-
-            # Компенсируем нитратный азот через компенсацию
-            self._compensate_element("N (NO3-)")
-
-            # Компенсируем серу через компенсацию
-            self._compensate_element("S")
-
-            # Компенсируем калий через компенсацию
-            self._compensate_element("K")
-
-            return self.results
-        except Exception as e:
-            print(f"Ошибка при расчёте: {str(e)}")
-            raise
-
-
 
     def calculate_ec(self):
         return round(self.total_ec, 2)