Update app.py

app.py

@@ -702,7 +702,7 @@ BASE_PROFILE = {
     'N (NH4+)': 0
 }
 
-# Полная база всех возможных удобрений (7 удобрений)
+# Полная база всех удобрений (7 удобрений)
 ALL_FERTILIZERS = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118, "Ca": 0.169
@@ -728,28 +728,16 @@ ALL_FERTILIZERS = {
 }
 
 class NutrientCalculator:
-    def __init__(self, volume_liters=1.0, use_k2so4=True, use_kno3=True):
+    def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
         self.final_profile = BASE_PROFILE.copy()
         self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
         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["Калий азотнокислый"]
+        self.fertilizers_db = ALL_FERTILIZERS
 
     def calculate(self, total_n, no3_ratio, nh4_ratio):
-        # Рассчитываем баланс азота
+        # 1. Сначала рассчитываем баланс азота
         total_parts = no3_ratio + nh4_ratio
         no3 = total_n * (no3_ratio / total_parts)
         nh4 = total_n * (nh4_ratio / total_parts)
@@ -759,16 +747,17 @@ class NutrientCalculator:
             'N (NH4+)': round(nh4, 1)
         }
         
-        # Добавляем азот в профиль
+        # Записываем азот в профиль
         self.final_profile['N (NO3-)'] = no3
         self.final_profile['N (NH4+)'] = nh4
 
-        # Последовательность внесения удобрений
+        # 2. Вносим удобрения в правильном порядке
         self._apply_magnesium_sulfate()
         self._apply_calcium_nitrate()
         self._apply_potassium_phosphate()
         self._apply_ammonium_nitrate()
-        self._apply_potassium_fertilizers()
+        self._apply_potassium_sulfate()
+        self._apply_potassium_nitrate()
         
         # Корректируем отрицательные значения
         self._adjust_negative_values()
@@ -783,7 +772,6 @@ class NutrientCalculator:
         self.results[fert_name].update(additions)
 
     def _adjust_negative_values(self):
-        # Корректируем отрицательные значения до 0
         for element in self.final_profile:
             if self.final_profile[element] < 0:
                 self.final_profile[element] = 0
@@ -827,39 +815,6 @@ class NutrientCalculator:
         self._apply_fertilizer("Калий фосфорнокислый", grams, {'внесет K': round(added_k, 1)})
         self.final_profile['P'] = 0
 
-    def _apply_potassium_fertilizers(self):
-        k_need = self.final_profile['K']
-        if k_need <= 0:
-            return
-
-        # Сначала используем калий сернокислый (если доступен)
-        if "Калий сернокислый" in self.fertilizers_db:
-            s_deficit = self.final_profile['S']
-            k2so4_grams = (k_need * self.volume) / (self.fertilizers_db["Калий сернокислый"]["K"] * 1000)
-            added_s = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
-
-            if s_deficit > 0 and added_s <= s_deficit:
-                self._apply_fertilizer("Калий сернокислый", k2so4_grams, {
-                    'внесет K': round(k_need, 2),
-                    'внесет S': round(added_s, 2)
-                })
-                self.final_profile['K'] = 0
-                self.final_profile['S'] -= added_s
-                return
-
-        # Если остался калий, используем калий азотнокислый
-        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
-
     def _apply_ammonium_nitrate(self):
         nh4_need = self.final_profile['N (NH4+)']
         if nh4_need <= 0:
@@ -873,6 +828,44 @@ class NutrientCalculator:
         self._apply_fertilizer("Аммоний азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
         self.final_profile['N (NH4+)'] = 0
 
+    def _apply_potassium_sulfate(self):
+        k_need = self.final_profile['K']
+        s_need = self.final_profile['S']
+        
+        if k_need <= 0 or s_need <= 0:
+            return
+            
+        k_content = self.fertilizers_db["Калий сернокислый"]["K"]
+        s_content = self.fertilizers_db["Калий сернокислый"]["S"]
+        
+        # Рассчитываем сколько можем внести по калию и сере
+        grams_by_k = (k_need * self.volume) / (k_content * 1000)
+        grams_by_s = (s_need * self.volume) / (s_content * 1000)
+        grams = min(grams_by_k, grams_by_s)
+        
+        added_k = grams * k_content * 1000 / self.volume
+        added_s = grams * s_content * 1000 / self.volume
+
+        self.final_profile['K'] -= added_k
+        self.final_profile['S'] -= added_s
+        self._apply_fertilizer("Калий сернокислый", grams, {
+            'внесет K': round(added_k, 2),
+            'внесет S': round(added_s, 2)
+        })
+
+    def _apply_potassium_nitrate(self):
+        k_need = self.final_profile['K']
+        if k_need <= 0:
+            return
+            
+        k_content = self.fertilizers_db["Калий азотнокислый"]["K"]
+        grams = (k_need * self.volume) / (k_content * 1000)
+        added_n = grams * self.fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+
+        self.final_profile['K'] = 0
+        self.final_profile['N (NO3-)'] -= added_n
+        self._apply_fertilizer("Калий азотнокислый", grams, {'внесет NO3': round(added_n, 2)})
+
     def calculate_ec(self):
         return round(self.total_ppm / 700, 2)
 
@@ -915,11 +908,9 @@ class NutrientCalculator:
 
 # Пример использования
 if __name__ == "__main__":
-    # Вариант с калием азотнокислым
-    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()
-
+    calculator = NutrientCalculator(volume_liters=10)
+    calculator.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
+    calculator.print_report()