Update app.py

app.py

@@ -692,41 +692,41 @@ class NutrientCalculator:
         self.results = {}
         self.target_profile = {
             'P': 0, 'K': 0, 'Mg': 0, 'Ca': 0, 'S': 0,
-            'N (NO3-)': 0, 'N (NH4+)': 0
+            'NO3': 0, 'NH4': 0  # Изменил ключи на NO3/NH4 без скобок
         }
         self.actual_profile = {
             'P': 0.0, 'K': 0.0, 'Mg': 0.0, 'Ca': 0.0, 'S': 0.0,
-            'N (NO3-)': 0.0, 'N (NH4+)': 0.0
+            'NO3': 0.0, 'NH4': 0.0
         }
         self.fertilizers = {}
         self.total_ec = 0.0
 
     def set_target_profile(self, profile_data):
-        """Устанавливаем целевые значения из запроса"""
+        """Устанавливаем целевые значения с преобразованием ключей"""
         self.target_profile.update({
             'P': float(profile_data.get('P', 0)),
             'K': float(profile_data.get('K', 0)),
             'Mg': float(profile_data.get('Mg', 0)),
             'Ca': float(profile_data.get('Ca', 0)),
             'S': float(profile_data.get('S', 0)),
-            'N (NO3-)': float(profile_data.get('N (NO3-)', 0)),  # Берём как есть
-            'N (NH4+)': float(profile_data.get('N (NH4+)', 0))   # Без пересчёта
+            'NO3': float(profile_data.get('N (NO3-)', 0)),  # Преобразуем ключ
+            'NH4': float(profile_data.get('N (NH4+)', 0))   # Преобразуем ключ
         })
 
     def set_fertilizers(self, fertilizers_data):
-        """Устанавливаем состав удобрений из входных данных"""
+        """Устанавливаем состав удобрений с преобразованием ключей"""
         self.fertilizers = {
             "Кальциевая селитра": {
-                "N (NO3-)": float(fertilizers_data["Кальциевая селитра"]["N (NO3-)"]),
+                "NO3": float(fertilizers_data["Кальциевая селитра"]["N (NO3-)"]),
                 "Ca": float(fertilizers_data["Кальциевая селитра"]["Ca"])
             },
             "Калий азотнокислый": {
-                "N (NO3-)": float(fertilizers_data["Калий азотнокислый"]["N (NO3-)"]),
+                "NO3": float(fertilizers_data["Калий азотнокислый"]["N (NO3-)"]),
                 "K": float(fertilizers_data["Калий азотнокислый"]["K"])
             },
             "Аммоний азотнокислый": {
-                "N (NO3-)": float(fertilizers_data["Аммоний азотнокислый"]["N (NO3-)"]),
-                "N (NH4+)": float(fertilizers_data["Аммоний азотнокислый"]["N (NH4+)"])
+                "NO3": float(fertilizers_data["Аммоний азотнокислый"]["N (NO3-)"]),
+                "NH4": float(fertilizers_data["Аммоний азотнокислый"]["N (NH4+)"])
             },
             "Сульфат магния": {
                 "Mg": float(fertilizers_data["Сульфат магния"]["Mg"]),
@@ -743,31 +743,32 @@ class NutrientCalculator:
         }
 
     def calculate(self):
-        """Оригинальная логика расчёта без изменений"""
+        """Логика расчёта с приоритетом по нитратам"""
         try:
             # 1. Вносим сульфат магния (Mg + S)
             self._apply_fertilizer("Сульфат магния", "Mg")
             
-            # 2. Вносим кальциевую селитру (Ca + NO3)
-            self._apply_fertilizer("Кальциевая селитра", "Ca")
-            
-            # 3. Вносим монофосфат калия (P + K)
+            # 2. Вносим монофосфат калия (P + K)
             self._apply_fertilizer("Монофосфат калия", "P")
             
-            # 4. Вносим аммоний азотнокислый (NH4 + NO3)
-            self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)")
+            # 3. Вносим аммоний азотнокислый (NH4 + NO3)
+            self._apply_fertilizer("Аммоний азотнокислый", "NH4")
             
