Add support for pinyin field with tone numbers

resolves #56

Author: sc-rwirnata

chinese/behavior.py

@@ -28,6 +28,7 @@
     sanitize_transcript,
     split_transcript,
     transcribe,
+    get_tone_number_pinyin,
 )
 from .translate import translate
 from .util import (
@@ -155,6 +156,19 @@ def fill_transcript(hanzi, note):
     return n_filled
 
 
+def fill_tone_numbers(hanzi, note):
+    filled = False
+    separated = split_hanzi(hanzi, grouped=False)
+
+    if get_first(config['fields']['toneNumbers'], note) == '':
+        trans = transcribe(separated, 'pinyin', 'simp')
+        tones = [get_tone_number_pinyin(t) for t in trans]
+        set_all(config['fields']['toneNumbers'], note, to=' '.join(tones))
+        filled = True
+
+    return filled
+
+
 def reformat_transcript(note, group, target):
     if target == 'bopomofo':
         return
@@ -311,6 +325,7 @@ def update_fields(note, focus_field, fields):
             fill_all_defs(hanzi, copy)
             fill_classifier(hanzi, copy)
             fill_transcript(hanzi, copy)
+            fill_tone_numbers(hanzi, copy)
             fill_trad(hanzi, copy)
             fill_color(hanzi, copy)
             fill_sound(hanzi, copy)

chinese/config.json

@@ -34,6 +34,11 @@
             "法語",
             "法语"
         ],
+        "toneNumbers": [
+            "Tone Numbers",
+            "Tones",
+            "NumPinyin"
+        ],
         "pinyin": [
             "Pinyin",
             "Reading",

tests/test_config.py

@@ -37,6 +37,7 @@ def test_get_fields_no_arg(self):
             'Hanzi (Color)',
             'Hanzi',
             'Measure Word',
+            "NumPinyin",
             'Pinyin (Taiwan)',
             'Pinyin',
             'Reading',
@@ -55,6 +56,8 @@ def test_get_fields_no_arg(self):
             'Sound (Mandarin)',
             'Sound',
             'Spoken',
+            'Tone Numbers',
+            'Tones',
             'Trad',
             'Trad.',
             'TradColor',