forked from amirbigg/python-design-patterns
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmediator.py
More file actions
86 lines (59 loc) · 1.8 KB
/
mediator.py
File metadata and controls
86 lines (59 loc) · 1.8 KB
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
"""
Mediator
- a behavioral design pattern that lets you reduce chaotic dependencies between objects.
The pattern restricts direct communications between the objects and forces them to collaborate
only via a mediator object.
"""
import abc
class AbstractMediator(abc.ABC):
@abc.abstractmethod
def notify(self, sender, event):
pass
class ConcreteMediator(AbstractMediator):
def __init__(self, comp_a, comp_b):
self.comp_a = comp_a
self.comp_a.set_mediator(self)
self.comp_b = comp_b
self.comp_b.set_mediator(self)
def notify(self, sender, event):
self.comp_b.receive(sender, event)
class AbstractComponent(abc.ABC):
def __init__(self, mediator = None):
self._mediator = mediator
def set_mediator(self, mediator):
self._mediator = mediator
@abc.abstractmethod
def receive(self, sender, event):
pass
@abc.abstractmethod
def notify(self, event):
pass
class Component1(AbstractComponent):
def receive(self, sender, event):
print(f'Component 1 received event ({sender.__class__.__name__} {event})')
def notify(self, event):
self._mediator.notify(self, event)
def do_a(self):
print('Component 1 does A.')
self.notify('A')
class Component2(AbstractComponent):
def receive(self, sender, event):
print(f'Component 2 received event ({sender.__class__.__name__} {event})')
def notify(self, event):
self._mediator.notify(self, event)
def do_b(self):
print('Component 2 does B.')
self.notify('B')
class Component3(AbstractComponent):
def receive(self, sender, event):
print(f'Component 3 received event ({sender.__class__.__name__} {event})')
def notify(self, event):
self._mediator.notify(self, event)
def do_c(self):
print('Component 3 does C.')
self.notify('C')
c1 = Component1()
c2 = Component2()
c3 = Component3()
mediator = ConcreteMediator(c1, c2)
c1.do_a()