Skip to content

Commit d7219ff

Browse files
authored
Merge pull request #344 from rednafi/master
Added type hints to mediator pattern
2 parents cdefb80 + f49d4e9 commit d7219ff

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

makefile renamed to Makefile

File renamed without changes.

patterns/behavioral/mediator.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,25 +8,27 @@
88
Encapsulates how a set of objects interact.
99
"""
1010

11+
from __future__ import annotations
12+
1113

1214
class ChatRoom:
1315
"""Mediator class"""
1416

15-
def display_message(self, user, message):
16-
print("[{} says]: {}".format(user, message))
17+
def display_message(self, user: User, message: str) -> None:
18+
print(f"[{user} says]: {message}")
1719

1820

1921
class User:
2022
"""A class whose instances want to interact with each other"""
2123

22-
def __init__(self, name):
24+
def __init__(self, name: str) -> None:
2325
self.name = name
2426
self.chat_room = ChatRoom()
2527

26-
def say(self, message):
28+
def say(self, message: str) -> None:
2729
self.chat_room.display_message(self, message)
2830

29-
def __str__(self):
31+
def __str__(self) -> str:
3032
return self.name
3133

3234

0 commit comments

Comments
 (0)