Update app.py

app.py

@@ -702,149 +702,105 @@ BASE_PROFILE = {
     'N (NH4+)': 0
 }
 
-# Полная база всех удобрений (7 удобрений)
-ALL_FERTILIZERS = {
-    "Кальциевая селитра": {
-        "N (NO3-)": 0.118, "Ca": 0.169
-    },
-    "Калий азотнокислый": {
-        "N (NO3-)": 0.138, "K": 0.387
-    },
-    "Калий сернокислый": {
-        "K": 0.448, "S": 0.184
-    },
-    "Аммоний азотнокислый": {
-        "N (NO3-)": 0.175, "N (NH4+)": 0.175
-    },
-    "Сульфат магния": {
-        "Mg": 0.098, "S": 0.13
-    },
-    "Калий фосфорнокислый": {
-        "P": 0.228, "K": 0.287
-    },
-    "Монофосфат калия": {
-        "P": 0.228, "K": 0.287
-    }
+# База удобрений (6 удобрений)
+FERTILIZERS = {
+    "Кальциевая селитра": {"N (NO3-)": 0.118, "Ca": 0.169},
+    "Калий азотнокислый": {"N (NO3-)": 0.138, "K": 0.387},
+    "Калий сернокислый": {"K": 0.448, "S": 0.184},
+    "Аммоний азотнокислый": {"N (NO3-)": 0.175, "N (NH4+)": 0.175},
+    "Сульфат магния": {"Mg": 0.098, "S": 0.13},
+    "Калий фосфорнокислый": {"P": 0.228, "K": 0.287}
 }
 
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
-        self.final_profile = BASE_PROFILE.copy()
+        self.profile = BASE_PROFILE.copy()
         self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
-        self.fertilizers_db = ALL_FERTILIZERS
-        self.target_nitrogen = {
-            'NO3': TOTAL_NITROGEN * (NO3_RATIO / (NO3_RATIO + NH4_RATIO)),
-            'NH4': TOTAL_NITROGEN * (NH4_RATIO / (NO3_RATIO + NH4_RATIO))
-        }
+        self.fertilizers = FERTILIZERS
+        
+        # Целевые значения азота
+        self.target_no3 = TOTAL_NITROGEN * (NO3_RATIO / (NO3_RATIO + NH4_RATIO))
+        self.target_nh4 = TOTAL_NITROGEN * (NH4_RATIO / (NO3_RATIO + NH4_RATIO))
 
     def calculate(self):
-        # 1. Вносим удобрения в строгом порядке
-        self._apply_magnesium_sulfate()
-        self._apply_calcium_nitrate()
-        self._apply_potassium_phosphate()
-        self._apply_ammonium_nitrate()
+        # 1. Вносим сульфат магния (покрываем Mg)
+        self._apply_fertilizer(
+            "Сульфат магния",
+            self.profile['Mg'],
+            "Mg",
+            {"S": "внесет S"}
+        )
+        
+        # 2. Вносим кальциевую селитру (покрываем Ca)
+        self._apply_fertilizer(
+            "Кальциевая селитра", 
+            self.profile['Ca'],
+            "Ca",
+            {"N (NO3-)": "внесет NO3"}
+        )
+        
+        # 3. Вносим калий фосфорнокислый (покрываем P)
+        self._apply_fertilizer(
+            "Калий фосфорнокислый",
+            self.profile['P'],
+            "P",
+            {"K": "внесет K"}
+        )
+        
+        # 4. Вносим аммоний азотнокислый (покрываем NH4)
+        nh4_needed = self.target_nh4
+        if nh4_needed > 0:
+            self._apply_fertilizer(
+                "Аммоний азотнокислый",
+                nh4_needed,
+                "N (NH4+)",
+                {"N (NO3-)": "внесет NO3"}
+            )
+        
+        # 5. Вносим калий сернокислый (покрываем K и S)
         self._apply_potassium_sulfate()
-        self._apply_potassium_nitrate()
-        self._apply_monopotassium_phosphate()
         
-        # 2. Корректируем отрицательные значения
-        self._adjust_negative_values()
+        # 6. Вносим калий азотнокислый (добираем K)
+        self._apply_potassium_nitrate()
         
-        # 3. Пересчитываем фактическое содержание азота
-        self._recalculate_actual_nitrogen()
+        # Пересчитываем фактическое содержание азота
+        self._recalculate_nitrogen()
         
         return self.results
 
-    def _apply_fertilizer(self, fert_name, grams, additions):
-        self.results[fert_name] = {
+    def _apply_fertilizer(self, name, need, main_element, additions):
+        if need <= 0:
+            return
+            
+        content = self.fertilizers[name][main_element]
+        grams = (need * self.volume) / (content * 1000)
+        
+        added = {}
+        for element, label in additions.items():
+            added_amount = grams * self.fertilizers[name][element] * 1000 / self.volume
+            added[label] = round(added_amount, 1)
+            self.profile[element] -= added_amount
+        
+        self.profile[main_element] = 0
+        self.results[name] = {
             'граммы': round(grams, 3),
             'миллиграммы': int(grams * 1000),
+            **added
         }
-        self.results[fert_name].update(additions)
-
-    def _adjust_negative_values(self):
-        for element in self.final_profile:
-            if self.final_profile[element] < 0:
-                self.final_profile[element] = 0
-
-    def _recalculate_actual_nitrogen(self):
-        # Пересчитываем фактическое содержание азота на основе внесенных удобрений
-        no3_added = 0
-        nh4_added = 0
-        
-        for fert in self.results.values():
-            if 'внесет NO3' in fert:
-                no3_added += fert['внесет NO3']
-            if 'внесет NH4' in fert:
-                nh4_added += fert['внесет NH4']
-        
-        self.final_profile['N (NO3-)'] = no3_added
-        self.final_profile['N (NH4+)'] = nh4_added
-
-    def _apply_magnesium_sulfate(self):
-        mg_need = self.final_profile['Mg']
-        if mg_need <= 0:
-            return
-            
-        mg_content = self.fertilizers_db["Сульфат магния"]["Mg"]
-        grams = (mg_need * self.volume) / (mg_content * 1000)
-        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)})
-        self.final_profile['Mg'] = 0
-
-    def _apply_calcium_nitrate(self):
-        ca_need = self.final_profile['Ca']
-        if ca_need <= 0:
-            return
-            
-        ca_content = self.fertilizers_db["Кальциевая селитра"]["Ca"]
-        grams = (ca_need * self.volume) / (ca_content * 1000)
-        added_n = grams * self.fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
-
-        self._apply_fertilizer("Кальциевая селитра", grams, {'внесет NO3': round(added_n, 1)})
-        self.final_profile['Ca'] = 0
-
-    def _apply_potassium_phosphate(self):
-        p_need = self.final_profile['P']
-        if p_need <= 0:
-            return
-            
-        p_content = self.fertilizers_db["Калий фосфорнокислый"]["P"]
-        grams = (p_need * self.volume) / (p_content * 1000)
-        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)})
-        self.final_profile['P'] = 0
-
-    def _apply_ammonium_nitrate(self):
-        nh4_need = self.target_nitrogen['NH4']
-        if nh4_need <= 0:
-            return
-            
-        nh4_content = self.fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"]
-        grams = (nh4_need * self.volume) / (nh4_content * 1000)
-        added_n = grams * self.fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
-
-        self._apply_fertilizer("Аммоний азотнокислый", grams, {
-            'внесет NO3': round(added_n, 1),
-            'внесет NH4': round(nh4_need, 1)
-        })
 
     def _apply_potassium_sulfate(self):
