Update app.py

app.py

@@ -700,58 +700,35 @@ BASE_PROFILE = {
     'S': 137.397
 }
 
-# Полная база всех возможных удобрений
-ALL_FERTILIZERS = {
+# База данных удобрений
+FERTILIZERS_DB = {
     "Кальциевая селитра": {
-        "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
-    },
-    "Сульфат калия": {
-        "K": 0.448, "S": 0.184  # Альтернатива калию сернокислому
+        "P": 0.227, 
+        "K": 0.287
     }
 }
 
 class NutrientCalculator:
-    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 = {}
-        
-        # Формируем активные удобрения на основе выбора
-        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 __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
@@ -759,10 +736,10 @@ class NutrientCalculator:
         self.total_ppm = 0
         self.initial_n_profile = {}
 
-    def calculate(self, base_profile, total_n, n_ratio):
-        total_parts = n_ratio[0] + n_ratio[1]
-        no3 = total_n * (n_ratio[0] / total_parts)
-        nh4 = total_n * (n_ratio[1] / total_parts)
+    def calculate(self, base_profile, 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.initial_n_profile = {
             'N (NO3-)': round(no3, 1),
@@ -791,9 +768,9 @@ class NutrientCalculator:
 
     def _apply_magnesium_sulfate(self):
         mg_need = self.final_profile['Mg']
-        mg_content = fertilizers_db["Сульфат магния"]["Mg"]
+        mg_content = FERTILIZERS_DB["Сульфат магния"]["Mg"]
         grams = (mg_need * self.volume) / (mg_content * 1000)
-        added_s = grams * fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
+        added_s = grams * FERTILIZERS_DB["Сульфат магния"]["S"] * 1000 / self.volume
 
         self.final_profile['S'] -= added_s
         self._apply_fertilizer("Сульфат магния", grams, {'внесет S': round(added_s, 1)})
@@ -801,9 +778,9 @@ class NutrientCalculator:
 
     def _apply_calcium_nitrate(self):
         ca_need = self.final_profile['Ca']
-        ca_content = fertilizers_db["Кальциевая селитра"]["Ca"]
+        ca_content = FERTILIZERS_DB["Кальциевая селитра"]["Ca"]
         grams = (ca_need * self.volume) / (ca_content * 1000)
-        added_n = grams * fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
+        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, 1)})
@@ -811,9 +788,9 @@ class NutrientCalculator:
 
     def _apply_mkp(self):
         p_need = self.final_profile['P']
-        p_content = fertilizers_db["Монофосфат калия"]["P"]
+        p_content = FERTILIZERS_DB["��онофосфат калия"]["P"]
         grams = (p_need * self.volume) / (p_content * 1000)
-        added_k = grams * fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
+        added_k = grams * FERTILIZERS_DB["Монофосфат калия"]["K"] * 1000 / self.volume
 
         self.final_profile['K'] -= added_k
         self._apply_fertilizer("Монофосфат калия", grams, {'внесет K': round(added_k, 1)})
@@ -826,14 +803,14 @@ class NutrientCalculator:
 
         s_deficit = max(0, self.final_profile['S'])
         if s_deficit > 0:
-            s_content = fertilizers_db["Калий сернокислый"]["S"]
+            s_content = FERTILIZERS_DB["Калий сернокислый"]["S"]
             k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
-            added_k = k2so4_grams * fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
+            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)
+                k2so4_grams = (k_need * self.volume) / (FERTILIZERS_DB["Калий сернокислый"]["K"] * 1000)
                 added_k = k_need
-                added_s = k2so4_grams * fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
+                added_s = k2so4_grams * FERTILIZERS_DB["Калий сернокислый"]["S"] * 1000 / self.volume
             else:
                 added_s = s_deficit
 
@@ -846,8 +823,8 @@ class NutrientCalculator:
             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
+            kno3_grams = (k_need * self.volume) / (FERTILIZERS_DB["Калий азотнокислый"]["K"] * 1000)
+            added_n = kno3_grams * FERTILIZERS_DB["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
             self._apply_fertilizer("Калий азотнокислый", kno3_grams, {'внесет NO3': round(added_n, 1)})
             self.final_profile['K'] = 0
@@ -857,9 +834,9 @@ class NutrientCalculator:
         nh4_need = self.final_profile['N (NH4+)']
         if nh4_need <= 0:
             return
-        nh4_content = fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"]
+        nh4_content = FERTILIZERS_DB["Аммоний азотнокислый"]["N (NH4+)"]
         grams = (nh4_need * self.volume) / (nh4_content * 1000)
-        added_n = grams * fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+        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, 1)})
@@ -905,58 +882,11 @@ class NutrientCalculator:
         else:
             print("  Все элементы покрыты полностью")
 
-    def _apply_potassium_fertilizers(self):
-        k_need = self.final_profile['K']
-        if k_need <= 0:
-            return
-
-        # Вариант 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(
-                "Калий азотнокислый", 
-                kno3_grams, 
-                {'внесет NO3': round(added_n, 2)}
-            )
-            self.final_profile['K'] = 0
-            self.final_profile['N (NO3-)'] -= added_n
-
-        # Если остался дефицит и нет подходящих удобрений
-        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()
-    
+    calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
+    calculator.calculate(BASE_PROFILE, TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
+    calculator.print_report()