Update app.py

app.py

@@ -687,9 +687,9 @@ from tabulate import tabulate
 
 # Константы для расчёта
 TOTAL_NITROGEN = 125  # ppm
-NO3_RATIO = 8.00         # Доля нитратного азота (можно задавать 1.23, 5.67 и т.д.)
-NH4_RATIO = 1.00         # Доля аммонийного азота
-VOLUME_LITERS = 10       # Объём раствора
+NO3_RATIO = 8.00      # Доля нитратного азота 
+NH4_RATIO = 1.00      # Доля аммонийного азота
+VOLUME_LITERS = 10    # Объём раствора
 
 # Базовый профиль (без азота)
 BASE_PROFILE = {
@@ -700,150 +700,91 @@ BASE_PROFILE = {
     'S': 137.397
 }
 
-# База данных удобрений
-fertilizers_db = {
+# Полная база всех возможных удобрений
+ALL_FERTILIZERS = {
     "Кальциевая селитра": {
-        "N (NO3-)": 0.118,
-        "Ca": 0.169,
+        "N (NO3-)": 0.118, "Ca": 0.169
     },
     "Калий азотнокислый": {
-        "N (NO3-)": 0.138,
-        "K": 0.387,
+        "N (NO3-)": 0.138, "K": 0.387
     },
     "Калий сернокислый": {
-        "K": 0.448,
-        "S": 0.184,
+        "K": 0.448, "S": 0.184
     },
     "Аммоний азотнокислый": {
-        "N (NO3-)": 0.175,
-        "N (NH4+)": 0.175,
+        "N (NO3-)": 0.175, "N (NH4+)": 0.175
     },
     "Сульфат магния": {
-        "Mg": 0.098,
-        "S": 0.13,
+        "Mg": 0.098, "S": 0.13
     },
     "Монофосфат калия": {
-        "P": 0.227,
-        "K": 0.287,
+        "P": 0.227, "K": 0.287
+    },
+    "Сульфат калия": {
+        "K": 0.448, "S": 0.184  # Альтернатива калию сернокислому
     }
 }
 
 class NutrientCalculator:
-    def __init__(self, volume_liters=1.0):
+    def __init__(self, volume_liters=1.0, use_k2so4=True, use_kno3=True):
+        """
+        :param volume_liters: Объём раствора
+        :param use_k2so4: Использовать калий сернокислый (True/False)
+        :param use_kno3: Использовать калий азотнокислый (True/False)
+        """
         self.volume = volume_liters
         self.results = {}
         self.final_profile = {}
         self.total_ppm = 0
         self.calculated_profile = {}
-
-    def calculate_nitrogen(self, total_n, no3_ratio, nh4_ratio):
-        """Расчёт распределения азота по формам с точными соотношениями"""
-        total_parts = no3_ratio + nh4_ratio
-        no3 = total_n * (no3_ratio / total_parts)
-        nh4 = total_n * (nh4_ratio / total_parts)
         
-        self.calculated_profile = {
-            'N (NO3-)': round(no3, 2),
-            'N (NH4+)': round(nh4, 2),
-            **BASE_PROFILE
-        }
-        return self.calculated_profile
-
-    def calculate(self, total_n, no3_ratio, nh4_ratio):
-        """Основной расчёт питательного раствора"""
-        # Сначала рассчитываем азот
-        profile = self.calculate_nitrogen(total_n, no3_ratio, nh4_ratio)
-        self.final_profile = profile.copy()
-        self.total_ppm = total_n + sum(BASE_PROFILE.values())
-
-        # Расчёт удобрений
-        self._apply_magnesium_sulfate()
-        self._apply_calcium_nitrate()
-        self._apply_mkp()
-        self._apply_potassium_fertilizers()
-        self._apply_ammonium_nitrate()
-        return self.results
-
-    def _apply_fertilizer(self, fert_name, grams, additions):
-        """Добавление удобрения в результаты"""
-        self.results[fert_name] = {
-            'граммы': round(grams, 3),
-            'миллиграммы': int(grams * 1000),
-            **additions
+        # Формируем активные удобрения на основе выбора
+        self.fertilizers_db = {
+            "Кальциевая селитра": ALL_FERTILIZERS["Кальциевая селитра"],
+            "Аммоний азотнокислый": ALL_FERTILIZERS["Аммоний азотнокислый"],
+            "Сульфат магния": ALL_FERTILIZERS["Сульфат магния"],
+            "Монофосфат калия": ALL_FERTILIZERS["Монофосфат калия"]
         }
+        
+        if use_k2so4:
+            self.fertilizers_db["Калий сернокислый"] = ALL_FERTILIZERS["Калий сернокислый"]
+        if use_kno3:
+            self.fertilizers_db["Калий азотнокислый"] = ALL_FERTILIZERS["Калий азотнокислый"]
 
-    def _apply_magnesium_sulfate(self):
-        """Расчёт сульфата магния"""
-        mg_need = self.final_profile['Mg']
-        mg_content = fertilizers_db["Сульфат магния"]["Mg"]
-        grams = (mg_need * self.volume) / (mg_content * 1000)
-        added_s = grams * fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
-
-        self.final_profile['S'] -= added_s
-        self._apply_fertilizer(
-            "Сульфат магния", 
-            grams, 
-            {'внесет S': round(added_s, 2)}
-        )
-        self.final_profile['Mg'] = 0
-
-    def _apply_calcium_nitrate(self):
-        ca_need = self.final_profile['Ca']
-        ca_content = fertilizers_db["Кальциевая селитра"]["Ca"]
-        grams = (ca_need * self.volume) / (ca_content * 1000)
-        added_n = grams * fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
-
-        self.final_profile['N (NO3-)'] -= added_n
-        self._apply_fertilizer(
-            "Кальциевая селитра", 
-            grams, 
-            {'внесет NO3': round(added_n, 2)}
-        )
-        self.final_profile['Ca'] = 0
-
-    def _apply_mkp(self):
-        p_need = self.final_profile['P']
-        p_content = fertilizers_db["Монофосфат калия"]["P"]
-        grams = (p_need * self.volume) / (p_content * 1000)
-        added_k = grams * fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
-
-        self.final_profile['K'] -= added_k
-        self._apply_fertilizer(
-            "Монофосфат калия", 
-            grams, 
-            {'внесет K': round(added_k, 2)}
-        )
-        self.final_profile['P'] = 0
+    # ... (остальные методы класса остаются без изменений)
 
     def _apply_potassium_fertilizers(self):
         k_need = self.final_profile['K']
         if k_need <= 0:
             return
 
-        s_deficit = max(0, self.final_profile['S'])
-        if s_deficit > 0:
-            s_content = fertilizers_db["Калий сернокислый"]["S"]
-            k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
-            added_k = k2so4_grams * fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
-
-            if added_k > k_need:
-                k2so4_grams = (k_need * self.volume) / (fertilizers_db["Калий сернокислый"]["K"] * 1000)
-                added_k = k_need
-                added_s = k2so4_grams * fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
-            else:
-                added_s = s_deficit
-
-            self._apply_fertilizer("Калий сернокислый", k2so4_grams, {
-                'внесет K': round(added_k, 2),
-                'внесет S': round(added_s, 2)
-            })
-            self.final_profile['K'] -= added_k
-            self.final_profile['S'] -= added_s
-            k_need = self.final_profile['K']
-
-        if k_need > 0:
-            kno3_grams = (k_need * self.volume) / (fertilizers_db["Калий азотнокислый"]["K"] * 1000)
-            added_n = kno3_grams * fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+        # Вариант 1: Если есть калий сернокислый
+        if "Калий сернокислый" in self.fertilizers_db:
+            s_deficit = max(0, self.final_profile['S'])
+            if s_deficit > 0:
+                s_content = self.fertilizers_db["Калий сернокислый"]["S"]
+                k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
+                added_k = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
+
+                if added_k > k_need:
+                    k2so4_grams = (k_need * self.volume) / (self.fertilizers_db["Калий сернокислый"]["K"] * 1000)
+                    added_k = k_need
+                    added_s = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
+                else:
+                    added_s = s_deficit
+
+                self._apply_fertilizer("Калий сернокислый", k2so4_grams, {
+                    'внесет K': round(added_k, 2),
+                    'внесет S': round(added_s, 2)
+                })
+                self.final_profile['K'] -= added_k
+                self.final_profile['S'] -= added_s
+                k_need = self.final_profile['K']
+
+        # Вариант 2: Если есть только калий азотнокислый
+        if k_need > 0 and "Калий азотнокислый" in self.fertilizers_db:
+            kno3_grams = (k_need * self.volume) / (self.fertilizers_db["Калий азотнокислый"]["K"] * 1000)
+            added_n = kno3_grams * self.fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
             self._apply_fertilizer(
                 "Калий азотнокислый", 
@@ -853,81 +794,21 @@ class NutrientCalculator:
             self.final_profile['K'] = 0
             self.final_profile['N (NO3-)'] -= added_n
 
-    def _apply_ammonium_nitrate(self):
-        nh4_need = self.final_profile['N (NH4+)']
-        if nh4_need <= 0:
-            return
-        nh4_content = fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"]
-        grams = (nh4_need * self.volume) / (nh4_content * 1000)
-        added_n = grams * fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
-
-        self.final_profile['N (NO3-)'] -= added_n
-        self._apply_fertilizer(
-            "Аммоний азотнокислый", 
-            grams, 
-            {'внесет NO3': round(added_n, 2)}
-        )
-        self.final_profile['N (NH4+)'] = 0
-
-    def calculate_ec(self):
-        """Расчёт электропроводимости"""
-        return round(self.total_ppm / 700, 3)
-
-    def print_report(self):
-        """Вывод результатов расчёта"""
-        # Выводим расчётный профиль после вычисления азота
-        print("\n" + "="*60)
-        print("РАСЧЁТНЫЙ ПРОФИЛЬ (после расчёта азота):")
-        print("="*60)
-        print(f"Соотношение NO3:NH4 = {NO3_RATIO:.2f}:{NH4_RATIO:.2f}")
-        
-        profile_table = [
-            [element, f"{value:.2f}"] 
-            for element, value in self.calculated_profile.items()
-        ]
-        print(tabulate(profile_table, headers=["Элемент", "Концентрация ppm"]))
-        
-        # Выводим результаты расчёта удобрений
-        print("\n" + "="*60)
-        print(f"РЕЗУЛЬТАТЫ РАСЧЁТА ДЛЯ {self.volume} ЛИТРОВ")
-        print("="*60)
-        print(f"Общая концентрация: {self.total_ppm:.2f} ppm")
-        print(f"EC: {self.calculate_ec()} mS/cm")
-
-        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
-        fert_table = []
-        for fert, data in self.results.items():
-            details = [
-                f"+{k}: {v:.2f} ppm" 
-                for k, v in data.items() 
-                if k.startswith('внесет')
-            ]
-            fert_table.append([
-                fert,
-                f"{data['граммы']:.3f} г",
-                f"{data['миллиграммы']} мг",
-                "\n".join(details)
-            ])
-        print(tabulate(fert_table, 
-                      headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
-        
-        print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
-        deficit = {
-            k: round(v, 2) 
-            for k, v in self.final_profile.items() 
-            if v > 0.01  # Более точный порог
-        }
-        if deficit:
-            for el, val in deficit.items():
-                print(f"  {el}: {val:.2f} ppm")
-        else:
-            print("  Все элементы покрыты полностью")
-
-
+        # Если остался дефицит и нет подходящих удобрений
+        if k_need > 0:
+            print(f"  Внимание: Не удалось полностью покрыть дефицит K ({k_need:.2f} ppm)")
+
+# Пример использования
+if __name__ == "__main__":
+    # Вариант 1: С калием сернокислым
+    calc_with_k2so4 = NutrientCalculator(volume_liters=10, use_k2so4=True, use_kno3=False)
+    calc_with_k2so4.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
+    calc_with_k2so4.print_report()
     
-calc = NutrientCalculator(volume_liters=VOLUME_LITERS)
-calc.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
-calc.print_report()
+    # Вариант 2: С калием азотнокислым
+    calc_with_kno3 = NutrientCalculator(volume_liters=10, use_k2so4=False, use_kno3=True)
+    calc_with_kno3.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
+    calc_with_kno3.print_report()