Update app.py

app.py

@@ -689,17 +689,21 @@ def nutri_call():
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
-        self.results = defaultdict(lambda: {'grams': 0.0, 'adds': defaultdict(float)})
-        self.target = {
+        self.results = {}
+        self.target_profile = {
             'P': 0, 'K': 0, 'Mg': 0, 'Ca': 0, 'S': 0,
-            'NO3': 0, 'NH4': 0
+            'NO3': 0, 'NH4': 0  # Упрощенные ключи
+        }
+        self.actual_profile = {
+            'P': 0.0, 'K': 0.0, 'Mg': 0.0, 'Ca': 0.0, 'S': 0.0,
+            'NO3': 0.0, 'NH4': 0.0
         }
-        self.actual = defaultdict(float)
         self.fertilizers = {}
         self.total_ec = 0.0
 
     def set_target_profile(self, profile_data):
-        self.target.update({
+        """Устанавливаем целевые значения с преобразованием ключей"""
+        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)),
@@ -710,6 +714,7 @@ class NutrientCalculator:
         })
 
     def set_fertilizers(self, fertilizers_data):
+        """Преобразуем входные данные в единый формат"""
         self.fertilizers = {
             "Сульфат магния": {
                 "Mg": float(fertilizers_data["Сульфат магния"]["Mg"]),
@@ -738,29 +743,30 @@ class NutrientCalculator:
         }
 
     def calculate(self):
+        """Новая логика расчёта с приоритетом по азоту"""
         try:
-            # 1. Обязательные компоненты
+            # 1. Вносим обязательные компоненты
             self._apply_fertilizer("Сульфат магния", "Mg")
             self._apply_fertilizer("Монофосфат калия", "P")
             
-            # 2. Азотный баланс
+            # 2. Контроль азота
             self._apply_fertilizer("Аммоний азотнокислый", "NH4")
             
             # 3. Кальций с контролем NO3
-            remaining_no3 = max(self.target['NO3'] - self.actual['NO3'], 0)
+            remaining_no3 = max(self.target_profile['NO3'] - self.actual_profile['NO3'], 0)
             if remaining_no3 > 0.1:
                 self._apply_fertilizer("Кальциевая селитра", "Ca")
             
-            # 4. Калий
-            k_deficit = self.target['K'] - self.actual['K']
+            # 4. Калий с минимальным NO3
+            k_deficit = self.target_profile['K'] - self.actual_profile['K']
             if k_deficit > 0.1:
-                if self.actual['NO3'] < self.target['NO3']:
+                if self.actual_profile['NO3'] < self.target_profile['NO3']:
                     self._apply_fertilizer("Калий азотнокислый", "K")
                 else:
                     self._apply_fertilizer("Калий сернокислый", "K")
             
             # 5. Сера
-            s_deficit = self.target['S'] - self.actual['S']
+            s_deficit = self.target_profile['S'] - self.actual_profile['S']
             if s_deficit > 0.1:
                 self._apply_fertilizer("Калий сернокислый", "S")
             
@@ -770,11 +776,12 @@ class NutrientCalculator:
             return False
 
     def _apply_fertilizer(self, name, main_element, required_ppm=None):
+        """Умное внесение удобрений с контролем NO3"""
         if name not in self.fertilizers:
             return
 
         if required_ppm is None:
-            required_ppm = self.target[main_element] - self.actual[main_element]
+            required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
         
         if required_ppm <= 0:
             return
@@ -783,22 +790,28 @@ class NutrientCalculator:
         if content <= 0:
             return
 
-        # Контроль NO3
+        # Корректировка для NO3
         if 'NO3' in self.fertilizers[name]:
-            no3_limit = max(self.target['NO3'] - self.actual['NO3'], 0)
+            no3_limit = max(self.target_profile['NO3'] - self.actual_profile['NO3'], 0)
             if no3_limit <= 0:
-                return
+                return  # Превышение NO3
             max_grams = (no3_limit * self.volume) / (self.fertilizers[name]['NO3'] * 1000)
             required_grams = (required_ppm * self.volume) / (content * 1000)
             grams = min(required_grams, max_grams)
         else:
             grams = (required_ppm * self.volume) / (content * 1000)
 
+        # Инициализация записи
+        if name not in self.results:
+            self.results[name] = {'grams': 0.0, 'adds': {}}
+        
         # Обновление данных
         self.results[name]['grams'] += grams
         for element, percent in self.fertilizers[name].items():
             added_ppm = (grams * percent * 1000) / self.volume
-            self.actual[element] += added_ppm
+            self.actual_profile[element] += added_ppm
+            if element not in self.results[name]['adds']:
+                self.results[name]['adds'][element] = 0.0
             self.results[name]['adds'][element] += added_ppm
 
         # Расчёт EC
@@ -813,6 +826,7 @@ class NutrientCalculator:
         )
 
     def get_results(self):
