Add note to Entry

2 files changed, 13 insertions, 10 deletions

diff --git a/kanji_info_fetcher/entry.py b/kanji_info_fetcher/entry.py
new file mode 100644
index 0000000..a361f67
--- /dev/null
+++ b/kanji_info_fetcher/entry.py
@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+
+@dataclass
+class Entry:
+  kanji: str
+  hiragana: str
+  english: str
+  meanings: str
+  note: str
diff --git a/kanji_info_fetcher/io.py b/kanji_info_fetcher/io.py
index 90c9507..bb0d69c 100644
--- a/kanji_info_fetcher/io.py
+++ b/kanji_info_fetcher/io.py
@@ -1,12 +1,5 @@
 from os import path
-from dataclasses import dataclass
-
-@dataclass
-class Entry:
-  kanji: str
-  hiragana: str
-  english: str
-  meanings: str
+from .entry import Entry
 
 def read_from_file(path_str: str, row_format: str) -> list:
   p = path.join(path_str)
@@ -32,13 +25,14 @@ def read_from_file(path_str: str, row_format: str) -> list:
       kanji=value_dict.get("kanji", "").strip(),
       hiragana=value_dict.get("hiragana", "").strip(),
       english=value_dict.get("english", "").strip(),
-      meanings=value_dict.get("meanings", "").strip()
+      meanings=value_dict.get("meanings", "").strip(),
+      note=value_dict.get("note", "").strip()
     ))
 
   return entries
 
 def _entry_to_row(entry: Entry, delimiter: str = "§") -> str:
-  s = delimiter.join([entry.kanji, entry.hiragana, entry.english, entry.meanings])
+  s = delimiter.join([entry.kanji, entry.hiragana, entry.english, entry.meanings, entry.note])
   return f"{s}\n"
 
 def write_to_file(path_str: str, entries: list):