pattern_state.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/2/5 15:05
# @Author  : ZhangSheng@xiaoyezi.com
# @File    : pattern_state.py

__author__ = 'kevinlu1010@qq.com'

from abc import ABCMeta, abstractmethod


class State():
    __metaclass__ = ABCMeta

    @abstractmethod
    def write_code(self):
        pass


class Morning(State):
    def write_code(self, work):
        if work.hour <= 12 and work.hour > 8:
            print '上午工作，精神百倍'
        else:

            work.set_status(Noon())
            work.write_code(work)


class Noon(State):
    def write_code(self, work):
        if work.hour <= 14 and work.hour>12 :
            print '中午工作，困，想午休'
        else:
            work.set_status(Afternoon())
            work.write_code(work)


class Afternoon(State):
    def write_code(self, work):
        if work.hour <= 18 and work.hour>14:
            print '下午工作，状态不错'
        else:
            work.set_status(Eve())
            work.write_code(work)


class Eve(State):
    def write_code(self, work):
        if work.hour <= 22 and work.hour>18:
            print '加班了，状态不太好'
        else:
            work.set_status(Night())
            work.write_code(work)


class Night(State):
    def write_code(self, work):
        if work.hour <= 8 or work.hour > 22:
            print '不行了，要睡觉了'
        else:
            work.set_status(Morning())
            work.write_code(work)


class Work():
    def __init__(self, hour):
        self.hour = hour
        self.state = Morning()

    def set_status(self, state):
        self.state = state

    def write_code(self, work):
        self.state.write_code(work)


if __name__ == '__main__':
    work = Work(10)
    for hour in (3, 11, 12, 13, 14, 17, 19, 22, 23,12):
        work.hour = hour
        print '%d点，' % hour
        work.write_code(work)