灵活请假功能：工厂模式与策略模式的结合

2022-11-12 03:55:33

工厂模式与策略模式：构建一个灵活的请假功能

工厂模式

想象一下你正在一家面包店工作，顾客可以定制各种面包。为了满足他们的需求，你可以使用一个工厂模式 。这个模式允许你创建一个统一的接口，来创建不同类型的面包，而无需指定每个面包的具体类型。

对于请假功能，你可以使用一个类似的方法。创建一个工厂模式，来创建不同类型的请假单。例如，你可以有一个病假单工厂、一个事假单工厂和一个年假单工厂。当用户需要请假时，你可以根据他们的请假类型，选择相应的工厂，然后使用该工厂创建请假单对象。

class LeaveFactory:
    @staticmethod
    def create_leave(leave_type):
        if leave_type == "sick":
            return SickLeave()
        elif leave_type == "casual":
            return CasualLeave()
        elif leave_type == "annual":
            return AnnualLeave()

策略模式

现在，让我们想象一下，你需要处理不同的付款方式。策略模式 可以帮助你实现这个目标。该模式允许你定义一系列算法，并将每个算法封装成单独的策略。这些策略可以根据需要进行互换。

对于请假功能，你可以使用策略模式来实现不同的请假策略。例如，你可以创建一个病假策略、一个事假策略和一个年假策略。当用户需要请假时，你可以根据他们的请假类型，选择相应的策略，然后使用该策略来计算请假天数或生成请假证明。

class LeaveStrategy:
    def calculate_days(self):
        pass

    def generate_proof(self):
        pass

class SickLeaveStrategy(LeaveStrategy):
    def calculate_days(self):
        return 5

    def generate_proof(self):
        return "Doctor's note"

class CasualLeaveStrategy(LeaveStrategy):
    def calculate_days(self):
        return 2

    def generate_proof(self):
        return "Self-declaration"

class AnnualLeaveStrategy(LeaveStrategy):
    def calculate_days(self, years_of_service):
        if years_of_service < 5:
            return 14
        elif years_of_service < 10:
            return 21
        else:
            return 28

    def generate_proof(self):
        return "Leave application"

结合工厂模式和策略模式

现在，让我们把工厂模式和策略模式结合起来，创建一个灵活且可扩展的请假功能。这种方法允许你根据不同的需求，创建不同的请假单和请假策略。你可以轻松地添加新的请假类型和请假策略，而无需更改核心代码。

class LeaveProcessor:
    def __init__(self, leave_factory, leave_strategy):
        self.leave_factory = leave_factory
        self.leave_strategy = leave_strategy

    def process_leave(self, leave_type, years_of_service=0):
        leave = self.leave_factory.create_leave(leave_type)
        days = self.leave_strategy.calculate_days(years_of_service)
        proof = self.leave_strategy.generate_proof()
        return leave, days, proof