-        k_need = self.final_profile['K']
-        s_need = self.final_profile['S']
+        k_need = self.profile['K']
+        s_need = self.profile['S']
         
         if k_need <= 0 or s_need <= 0:
             return
             
-        k_content = self.fertilizers_db["Калий сернокислый"]["K"]
-        s_content = self.fertilizers_db["Калий сернокислый"]["S"]
+        k_content = self.fertilizers["Калий сернокислый"]["K"]
+        s_content = self.fertilizers["Калий сернокислый"]["S"]
         
+        # Рассчитываем максимально возможное внесение
         grams = min(
             (k_need * self.volume) / (k_content * 1000),
             (s_need * self.volume) / (s_content * 1000)
@@ -853,63 +809,62 @@ class NutrientCalculator:
         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)
-        })
+        self.profile['K'] -= added_k
+        self.profile['S'] -= added_s
+        self.results["Калий сернокислый"] = {
+            'граммы': round(grams, 3),
+            'миллиграммы': int(grams * 1000),
+            'внесет K': round(added_k, 1),
+            'внесет S': round(added_s, 1)
+        }
 
     def _apply_potassium_nitrate(self):
-        k_need = self.final_profile['K']
+        k_need = self.profile['K']
         if k_need <= 0:
             return
             
-        k_content = self.fertilizers_db["Калий азотнокислый"]["K"]
+        k_content = self.fertilizers["Калий азотнокислый"]["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._apply_fertilizer("Калий азотнокислый", grams, {'внесет NO3': round(added_n, 2)})
+        added_n = grams * self.fertilizers["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
-    def _apply_monopotassium_phosphate(self):
-        # Добавляем дополнительное фосфорное удобрение для полного покрытия
-        p_need = self.final_profile['P']
-        if p_need <= 0:
-            return
-            
-        p_content = self.fertilizers_db["Монофосфат калия"]["P"]
-        grams = (p_need * self.volume) / (p_content * 1000)
-        added_k = grams * self.fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
+        self.profile['K'] = 0
+        self.results["Калий азотнокислый"] = {
+            'граммы': round(grams, 3),
+            'миллиграммы': int(grams * 1000),
+            'внесет NO3': round(added_n, 1)
+        }
 
-        self.final_profile['K'] -= added_k
-        self._apply_fertilizer("Монофосфат калия", grams, {
-            'внесет P': round(p_need, 1),
-            'внесет K': round(added_k, 1)
-        })
-        self.final_profile['P'] = 0
+    def _recalculate_nitrogen(self):
+        # Считаем фактически внесенный азот
+        no3 = 0
+        nh4 = 0
+        
+        for fert in self.results.values():
+            if 'внесет NO3' in fert:
+                no3 += fert['внесет NO3']
+            if 'внесет NH4' in fert:
+                nh4 += fert['внесет NH4']
+        
+        self.profile['N (NO3-)'] = no3
+        self.profile['N (NH4+)'] = nh4
 
     def calculate_ec(self):
         return round(self.total_ppm / 700, 2)
 
     def print_report(self):
         print("\n" + "="*60)
-        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ПЕРЕСЧИТАННЫЙ):")
+        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА:")
         print("="*60)
-        table = [[el, round(val, 1)] for el, val in self.final_profile.items()]
+        table = [[el, round(val, 1)] for el, val in self.profile.items()]
         print(tabulate(table, headers=["Элемент", "ppm"]))
         
-        print("\nЦелевые значения азота:")
-        print(f"  N (NO3-): {round(self.target_nitrogen['NO3'], 1)} ppm")
-        print(f"  N (NH4+): {round(self.target_nitrogen['NH4'], 1)} ppm")
-
         print("\n" + "="*60)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*60)
         print(f"Общая концентрация: {round(self.total_ppm, 1)} ppm")
         print(f"EC: {self.calculate_ec()} mS/cm")
 
-        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
+        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ (6 штук):")
         fert_table = []
         for fert, data in self.results.items():
             details = [f"+{k}: {v} ppm" for k, v in data.items() if k.startswith('внесет')]
@@ -922,10 +877,10 @@ class NutrientCalculator:
         print(tabulate(fert_table, headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
 
         print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
-        deficit = {k: v for k, v in self.final_profile.items() if v > 0.1}
+        deficit = {k: round(v, 1) for k, v in self.profile.items() if v > 0.1}
         if deficit:
             for el, val in deficit.items():
-                print(f"  {el}: {round(val, 1)} ppm")
+                print(f"  {el}: {val} ppm")
         else:
             print("  Все элементы покрыты полностью")