Update app.py

app.py

@@ -686,7 +686,6 @@ def nutri_call():
 
 
 
-
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
@@ -703,77 +702,120 @@ class NutrientCalculator:
         self.total_ec = 0.0
 
     def set_target_profile(self, profile_data):
-        """Устанавливаем целевые значения"""
-        for key in profile_data:
-            if key in self.target_profile:
-                self.target_profile[key] = float(profile_data[key])
+        """Устанавливаем целевые значения из запроса"""
+        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))   # Без пересчёта
+        })
 
     def set_fertilizers(self, fertilizers_data):
-        """Принимаем удобрения в любом формате (рус/англ)"""
-        self.fertilizers = fertilizers_data
+        """Устанавливаем состав удобрений из входных данных"""
+        self.fertilizers = {
+            "Кальциевая селитра": {
+                "N (NO3-)": float(fertilizers_data["Кальциевая селитра"]["N (NO3-)"]),
+                "Ca": float(fertilizers_data["Кальциевая селитра"]["Ca"])
+            },
+            "Калий азотнокислый": {
+                "N (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+)"])
+            },
+            "Сульфат магния": {
+                "Mg": float(fertilizers_data["Сульфат магния"]["Mg"]),
+                "S": float(fertilizers_data["Сульфат магния"]["S"])
+            },
+            "Монофосфат калия": {
+                "P": float(fertilizers_data["Монофосфат калия"]["P"]),
+                "K": float(fertilizers_data["Монофосфат калия"]["K"])
+            },
+            "Калий сернокислый": {
+                "K": float(fertilizers_data["Калий сернокислый"]["K"]),
+                "S": float(fertilizers_data["Калий сернокислый"]["S"])
+            }
+        }
 
     def calculate(self):
-        """Основная логика расчёта без изменений"""
+        """Оригинальная логика расчёта без изменений"""
         try:
-            # Порядок внесения удобрений
-            self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
-            self._apply("Монофосфат калия", "P", self.target_profile['P'])
-            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+            # 1. Вносим сульфат магния (Mg + S)
+            self._apply_fertilizer("Сульфат магния", "Mg")
+            
+            # 2. Вносим кальциевую селитру (Ca + NO3)
+            self._apply_fertilizer("Кальциевая селитра", "Ca")
             
-            # Довносим NO3 если нужно
-            no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile['N (NO3-)']
-            if no3_needed > 0.1:
-                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+            # 3. Вносим монофосфат калия (P + K)
+            self._apply_fertilizer("Монофосфат калия", "P")
             
-            # Довносим калий если нужно
+            # 4. Вносим аммоний азотнокислый (NH4 + NO3)
+            self._apply_fertilizer("Аммоний азотнокислый", "N (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)
+            
+            # 6. Довносим калий при необходимости
             k_deficit = self.target_profile['K'] - self.actual_profile['K']
             if k_deficit > 0.1:
-                self._apply("Калий азотнокислый", "K", k_deficit)
+                self._apply_fertilizer("Калий азотнокислый", "K", k_deficit)
+            
+            # 7. Довносим серу через сульфат калия при необходимости
+            s_deficit = self.target_profile['S'] - self.actual_profile['S']
+            if s_deficit > 0.1:
+                self._apply_fertilizer("Калий сернокислый", "S", s_deficit)
             
             return True
         except Exception as e:
             print(f"Ошибка расчёта: {str(e)}")
             return False
 
-    def _apply(self, fert_name, element, required_ppm):
-        """Универсальный метод внесения удобрений"""
-        if fert_name not in self.fertilizers or required_ppm <= 0:
+    def _apply_fertilizer(self, name, main_element, required_ppm=None):
+        """Метод внесения удобрений без изменений"""
+        if required_ppm is None:
+            required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
+        
+        if required_ppm <= 0 or name not in self.fertilizers:
             return
 
-        content = self.fertilizers[fert_name].get(element, 0)
+        content = self.fertilizers[name].get(main_element, 0)
         if content <= 0:
             return
 
         grams = (required_ppm * self.volume) / (content * 1000)
         
-        # Сохраняем результаты
-        if fert_name not in self.results:
-            self.results[fert_name] = {
-                'grams': 0.0,
-                'adds': {}
-            }
+        # Запись результатов
+        if name not in self.results:
+            self.results[name] = {'grams': 0.0, 'adds': {}}
         
-        self.results[fert_name]['grams'] += grams
+        self.results[name]['grams'] += grams
         
-        # Обновляем профиль
-        for nutrient, percent in self.fertilizers[fert_name].items():
+        # Обновление профиля
+        for element, percent in self.fertilizers[name].items():
             added_ppm = (grams * percent * 1000) / self.volume
-            self.actual_profile[nutrient] = self.actual_profile.get(nutrient, 0) + added_ppm
-            self.results[fert_name]['adds'][nutrient] = self.results[fert_name]['adds'].get(nutrient, 0) + added_ppm
+            self.actual_profile[element] += added_ppm
+            self.results[name]['adds'][element] = self.results[name]['adds'].get(element, 0) + added_ppm
         
         # Расчёт EC
+        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
+        }
         self.total_ec += sum(
-            added_ppm * {
-                'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015,
-                'Ca': 0.0016, 'S': 0.0014,
-                'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
-            }.get(nutrient, 0.0015)
-            for nutrient, added_ppm in self.results[fert_name]['adds'].items()
+            added_ppm * ec_coefficients.get(element, 0.0015)
+            for element, added_ppm in self.results[name]['adds'].items()
         )
 
     def get_results(self):
-        """Форматируем результаты"""
+        """Форматирование результатов без изменений"""
         return {
             'fertilizers': [
                 {
@@ -800,17 +842,21 @@ def handle_calculation():
     try:
         data = request.json
         
-        # Проверяем обязательные поля
-        if 'fertilizerConstants' not in data or 'profileSettings' not in data:
-            return jsonify({'status': 'error', 'message': 'Missing required fields'}), 400
+        # Валидация входных данных
+        required_fields = ['P', 'K', 'Mg', 'Ca', 'S', 'N (NO3-)', 'N (NH4+)', 'liters']
+        if not all(field in data['profileSettings'] for field in required_fields):
+            return jsonify({'status': 'error', 'message': 'Missing required profile fields'}), 400
         
-        # Создаём калькулятор
-        calculator = NutrientCalculator(volume_liters=data['profileSettings']['liters'])
+        required_fertilizers = [
+            "Кальциевая селитра", "Калий азотнокислый", "Аммоний азотнокислый",
+            "Сульфат магния", "Монофосфат калия", "Калий сернокислый"
+        ]
+        if not all(fert in data['fertilizerConstants'] for fert in required_fertilizers):
+            return jsonify({'status': 'error', 'message': 'Missing required fertilizers'}), 400
         
-        # Настраиваем профиль (принимаем как есть)
+        # Иниц��ализация калькулятора
+        calculator = NutrientCalculator(volume_liters=data['profileSettings']['liters'])
         calculator.set_target_profile(data['profileSettings'])
-        
-        # Настраиваем удобрения (принимаем как есть, без преобразований)
         calculator.set_fertilizers(data['fertilizerConstants'])
         
         # Расчёт
@@ -828,9 +874,8 @@ def handle_calculation():
     except Exception as e:
         return jsonify({
             'status': 'error',
-            'message': str(e)
-        }), 400
-
+            'message': f"Server error: {str(e)}"
+        }), 500