-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexample.py
83 lines (65 loc) · 1.89 KB
/
example.py
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env python3
import csv
import random
import genanki
# Filename of the data file
data_filename = "HSK Official With Definitions 2012 L1.txt"
# Filename of the Anki deck to generate
deck_filename = "HSK1.apkg"
# Title of the deck as shown in Anki
anki_deck_title = "HSK1"
# Name of the card model
anki_model_name = "HSK"
# Create the deck model
model_id = random.randrange(1 << 30, 1 << 31)
style = """
.card {
font-family: arial;
font-size: 24px;
text-align: center;
color: black;
background-color: white;
}
.hanzi {
font-size: 64px;
}
"""
anki_model = genanki.Model(
model_id,
anki_model_name,
fields=[{"name": "hanzi"}, {"name": "pinyin"}, {"name": "meaning"}],
templates=[
{
"name": "Card 1",
"qfmt": '<p class="hanzi">{{hanzi}}</p>',
"afmt": '{{FrontSide}}<hr id="answer"><p class="pinyin">{{pinyin}}</p><p class="meaning">{{meaning}}</p>',
},
{
"name": "Card 2",
"qfmt": '<p class="meaning">{{meaning}}</p>',
"afmt": '{{FrontSide}}<hr id="answer"><p class="hanzi">{{hanzi}}</p><p class="pinyin">{{pinyin}}</p>',
},
],
css=style,
)
# The list of flashcards
anki_notes = []
with open(data_filename, "r") as csv_file:
csv_reader = csv.reader(csv_file, delimiter="\t")
for row in csv_reader:
anki_note = genanki.Note(
model=anki_model,
# simplified writing, pinyin, meaning
fields=[row[0], row[3], row[4]],
)
anki_notes.append(anki_note)
# Shuffle flashcards
random.shuffle(anki_notes)
anki_deck = genanki.Deck(model_id, anki_deck_title)
anki_package = genanki.Package(anki_deck)
# Add flashcards to the deck
for anki_note in anki_notes:
anki_deck.add_note(anki_note)
# Save the deck to a file
anki_package.write_to_file(deck_filename)
print("Created deck with {} flashcards".format(len(anki_deck.notes)))