oop-solid-design-patterns/problem2_isp.py at main · khadijafatima409/oop-solid-design-patterns · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""
Problem 2: Interface Segregation Principle (ISP)
"""

from abc import ABC, abstractmethod


# ---------- BAD DESIGN ----------
class Worker(ABC):
    @abstractmethod
    def work(self): pass

    @abstractmethod
    def eat(self): pass

    @abstractmethod
    def sleep(self): pass

    @abstractmethod
    def code(self): pass

    @abstractmethod
    def manage(self): pass

    @abstractmethod
    def recharge(self): pass


class BadRobot(Worker):
    def work(self): print("Robot working")
    def eat(self): pass
    def sleep(self): pass
    def code(self): pass
    def manage(self): pass
    def recharge(self): print("Robot recharging")


# ---------- GOOD DESIGN ----------
class Workable(ABC):
    @abstractmethod
    def work(self): pass


class Eatable(ABC):
    @abstractmethod
    def eat(self): pass


class Sleepable(ABC):
    @abstractmethod
    def sleep(self): pass


class Codable(ABC):
    @abstractmethod
    def code(self): pass


class Manageable(ABC):
    @abstractmethod
    def manage(self): pass


class Rechargeable(ABC):
    @abstractmethod
    def recharge(self): pass


class Developer(Workable, Eatable, Sleepable, Codable):
    def work(self): print("Developer working")
    def eat(self): print("Developer eating")
    def sleep(self): print("Developer sleeping")
    def code(self): print("Developer coding")


class Manager(Workable, Eatable, Sleepable, Manageable):
    def work(self): print("Manager working")
    def eat(self): print("Manager eating")
    def sleep(self): print("Manager sleeping")
    def manage(self): print("Manager managing")


class Robot(Workable, Rechargeable):
    def work(self): print("Robot working")
    def recharge(self): print("Robot recharging")


if __name__ == "__main__":
    dev = Developer()
    manager = Manager()
    robot = Robot()

    dev.code()
    manager.manage()
    robot.recharge()