Fix Writer Model

Author: soheilrt

models/writer.py

@@ -1,5 +1,5 @@
 from abc import abstractmethod, ABC
-from typing import List
+from typing import List, Dict
 
 from models.word import WordDefinition
 
@@ -10,7 +10,7 @@ class Writer(ABC):
     """
 
     @abstractmethod
-    def write(self, data: List[WordDefinition]):
+    def write(self, data: Dict[str, List[WordDefinition]]):
         """
         Write data to a file.
         """

writers/json_writer.py

@@ -1,5 +1,5 @@
-from typing import List
 import json
+from typing import List, Dict
 
 from models.word import WordDefinition
 from models.writer import Writer
@@ -13,5 +13,12 @@ def __init__(self, file_name: str, indent=4, sort_keys=True, ensure_ascii=False)
         self.sort_keys = sort_keys
         self.ensure_ascii = ensure_ascii
 
-    def write(self, data: List[WordDefinition]):
-        json.dump(data, self.file, cls=WordDefinitionJsonEncoder, indent=self.indent, sort_keys=self.sort_keys,ensure_ascii=self.ensure_ascii)
+    def write(self, data: Dict[str, List[WordDefinition]]):
+        json.dump(
+            data,
+            self.file,
+            cls=WordDefinitionJsonEncoder,
+            indent=self.indent,
+            sort_keys=self.sort_keys,
+            ensure_ascii=self.ensure_ascii
+        )