Update app.py

app.py

@@ -979,45 +979,140 @@ if __name__ == "__main__":
 
 
 
-@app.route('/calculation', methods=['POST'])
-def handle_calculation():
-    try:
-        data = request.json
+class WebNutrientCalculator:
+    def __init__(self, volume_liters=1.0):
+        self.volume = volume_liters
+        self.results = {}
+        self.target_profile = {}
+        self.actual_profile = {}
+        self.fertilizers = {}
+        self.total_ec = 0.0
+
+    def web_calculate(self, profile_settings, fertilizer_constants):
+        """Основная функция расчета для веб-интерфейса"""
+        try:
+            # 1. Устанавливаем целевой профиль из параметров
+            self.target_profile = {
+                'P': float(profile_settings['P']),
+                'K': float(profile_settings['K']),
+                'Mg': float(profile_settings['Mg']),
+                'Ca': float(profile_settings['Ca']),
+                'S': float(profile_settings['S']),
+                'N (NO3-)': float(profile_settings['N (NO3-)']),
+                'N (NH4+)': float(profile_settings['N (NH4+)'])
+            }
+            
+            # 2. Устанавливаем параметры удобрений
+            self.fertilizers = fertilizer_constants
+            
+            # 3. Выполняем расчет по новой логике
+            self._web_apply("Сульфат магния", "Mg", self.target_profile['Mg'])
+            self._web_apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+            self._web_apply("Монофосфат калия", "P", self.target_profile['P'])
+            self._web_apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+            
+            # 4. Коррекция NO3
+            current_no3 = self.actual_profile.get('N (NO3-)', 0)
+            no3_needed = self.target_profile['N (NO3-)'] - current_no3
+            if no3_needed > 0.1:
+                self._web_apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+            
+            # 5. Коррекция K и S
+            self._web_apply_k_sulfate()
+            
+            # 6. Финалный расчет EC
+            self.total_ec = self._web_calculate_ec()
+            
+            return self._web_get_results()
+            
+        except Exception as e:
+            print(f"WebCalculator error: {str(e)}")
+            raise
+
+    def _web_apply(self, fert_name, main_element, required_ppm):
+        """Аналог оригинального _apply для веб-версии"""
+        if required_ppm <= 0:
+            return
+            
+        content = self.fertilizers[fert_name][main_element]
+        grams = (required_ppm * self.volume) / (content * 1000)
         
-        # 1. СОЗДАЕМ КАЛЬКУЛЯТОР С НУЖНЫМ ОБЪЕМОМ
-        calculator = NutrientCalculator(volume_liters=data['profileSettings']['liters'])
+        if fert_name not in self.results:
+            self.results[fert_name] = {
+                'grams': 0.0,
+                'adds': {}
+            }
+            
+        self.results[fert_name]['grams'] += grams
         
-        # 2. ПЕРЕЗАПИСЫВАЕМ ВСЕ ЗНАЧЕНИЯ ИЗ ЗАПРОСА (БЕЗ СОХРАНЕНИЯ СТАРЫХ)
-        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']['N (NO3-)']), # Берем ТОЛЬКО из запроса
-            'N (NH4+)': float(data['profileSettings']['N (NH4+)'])  # Берем ТОЛЬКО из запроса
+        # Расчет вносимых элементов
+        for element, percent in self.fertilizers[fert_name].items():
+            added_ppm = (grams * percent * 1000) / self.volume
+            self.results[fert_name]['adds'][element] = added_ppm
+            self.actual_profile[element] = self.actual_profile.get(element, 0) + added_ppm
+
+    def _web_apply_k_sulfate(self):
+        """Аналог оригинального _apply_k_sulfate"""
+        fert = "Калий сернокислый"
+        k_def = self.target_profile['K'] - self.actual_profile.get('K', 0)
+        s_def = self.target_profile['S'] - self.actual_profile.get('S', 0)
+        
+        if s_def > 0.1:
+            self._web_apply(fert, "S", s_def)
+
+    def _web_calculate_ec(self):
+        """Упрощенный расчет 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
         }
         
-        # 3. УСТАНАВЛИВАЕМ ПАРАМЕТРЫ УДОБРЕНИЙ ИЗ ЗАПРОСА
-        calculator.fertilizers = {
-            fert: {el: float(p) for el, p in nutrients.items()}
-            for fert, nutrients in data['fertilizerConstants'].items()
+        total_ec = 0.0
+        for element, ppm in self.actual_profile.items():
+            total_ec += ppm * ec_coefficients.get(element, 0.0015)
+            
+        return round(total_ec, 2)
+
+    def _web_get_results(self):
+        """Форматирование результатов для веб"""
+        return {
+            'fertilizers': [
+                {
+                    'name': name,
+                    'grams': round(data['grams'], 3),
+                    'adds': {k: round(v, 2) for k, v in data['adds'].items()}
+                }
+                for name, data in self.results.items()
+            ],
+            'profile': [
+                {'element': k, 'ppm': round(v, 2)}
+                for k, v in self.actual_profile.items()
+            ],
+            'ec': self.total_ec,
+            'volume': self.volume
         }
+        }), 500
 
-        # 4. ВЫПОЛНЯЕМ РАСЧЕТ (ИСПОЛЬЗУЕТ ТОЛЬКО ПЕРЕДАННЫЕ ДАННЫЕ)
-        calculator.calculate()
+
+@app.route('/calculation', methods=['POST'])
+def handle_calculation():
+    try:
+        data = request.json
+        
+        # Создаем ВЕБ-версию калькулятора
+        web_calculator = WebNutrientCalculator(volume_liters=data['profileSettings']['liters'])
+        
+        # Выполняем расчет
+        result = web_calculator.web_calculate(
+            profile_settings=data['profileSettings'],
+            fertilizer_constants=data['fertilizerConstants']
+        )
         
-        # 5. ВОЗВРАЩАЕМ РЕЗУЛЬТАТ
-        return jsonify(calculator.get_web_results())
+        return jsonify(result)
         
     except Exception as e:
-        return jsonify({
-            "error": str(e),
-            "message": "Ошибка расчета"
-        }), 500
-
-
-
+        return jsonify({"error": str(e)}), 500