summaryrefslogtreecommitdiffstats
path: root/attain_vocab_to_anki_py/io.py
blob: 4288e06d99f7ba5a059f00016def1b6bd4ea0886 (plain)
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
from os import path
from dataclasses import dataclass

@dataclass
class Entry:
  kanji: str
  hiragana: str
  english: str
  meanings: str

def read_from_file(path_str: str) -> list:
  p = path.join(path_str)
  lines = []

  with open(p, "r") as f:
    lines = f.readlines()

  entries = []
  for line in lines:
    kanji, hiragana, english = line.split(",", 2)
    entries.append(Entry(kanji=kanji.strip(),
      hiragana=hiragana.strip(),
      english=english.strip(),
      meanings=""))

  return entries

def _entry_to_row(entry: Entry, delimiter: str = ";") -> str:
  s = delimiter.join([entry.kanji, entry.hiragana, entry.english, entry.meanings])
  return f"{s}\n"

def write_to_file(path_str: str, entries: list):
  p = path.join(path_str)

  lines = map(_entry_to_row, entries)

  with open(p, "w") as f:
    f.writelines(lines)