-            # 5. Довносим NO3 через калийную селитру при необходимости
-            no3_deficit = self.target_profile['N (NO3-)'] - self.actual_profile['N (NO3-)']
-            if no3_deficit > 0.1:
-                self._apply_fertilizer("Калий азотнокислый", "N (NO3-)", no3_deficit)
+            # 4. Жёстко контролируем NO3:
+            #    - Если нужно меньше, чем уже внесено, пропускаем кальциевую селитру
+            if self.target_profile['NO3'] > self.actual_profile['NO3']:
+                self._apply_fertilizer("Кальциевая селитра", "Ca")
             
-            # 6. Довносим калий при необходимости
+            # 5. Довносим K через калийную селитру или сульфат калия
             k_deficit = self.target_profile['K'] - self.actual_profile['K']
             if k_deficit > 0.1:
-                self._apply_fertilizer("Калий азотнокислый", "K", k_deficit)
+                # Выбираем удобрение с минимальным NO3
+                if self.actual_profile['NO3'] < self.target_profile['NO3']:
+                    self._apply_fertilizer("Калий азотнокислый", "K", k_deficit)
+                else:
+                    self._apply_fertilizer("Калий сернокислый", "K", k_deficit)
             
-            # 7. Довносим серу через сульфат калия при необходимости
+            # 6. Довносим серу при необходимости
             s_deficit = self.target_profile['S'] - self.actual_profile['S']
             if s_deficit > 0.1:
                 self._apply_fertilizer("Калий сернокислый", "S", s_deficit)
@@ -778,7 +779,7 @@ class NutrientCalculator:
             return False
 
     def _apply_fertilizer(self, name, main_element, required_ppm=None):
-        """Метод внесения удобрений без изменений"""
+        """Метод внесения удобрений с контролем NO3"""
         if required_ppm is None:
             required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
         
@@ -789,6 +790,12 @@ class NutrientCalculator:
         if content <= 0:
             return
 
+        # Корректируем граммы, если NO3 превышает целевое значение
+        if 'NO3' in self.fertilizers[name]:
+            no3_added = (required_ppm / content) * self.fertilizers[name]['NO3']
+            if self.actual_profile['NO3'] + no3_added > self.target_profile['NO3']:
+                required_ppm = max(0, (self.target_profile['NO3'] - self.actual_profile['NO3']) * content / self.fertilizers[name]['NO3'])
+
         grams = (required_ppm * self.volume) / (content * 1000)
         
         # Запись результатов
@@ -807,7 +814,7 @@ class NutrientCalculator:
         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
+            'NO3': 0.0017, 'NH4': 0.0019
         }
         self.total_ec += sum(
             added_ppm * ec_coefficients.get(element, 0.0015)
@@ -815,7 +822,7 @@ class NutrientCalculator:
         )
 
     def get_results(self):
-        """Форматирование результатов без изменений"""
+        """Форматирование результатов с обратным преобразованием ключей"""
         return {
             'fertilizers': [
                 {
@@ -826,13 +833,14 @@ class NutrientCalculator:
             ],
             'profile': [
                 {
-                    'element': k.replace(' (NO3-)', '-NO3').replace(' (NH4+)', '-NH4'),
+                    'element': 'N-NO3' if k == 'NO3' else 'N-NH4' if k == 'NH4' else k,
                     'ppm': round(v, 3)
                 } for k, v in self.actual_profile.items()
             ],
             'ec': round(self.total_ec, 2),
             'deficits': {
-                k: max(self.target_profile[k] - self.actual_profile.get(k, 0), 0)
+                'N (NO3-)' if k == 'NO3' else 'N (NH4+)' if k == 'NH4' else k: 
+                max(self.target_profile[k] - self.actual_profile.get(k, 0), 0)
                 for k in self.target_profile
             }
         }
@@ -894,6 +902,5 @@ def handle_calculation():
 
 
 
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))