Update the CityFormatter to have the header as an argument.

This commit is contained in:
2023-06-20 13:38:56 +03:00
parent ca3bb35b74
commit d99c246de2
2 changed files with 20 additions and 2 deletions
+2 -2
View File
@@ -5,8 +5,8 @@ class CityFormatter(AbstractFormatter):
name = city['name']
return f"- {name}\n"
def format_all(self, cities: dict) -> str:
formated_message = "Cities:\n\n"
def format_all(self, cities: dict, header: str = "# Міста\n\n") -> str:
formated_message = header
for city in cities['data']:
formated_item = self.format(city)