Add tone number pinyin field

resolves jdlorimer#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_numpinyin(hanzi, note):
+    key = 'numpinyin'
+    separated = split_hanzi(hanzi, grouped=False)
+
+    if get_first(config['fields'][key], note) == '':
+        trans = transcribe(separated, 'pinyin', 'simp')
+        tone = ""
+        for t in trans:
+            tone += get_tone_number_pinyin(t) + ' '
+
+        set_all(config['fields'][key], note, to=tone.strip())
+
+
 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_numpinyin(hanzi, copy)
             fill_trad(hanzi, copy)
             fill_color(hanzi, copy)
             fill_sound(hanzi, copy)

chinese/config.json

@@ -34,6 +34,10 @@
             "法語",
             "法语"
         ],
+        "numpinyin": [
+            "NumPinyin",
+            "Tone Number"
+        ],
         "pinyin": [
             "Pinyin",
             "Reading",