From eafd3c94ddc6a17a25c7f434dc3d1156194ff9db Mon Sep 17 00:00:00 2001
From: Andreas Kienle <andreaskienle@gmx.de>
Date: Sun, 23 Feb 2020 11:59:19 +0100
Subject: [PATCH] Accept multiple audio files in note creation and update

---
 plugin/__init__.py | 56 +++++++++++++++++++++++++++-------------------
 1 file changed, 33 insertions(+), 23 deletions(-)

diff --git a/plugin/__init__.py b/plugin/__init__.py
index c8a18ff..d158709 100644
--- a/plugin/__init__.py
+++ b/plugin/__init__.py
@@ -449,29 +449,8 @@ class AnkiConnect:
     def addNote(self, note):
         ankiNote = self.createNote(note)
 
-        audio = note.get('audio')
-        if audio is not None and len(audio['fields']) > 0:
-            try:
-                data = util.download(audio['url'])
-                skipHash = audio.get('skipHash')
-                if skipHash is None:
-                    skip = False
-                else:
-                    m = hashlib.md5()
-                    m.update(data)
-                    skip = skipHash == m.hexdigest()
-
-                if not skip:
-                    audioFilename = self.media().writeData(audio['filename'], data)
-                    for field in audio['fields']:
-                        if field in ankiNote:
-                            ankiNote[field] += u'[sound:{}]'.format(audioFilename)
-
-            except Exception as e:
-                errorMessage = str(e).replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
-                for field in audio['fields']:
-                    if field in ankiNote:
-                        ankiNote[field] += errorMessage
+        audioObjectOrList = note.get('audio')
+        self.addAudio(ankiNote, audioObjectOrList)
 
         collection = self.collection()
         self.startEditing()
@@ -484,6 +463,34 @@ class AnkiConnect:
         return ankiNote.id
 
 
+    def addAudio(self, ankiNote, audioObjectOrList):
+        if audioObjectOrList is not None:
+            audioList = audioObjectOrList if isinstance(audioObjectOrList, list) else [audioObjectOrList]
+            for audio in audioList:
+                if audio is not None and len(audio['fields']) > 0:
+                    try:
+                        data = util.download(audio['url'])
+                        skipHash = audio.get('skipHash')
+                        if skipHash is None:
+                            skip = False
+                        else:
+                            m = hashlib.md5()
+                            m.update(data)
+                            skip = skipHash == m.hexdigest()
+
+                        if not skip:
+                            audioFilename = self.media().writeData(audio['filename'], data)
+                            for field in audio['fields']:
+                                if field in ankiNote:
+                                    ankiNote[field] += u'[sound:{}]'.format(audioFilename)
+
+                    except Exception as e:
+                        errorMessage = str(e).replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
+                        for field in audio['fields']:
+                            if field in ankiNote:
+                                ankiNote[field] += errorMessage
+
+
     @util.api()
     def canAddNote(self, note):
         try:
@@ -502,6 +509,9 @@ class AnkiConnect:
             if name in ankiNote:
                 ankiNote[name] = value
 
+        audioObjectOrList = note['audio']
+        self.addAudio(ankiNote, audioObjectOrList)
+
         ankiNote.flush()