上传文件

equipment_calculation/software_calculators.py

@@ -401,6 +401,14 @@ class MainGridBudgetCalculator(CalculatorBase):
         """创建主网预算计算策略"""
         return MainGridBudgetCalculationStrategy()
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def apply_quantity_fee_rules(self) -> None:
         """应用主网预算特定的工程量取费规则"""
         print(f"应用{self.software_type.name}特定的工程量取费规则 - 对节点进行预处理")
@@ -467,6 +475,14 @@ class MainGridBillCalculator(CalculatorBase):
         """创建主网清单计算策略"""
         return MainGridBillCalculationStrategy()
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def apply_quantity_fee_rules(self) -> None:
         """应用主网清单特定的工程量取费规则"""
         print(f"应用{self.software_type.name}特定的工程量取费规则 - 对节点进行预处理")
@@ -531,6 +547,14 @@ class DistributionBudgetCalculator(CalculatorBase):
     def __init__(self):
         super().__init__(DISTRIBUTION_BUDGET)
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def create_calculation_strategy(self) -> CalculationStrategy:
         """创建配网预算计算策略"""
         return DistributionBudgetCalculationStrategy()
@@ -562,6 +586,14 @@ class DistributionBillCalculator(CalculatorBase):
         """创建配网清单计算策略"""
         return DistributionBillCalculationStrategy()
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def apply_quantity_fee_rules(self) -> None:
         """应用配网清单特定的工程量取费规则"""
         print(f"应用{self.software_type.name}特定的工程量取费规则")
@@ -589,6 +621,14 @@ class TechnicalRenovationBudgetCalculator(CalculatorBase):
         """创建技改预算计算策略"""
         return TechnicalRenovationBudgetCalculationStrategy()
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def apply_quantity_fee_rules(self) -> None:
         """应用技改预算特定的工程量取费规则"""
         print(f"应用{self.software_type.name}特定的工程量取费规则")
@@ -616,6 +656,14 @@ class TechnicalRenovationBillCalculator(CalculatorBase):
         """创建技改清单计算策略"""
         return TechnicalRenovationBillCalculationStrategy()
 
+    def set_output_dir(self, output_dir):
+        """
+        设置计算结果的输出目录
+
+        :param output_dir: 输出目录的路径
+        """
+        self.output_dir = output_dir
+
     def apply_quantity_fee_rules(self) -> None:
         """应用技改清单特定的工程量取费规则"""
         print(f"应用{self.software_type.name}特定的工程量取费规则")