Update app.py

app.py

@@ -684,26 +684,18 @@ NO3_RATIO = 9.000
 NH4_RATIO = 1.00
 VOLUME_LITERS = 100
 
-# Коэффициенты электропроводности для каждого элемента (мСм/см на ppm)
+# Коэффициенты электропроводности
 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
+    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015, 
+    'Ca': 0.0016, 'S': 0.0014, 
+    'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
 }
 
 # Целевые значения
 BASE_PROFILE = {
-    'P': 31.000,
-    'K': 210.000,
-    'Mg': 24.000,
-    'Ca': 84.000,
-    'S': 56.439,
-    'N (NO3-)': 0,
-    'N (NH4+)': 0
+    'P': 31.000, 'K': 210.000, 'Mg': 24.000,
+    'Ca': 84.000, 'S': 56.439,
+    'N (NO3-)': 0, 'N (NH4+)': 0
 }
 
 FERTILIZERS = {
@@ -722,7 +714,7 @@ class NutrientCalculator:
         self.target_profile = BASE_PROFILE.copy()
         self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
         self.fertilizers = FERTILIZERS
-        self.total_ec = 0.0  # Накопленное значение EC
+        self.total_ec = 0.0
 
         # Расчёт азота
         total_parts = NO3_RATIO + NH4_RATIO
@@ -733,108 +725,155 @@ class NutrientCalculator:
             "NH4+": self.target_profile['N (NH4+)']
         }
 
+    def _label(self, element):
+        """Форматирование названий элементов для вывода"""
+        labels = {
+            'N (NO3-)': 'NO3',
+            'N (NH4+)': 'NH4'
+        }
+        return labels.get(element, element)
+
     def calculate(self):
-        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+)'])
-        
-        current_no3 = self.actual_profile['N (NO3-)']
-        no3_needed = self.target_profile['N (NO3-)'] - current_no3
-        
-        if no3_needed > 0.1:
-            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
-        
-        self._apply_k_sulfate()
-        
-        k_deficit = self.target_profile['K'] - self.actual_profile['K']
-        if k_deficit > 0.1:
-            self._apply("Калий азотнокислый", "K", k_deficit)
-        
-        return self.results
+        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+)'])
+            
+            current_no3 = self.actual_profile['N (NO3-)']
+            no3_needed = self.target_profile['N (NO3-)'] - current_no3
+            
+            if no3_needed > 0.1:
+                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+            
+            self._apply_k_sulfate()
+            
+            k_deficit = self.target_profile['K'] - self.actual_profile['K']
+            if k_deficit > 0.1:
+                self._apply("Калий азотнокислый", "K", k_deficit)
+            
+            return self.results
+        except Exception as e:
+            print(f"Ошибка при расчёте: {str(e)}")
+            raise
 
     def _apply(self, fert_name, main_element, required_ppm):
         if required_ppm <= 0:
             return
             
-        content = self.fertilizers[fert_name][main_element]
-        grams = required_ppm * self.volume / (content * 1000)
-        
-        if fert_name not in self.results:
-            result = {
-                'граммы': 0.0,
-                'миллиграммы': 0,
-                'вклад в EC': 0.0
-            }
-            for element in self.fertilizers[fert_name]:
-                result[f'внесет {self._label(element)}'] = 0.0
-            self.results[fert_name] = result
+        try:
+            content = self.fertilizers[fert_name][main_element]
+            grams = required_ppm * self.volume / (content * 1000)
             
-        self.results[fert_name]['граммы'] += grams
-        self.results[fert_name]['миллиграммы'] += int(grams * 1000)
-        
-        fert_ec = 0.0
-        for element, percent in self.fertilizers[fert_name].items():
-            added_ppm = grams * percent * 1000 / self.volume
-            self.results[fert_name][f'внесет {self._label(element)}'] += added_ppm
-            self.actual_profile[element] += added_ppm
-            fert_ec += added_ppm * EC_COEFFICIENTS[element]
+            if fert_name not in self.results:
+                result = {
+                    'граммы': 0.0,
+                    'миллиграммы': 0,
+                    'вклад в EC': 0.0
+                }
+                for element in self.fertilizers[fert_name]:
+                    result[f'внесет {self._label(element)}'] = 0.0
+                self.results[fert_name] = result
+                
+            self.results[fert_name]['граммы'] += grams
+            self.results[fert_name]['миллиграммы'] += int(grams * 1000)
+            
+            fert_ec = 0.0
+            for element, percent in self.fertilizers[fert_name].items():
+                added_ppm = grams * percent * 1000 / self.volume
+                self.results[fert_name][f'внесет {self._label(element)}'] += added_ppm
+                self.actual_profile[element] += added_ppm
+                fert_ec += added_ppm * EC_COEFFICIENTS.get(element, 0.0015)  # Значение по умолчанию
+                
+            self.results[fert_name]['вклад в EC'] += fert_ec
+            self.total_ec += fert_ec
+        except KeyError as e:
+            print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
+            raise
+
+    def _apply_k_sulfate(self):
+        fert = "Калий сернокислый"
+        k_def = self.target_profile['K'] - self.actual_profile['K']
+        s_def = self.target_profile['S'] - self.actual_profile['S']
         
