Update app.py

app.py

@@ -982,65 +982,97 @@ if __name__ == "__main__":
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try:
-        # Получаем данные как есть
         data = request.json
-        
-        # Создаем калькулятор
-        calculator = NutrientCalculator(volume_liters=data['profileSettings']['liters'])
-        
-        # Устанавливаем целевой профиль из полученных данных
-        calculator.target_profile = {
-            'P': float(data['profileSettings']['P']),
-            'K': float(data['profileSettings']['K']),
-            'Mg': float(data['profileSettings']['Mg']),
-            'Ca': float(data['profileSettings']['Ca']),
-            'S': float(data['profileSettings']['S']),
-            'N (NO3-)': float(data['profileSettings']['NO3']),
-            'N (NH4+)': float(data['profileSettings']['NH4'])
-        }
-        
-        # Конвертируем удобрения в нужный формат
-        calculator.fertilizers = {
+        print("Полученные данные:", data)  # Логирование входящих данных
+
+        # Проверка обязательных полей
+        if not data or 'fertilizerConstants' not in data or 'profileSettings' not in data:
+            return jsonify({"error": "Не хватает обязательных данных"}), 400
+
+        # Конвертация удобрений
+        fertilizers = {
             "Кальциевая селитра": {
-                "N (NO3-)": data['fertilizerConstants']['CaN2O6']['NO3'] / 100,
-                "Ca": data['fertilizerConstants']['CaN2O6']['Ca'] / 100
+                "N (NO3-)": float(data['fertilizerConstants']['Кальциевая селитра']["N (NO3-)"]),
+                "Ca": float(data['fertilizerConstants']['Кальциевая селитра']["Ca"])
             },
             "Калий азотнокислый": {
-                "N (NO3-)": data['fertilizerConstants']['KNO3']['NO3'] / 100,
-                "K": data['fertilizerConstants']['KNO3']['K'] / 100
+                "N (NO3-)": float(data['fertilizerConstants']['Калий азотнокислый']["N (NO3-)"]),
+                "K": float(data['fertilizerConstants']['Калий азотнокислый']["K"])
             },
             "Аммоний азотнокислый": {
-                "N (NO3-)": data['fertilizerConstants']['NH4NO3']['NO3'] / 100,
-                "N (NH4+)": data['fertilizerConstants']['NH4NO3']['NH4'] / 100
+                "N (NO3-)": float(data['fertilizerConstants']['Аммоний азотнокислый']["N (NO3-)"]),
+                "N (NH4+)": float(data['fertilizerConstants']['Аммоний азотнокислый']["N (NH4+)"])
             },
             "Сульфат магния": {
-                "Mg": data['fertilizerConstants']['MgSO4']['Mg'] / 100,
-                "S": data['fertilizerConstants']['MgSO4']['S'] / 100
+                "Mg": float(data['fertilizerConstants']['Сульфат магния']["Mg"]),
+                "S": float(data['fertilizerConstants']['Сульфат магния']["S"])
             },
             "Монофосфат калия": {
-                "P": data['fertilizerConstants']['KH2PO4']['P'] / 100,
-                "K": data['fertilizerConstants']['KH2PO4']['K'] / 100
+                "P": float(data['fertilizerConstants']['Монофосфат калия']["P"]),
+                "K": float(data['fertilizerConstants']['Монофосфат калия']["K"])
             },
             "Калий сернокислый": {
-                "K": data['fertilizerConstants']['K2SO4']['K'] / 100,
-                "S": data['fertilizerConstants']['K2SO4']['S'] / 100
+                "K": float(data['fertilizerConstants']['Калий сернокислый']["K"]),
+                "S": float(data['fertilizerConstants']['Калий сернокислый']["S"])
             }
         }
+
+        # Настройки профиля
+        profile = {
+            'P': float(data['profileSettings']['P']),
+            'K': float(data['profileSettings']['K']),
+            'Mg': float(data['profileSettings']['Mg']),
+            'Ca': float(data['profileSettings']['Ca']),
+            'S': float(data['profileSettings']['S']),
+            'N (NO3-)': float(data['profileSettings']['N (NO3-)']),
+            'N (NH4+)': float(data['profileSettings']['N (NH4+)']),
+        }
+        liters = int(data['profileSettings']['liters'])
+
+        # Создаем и настраиваем калькулятор
+        calculator = NutrientCalculator(volume_liters=liters)
+        calculator.fertilizers = fertilizers
+        calculator.target_profile.update(profile)
         
         # Выполняем расчет
         calculator.calculate()
         
-        # Возвращаем результаты
-        return jsonify({
-            "fertilizers": calculator._format_fertilizers(),
-            "profile": calculator._format_profile(),
+        # Формируем ответ
+        response = {
+            "fertilizers": [],
+            "profile": [],
             "ec": calculator.calculate_ec(),
             "deficits": calculator.calculate_deficits()
-        })
-        
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        }
 
+        # Заполняем удобрения
+        for fert_name, result in calculator.results.items():
+            response["fertilizers"].append({
+                "name": fert_name,
+                "grams": round(result['граммы'], 3),
+                "adds": {k.replace('внесет ', ''): v 
+                         for k, v in result.items() 
+                         if k.startswith('внесет')}
+            })
+
+        # Заполняем профиль
+        for element, value in calculator.actual_profile.items():
+            response["profile"].append({
+                "element": element.replace(' (NO3-)', '-NO3').replace(' (NH4+)', '-NH4'),
+                "ppm": round(value, 1)
+            })
+
+        print("Результат расчета:", response)  # Логирование результата
+        return jsonify(response)
+
+    except KeyError as e:
+        error_msg = f"Отсутствует обязательное поле: {str(e)}"
+        print(error_msg)
+        return jsonify({"error": error_msg}), 400
+    except Exception as e:
+        error_msg = f"Ошибка расчета: {str(e)}"
+        print(error_msg)
+        return jsonify({"error": error_msg}), 500