create deck when importing, close #20

src/__main__.py

@@ -8,7 +8,7 @@
 
 from NoteType import Cloze, Choices, ListCloze, TableCloze, QA
 from helper.genankiHelper import getDeck, exportDeck
-from helper.ankiConnectHelper import addNotes
+from helper.ankiConnectHelper import addNotes, checkOnline
 import config
 
 
@@ -60,6 +60,9 @@ def HandlePost(text):
 def main():
     print("Starting...")
     noteLists = []
+    if config.output == "" and not checkOnline():  # import into Anki
+        print("ERROR: failed to connect with Anki Connect.")
+        sys.exit()
     for file in config.file_list:
         print("file: " + file)
         f = open(file, "r", encoding="utf-8")

src/helper/ankiConnectHelper.py

@@ -2,6 +2,14 @@
 import json
 
 
+def checkOnline():
+    try:
+        getDeckNames()
+    except Exception:
+        return False
+    return True
+
+
 def request(action, **params):
     return {"action": action, "params": params, "version": 6}
 
@@ -23,7 +31,7 @@ def invoke(action, **params):
 
 def addNote(target, deck, options={"allowDuplicate": True}, retry=True):
     try:
-        invoke("addNote", note={
+        return invoke("addNote", note={
             "deckName": deck,
             "modelName": target.model.modelName,
             "fields": target.outputfields,
@@ -32,17 +40,28 @@ def addNote(target, deck, options={"allowDuplicate": True}, retry=True):
         }
         )
     except Exception as e:
-        if len(e.args) > 0 and "model" in e.args[0] and \
-                target.model.modelName not in getModelNamesAndIds().keys():
+        if len(e.args) == 0:
+            return e
+        if "model" in e.args[0] and target.model.modelName not in getModelNamesAndIds().keys():
             createModel(target.model)
-            if retry:
-                addNote(target, deck, options, False)
+        elif "deck was not found" in e.args[0] and deck not in getDeckNames():
+            createDeck(deck)
+        if retry:
+            return addNote(target, deck, options, False)
+
+
+def createDeck(deckName):
+    return invoke("createDeck", deck=deckName)
 
 
 def getModelNamesAndIds():
     return invoke("modelNamesAndIds")
 
 
+def getDeckNames():
+    return invoke("deckNames")
+
+
 def createModel(model):
     return invoke("createModel",
                   modelName=model.modelName,