+        """Форматируем выходные данные"""
         return {
             'fertilizers': [
                 {
@@ -822,40 +836,49 @@ class NutrientCalculator:
                 } for name, data in self.results.items()
             ],
             'profile': [
-                {'element': 'P', 'ppm': round(self.actual['P'], 3)},
-                {'element': 'K', 'ppm': round(self.actual['K'], 3)},
-                {'element': 'Mg', 'ppm': round(self.actual['Mg'], 3)},
-                {'element': 'Ca', 'ppm': round(self.actual['Ca'], 3)},
-                {'element': 'S', 'ppm': round(self.actual['S'], 3)},
-                {'element': 'N-NO3', 'ppm': round(self.actual['NO3'], 3)},
-                {'element': 'N-NH4', 'ppm': round(self.actual['NH4'], 3)}
+                {'element': 'P', 'ppm': round(self.actual_profile['P'], 3)},
+                {'element': 'K', 'ppm': round(self.actual_profile['K'], 3)},
+                {'element': 'Mg', 'ppm': round(self.actual_profile['Mg'], 3)},
+                {'element': 'Ca', 'ppm': round(self.actual_profile['Ca'], 3)},
+                {'element': 'S', 'ppm': round(self.actual_profile['S'], 3)},
+                {'element': 'N-NO3', 'ppm': round(self.actual_profile['NO3'], 3)},
+                {'element': 'N-NH4', 'ppm': round(self.actual_profile['NH4'], 3)}
             ],
             'ec': round(self.total_ec, 2),
             'deficits': {
-                'Ca': max(self.target['Ca'] - self.actual['Ca'], 0),
-                'K': max(self.target['K'] - self.actual['K'], 0),
-                'Mg': max(self.target['Mg'] - self.actual['Mg'], 0),
-                'N (NH4+)': max(self.target['NH4'] - self.actual['NH4'], 0),
-                'N (NO3-)': max(self.target['NO3'] - self.actual['NO3'], 0),
-                'P': max(self.target['P'] - self.actual['P'], 0),
-                'S': max(self.target['S'] - self.actual['S'], 0)
+                'Ca': max(self.target_profile['Ca'] - self.actual_profile['Ca'], 0),
+                'K': max(self.target_profile['K'] - self.actual_profile['K'], 0),
+                'Mg': max(self.target_profile['Mg'] - self.actual_profile['Mg'], 0),
+                'N (NH4+)': max(self.target_profile['NH4'] - self.actual_profile['NH4'], 0),
+                'N (NO3-)': max(self.target_profile['NO3'] - self.actual_profile['NO3'], 0),
+                'P': max(self.target_profile['P'] - self.actual_profile['P'], 0),
+                'S': max(self.target_profile['S'] - self.actual_profile['S'], 0)
             }
         }
 
-@app.route('/calculation', methods=['POST'])  # Точный эндпоинт из ошибки
+@app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try:
         data = request.json
         
         # Валидация
-        required = ['P', 'K', 'Mg', 'Ca', 'S', 'N (NO3-)', 'N (NH4+)', 'liters']
-        if not all(k in data['profileSettings'] for k in required):
-            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
+        
+        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'])
         
+        # Расчёт
         if calculator.calculate():
             return jsonify({
                 'status': 'success',