-        self.results[fert_name]['вклад в EC'] += fert_ec
-        self.total_ec += fert_ec
-
-    # ... остальные методы остаются без изменений ...
+        if k_def <= 0 and s_def <= 0:
+            return
+            
+        try:
+            if s_def > 0.1:
+                s_content = self.fertilizers[fert]["S"]
+                grams_s = s_def * self.volume / (s_content * 1000)
+                
+                k_content = self.fertilizers[fert]["K"]
+                k_from_s = grams_s * k_content * 1000 / self.volume
+                
+                if k_from_s > k_def and k_def > 0.1:
+                    grams = k_def * self.volume / (k_content * 1000)
+                else:
+                    grams = grams_s
+                    
+                self._apply(fert, "S", s_def)
+        except Exception as e:
+            print(f"Ошибка при расчёте сульфата калия: {str(e)}")
+            raise
 
     def calculate_ec(self):
         return round(self.total_ec, 2)
 
     def print_report(self):
-        print("\n" + "="*60)
-        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
-        print("="*60)
-        table = [[el, round(self.actual_profile[el], 1)] for el in self.actual_profile]
-        print(tabulate(table, headers=["Элемент", "ppm"]))
-
-        print("\nИсходный расчёт азота:")
-        for form, val in self.initial_n_profile.items():
-            print(f"  {form}: {round(val, 1)} ppm")
-
-        print("\n" + "="*60)
-        print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
-        print("="*60)
-        print(f"Общая концентрация: {round(sum(self.actual_profile.values()), 1)} ppm")
-        print(f"EC: {self.calculate_ec()} mS/cm")
-
-        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
-        fert_table = []
-        for fert, data in self.results.items():
-            adds = [f"+{k}: {v:.1f} ppm" for k, v in data.items() if k.startswith('внесет')]
-            fert_table.append([
-                fert,
-                round(data['граммы'], 3),
-                data['миллиграммы'],
-                round(data['вклад в EC'], 3),
-                "\n".join(adds)
-            ])
-        print(tabulate(fert_table, 
-                     headers=["Удобрение", "Граммы", "Миллиграммы", "EC (мСм/см)", "Добавит"]))
-
-        print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
-        deficit = {
-            k: round(self.target_profile[k] - self.actual_profile[k], 1)
-            for k in self.target_profile
-            if abs(self.target_profile[k] - self.actual_profile[k]) > 0.1
-        }
-        if deficit:
-            for el, val in deficit.items():
-                print(f"  {el}: {val} ppm")
-        else:
-            print("  Все элементы покрыты полностью")
+        try:
+            print("\n" + "="*60)
+            print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
+            print("="*60)
+            table = [[el, round(self.actual_profile[el], 1)] for el in self.actual_profile]
+            print(tabulate(table, headers=["Элемент", "ppm"]))
+
+            print("\nИсходный расчёт азота:")
+            for form, val in self.initial_n_profile.items():
+                print(f"  {form}: {round(val, 1)} ppm")
+
+            print("\n" + "="*60)
+            print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
+            print("="*60)
+            print(f"Общая концентрация: {round(sum(self.actual_profile.values()), 1)} ppm")
+            print(f"EC: {self.calculate_ec()} mS/cm")
+
+            print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
+            fert_table = []
+            for fert, data in self.results.items():
+                adds = [f"+{k}: {v:.1f} ppm" for k, v in data.items() if k.startswith('внесет')]
+                fert_table.append([
+                    fert,
+                    round(data['граммы'], 3),
+                    data['миллиграммы'],
+                    round(data['вклад в EC'], 3),
+                    "\n".join(adds)
+                ])
+            print(tabulate(fert_table, 
+                         headers=["Удобрение", "Граммы", "Миллиграммы", "EC (мСм/см)", "Добавит"]))
+
+            print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
+            deficit = {
+                k: round(self.target_profile[k] - self.actual_profile[k], 1)
+                for k in self.target_profile
+                if abs(self.target_profile[k] - self.actual_profile[k]) > 0.1
+            }
+            if deficit:
+                for el, val in deficit.items():
+                    print(f"  {el}: {val} ppm")
+            else:
+                print("  Все элементы покрыты полностью")
+        except Exception as e:
+            print(f"Ошибка при выводе отчёта: {str(e)}")
+            raise
 
 if __name__ == "__main__":
-    calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
-    calculator.calculate()
-    calculator.print_report()
+    try:
+        calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
+        calculator.calculate()
+        calculator.print_report()
+    except Exception as e:
+        print(f"Критическая ошибка: {str(e)}")