Python设计模式之状态模式原理与用法详解

状态模式的核心原理

状态模式（State Pattern）是一种行为型设计模式，其核心思想是：当一个对象的内在状态改变时，允许改变其行为，这个对象看起来就像是改变了其类。这种模式特别适用于以下场景：

• 控制对象状态转换的条件表达式过于复杂，导致代码难以维护。
• 对象的行为高度依赖于其当前状态，且必须在运行时刻根据状态动态改变行为。
• 需要将状态的判断逻辑从主业务逻辑中剥离，转移到表示不同状态的一系列类中。

通过状态模式，可以将复杂的条件判断逻辑简化，使代码结构更加清晰，易于扩展和维护。

Python实现示例

以下代码展示了状态模式在Python中的具体实现。我们以一个工作场景为例，根据当前时间（小时）的不同，输出不同的工作状态。

#!/usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = 'Andy'
"""
大话设计模式
设计模式——状态模式
状态模式(State Pattern):当一个对象的内在状态改变时允许改变其行为，这个对象看起来像是改变了其类
应用场景:当控制一个对象的状态转换的条件表达式过于复杂时,把状态的判断逻辑转移到表示不同状态的一系列类当中,可以把复杂的判断逻辑简化
(当一个对象的行为取决于它的状态,并且它必须在运行时刻根据状态改变他的行为)
"""
class State(object):
  def __init__(self):
    pass
  def write_program(self, w):
    pass
class Work(object):
  def __init__(self):
    self.hour = 9
    self.curr = ForenoonState()
  def set_state(self, s):
    self.curr = s
  def write_program(self):
    self.curr.write_program(self)
class ForenoonState(State):
  def write_program(self, w):
    if w.hour < 12:
      print "当前时间:%s点,"%w.hour,"精神百倍"
    else:
      w.set_state(AfternoonState())
      w.write_program()
class AfternoonState(State):
  def write_program(self, w):
    if w.hour < 17:
      print "当前时间:%s点,"%w.hour,"状态还行,继续努力"
    else:
      w.set_state(EveningState())
      w.write_program()
class EveningState(State):
  def write_program(self, w):
    if w.hour < 21:
      print "当前时间:%s点,"%w.hour,"加班呢,疲劳了"
    else:
      w.set_state(SleepState())
      w.write_program()
class SleepState(State):
  def write_program(self, w):
    print "当前时间:%s点,"%w.hour,"不行了,睡着了"
if __name__=="__main__":
  work = Work()
  work.hour = 9
  work.write_program()
  work.hour = 15
  work.write_program()
  work.hour = 20
  work.write_program()
  work.hour = 22
  work.write_program()

代码结构解析

上述代码包含以下几个关键部分：

1. State类：定义状态接口，包含write_program方法，所有具体状态类都继承此类。
2. Work类：上下文类，维护当前状态curr，并提供set_state方法切换状态。其write_program方法委托给当前状态对象执行。
3. 具体状态类（如ForenoonState、AfternoonState等）：继承自State，实现具体的行为逻辑。每个状态类中根据当前时间判断是否切换状态，并调用Work的set_state方法切换至下一个状态。

运行结果与状态切换

通过设置不同的hour值，可以观察到状态的自然切换：

• 当hour = 9时，输出“精神百倍”，处于上午状态。
• 当hour = 15时，输出“状态还行,继续努力”，处于下午状态。
• 当hour = 20时，输出“加班呢,疲劳了”，处于晚间状态。
• 当hour = 22时，输出“不行了,睡着了”，进入睡眠状态。

这种设计避免了在Work类中使用大量的if-else或switch-case语句，使得新增状态或修改状态逻辑时，只需修改或新增对应的状态类，符合开闭原则。

总结

状态模式通过将状态相关的行为封装到独立的类中，有效简化了复杂的状态转换逻辑。在Python中，利用其动态特性，可以灵活地实现状态切换，提升代码的可读性和可维护性。对于行为随状态动态变化的场景，状态模式是一个值得推荐的设计方案。