Update app.py

app.py

@@ -700,43 +700,58 @@ 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
+    },
+    "Сульфат калия": {
+        "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
     }
 }
 
 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.initial_n_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 calculate(self, base_profile, total_n, no3_ratio, nh4_ratio):
+    def calculate(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)
@@ -746,10 +761,10 @@ class NutrientCalculator:
             'N (NH4+)': round(nh4, 1)
         }
 
-        self.final_profile = base_profile.copy()
+        self.final_profile = BASE_PROFILE.copy()
         self.final_profile['N (NO3-)'] = no3
         self.final_profile['N (NH4+)'] = nh4
-        self.total_ppm = total_n + sum(base_profile.values())
+        self.total_ppm = total_n + sum(BASE_PROFILE.values())
 
         # Расчёт удобрений
         self._apply_magnesium_sulfate()
@@ -768,9 +783,9 @@ class NutrientCalculator:
 
     def _apply_magnesium_sulfate(self):
         mg_need = self.final_profile['Mg']
-        mg_content = FERTILIZERS_DB["Сульфат магния"]["Mg"]
+        mg_content = self.fertilizers_db["Сульфат магния"]["Mg"]
         grams = (mg_need * self.volume) / (mg_content * 1000)
-        added_s = grams * FERTILIZERS_DB["Сульфат магния"]["S"] * 1000 / self.volume
+        added_s = grams * self.fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
 
         self.final_profile['S'] -= added_s
         self._apply_fertilizer("Сульфат магния", grams, {'внесет S': round(added_s, 1)})
@@ -778,9 +793,9 @@ class NutrientCalculator:
 
     def _apply_calcium_nitrate(self):
         ca_need = self.final_profile['Ca']
-        ca_content = FERTILIZERS_DB["Кальциевая селитра"]["Ca"]
+        ca_content = self.fertilizers_db["Кальциевая селитра"]["Ca"]
         grams = (ca_need * self.volume) / (ca_content * 1000)
-        added_n = grams * FERTILIZERS_DB["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
+        added_n = grams * self.fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
 
         self.final_profile['N (NO3-)'] -= added_n
         self._apply_fertilizer("Кальциевая селитра", grams, {'внесет NO3': round(added_n, 1)})
@@ -788,9 +803,9 @@ class NutrientCalculator:
 
     def _apply_mkp(self):
         p_need = self.final_profile['P']
-        p_content = FERTILIZERS_DB["Монофосфат калия"]["P"]
+        p_content = self.fertilizers_db["Монофосфат калия"]["P"]
         grams = (p_need * self.volume) / (p_content * 1000)
-        added_k = grams * FERTILIZERS_DB["Монофосфат калия"]["K"] * 1000 / self.volume
+        added_k = grams * self.fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
 
         self.final_profile['K'] -= added_k
         self._apply_fertilizer("Монофосфат калия", grams, {'внесет K': round(added_k, 1)})
@@ -801,42 +816,53 @@ class NutrientCalculator:
         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, 1),
-                'внесет S': round(added_s, 1)
-            })
-            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
-
-            self._apply_fertilizer("Калий азотнокислый", kno3_grams, {'внесет NO3': round(added_n, 1)})
+        # Вариант 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)")
+
     def _apply_ammonium_nitrate(self):
         nh4_need = self.final_profile['N (NH4+)']
         if nh4_need <= 0:
             return
-        nh4_content = FERTILIZERS_DB["Аммоний азотнокислый"]["N (NH4+)"]
+        nh4_content = self.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 * self.fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
         self.final_profile['N (NO3-)'] -= added_n
         self._apply_fertilizer("Аммони�� азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
@@ -882,11 +908,9 @@ class NutrientCalculator:
         else:
             print("  Все элементы покрыты полностью")
 
-
-
-
 # Пример использования
 if __name__ == "__main__":
+
     
     # Вариант 2: С калием азотнокислым
     calc_with_kno3 = NutrientCalculator(volume_liters=10, use_k2so4=False, use_kno3=True)
@@ -898,6 +922,5 @@ if __name__ == "__main__":
 
 
 
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))