From 4edd72c103afc700e5d504a755094545717eee45 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Sun, 17 Jul 2016 11:38:03 -0700
Subject: [PATCH] Fixups

---
 yomi_base/anki_bridge.py  | 62 +++++++++++++++++++++++++++++++++++++--
 yomi_base/anki_connect.py |  3 +-
 2 files changed, 62 insertions(+), 3 deletions(-)

diff --git a/yomi_base/anki_bridge.py b/yomi_base/anki_bridge.py
index ad4ade8..46e71e6 100644
--- a/yomi_base/anki_bridge.py
+++ b/yomi_base/anki_bridge.py
@@ -18,10 +18,56 @@
 
 import anki
 import aqt
+import hashlib
+import urllib2
 
 
+#
+# Audio helpers
+#
+
+def audioBuildFilename(kana, kanji):
+    filename = u'yomichan_{}'.format(kana)
+    if kanji:
+        filename += u'_{}'.format(kanji)
+    filename += u'.mp3'
+    return filename
+
+
+def audioDownload(kana, kanji):
+    url = 'http://assets.languagepod101.com/dictionary/japanese/audiomp3.php?kanji={}'.format(urllib2.quote(kanji.encode('utf-8')))
+    if kana:
+        url += '&kana={}'.format(urllib2.quote(kana.encode('utf-8')))
+
+    try:
+        resp = urllib2.urlopen(url)
+    except urllib2.URLError:
+        return None
+
+    if resp.code != 200:
+        return None
+
+    return resp.read()
+
+
+def audioIsPlaceholder(data):
+    m = hashlib.md5()
+    m.update(data)
+    return m.hexdigest() == '7e2c2f954ef6051373ba916f000168dc'
+
+
+def audioInject(note, fields, filename):
+    for field in fields:
+        if field in note:
+            note[field] += u'[sound:{}]'.format(filename)
+
+
+#
+# Anki
+#
+
 class Anki:
-    def addNote(self, deckName, modelName, fields, tags=[]):
+    def addNote(self, deckName, modelName, fields, tags, audio):
         collection = self.collection()
         if collection is None:
             return
@@ -30,8 +76,14 @@ class Anki:
         if note is None:
             return
 
-        self.startEditing()
+        if audio is not None and len(audio['fields']) > 0:
+            data = audioDownload(audio['kana'], audio['kanji'])
+            if data is not None and not audioIsPlaceholder(data):
+                filename = audioBuildFilename(audio['kana'], audio['kanji'])
+                audioInject(note, audio['fields'], filename)
+                self.media().writeData(filename, data)
 
+        self.startEditing()
         collection.addNote(note)
         collection.autosave()
 
@@ -94,6 +146,12 @@ class Anki:
         return self.window().col
 
 
+    def media(self):
+        collection = self.collection()
+        if collection is not None:
+            return collection.media
+
+
     def modelNames(self):
         collection = self.collection()
         if collection is not None:
diff --git a/yomi_base/anki_connect.py b/yomi_base/anki_connect.py
index d3b554e..6783b6b 100644
--- a/yomi_base/anki_connect.py
+++ b/yomi_base/anki_connect.py
@@ -70,7 +70,8 @@ class AnkiConnect:
             note['deckName'],
             note['modelName'],
             note['fields'],
-            note['tags']
+            note['tags'],
+            note.get('audio')
         )