main.py

#!/usr/bin/env python
import sys
import json
import importlib
import os
import click

sys.path.append('anki')
from anki import Collection as aopen

def initAnkiModule(data, collection, card_type):
    if bool(collection) == False or "deck" not in data:
        print('Collection or deck is not found!')
        return None
    deck = aopen(collection)
    deckId = deck.decks.id(data['deck'])

    deck.decks.select(deckId)
    model = deck.models.byName(card_type.GetCardType(deck.models))
    if model is not None:
        model['did'] = deckId
        deck.models.save(model)
        deck.models.setCurrent(model)
    return deck

def handleProfile(data, collection, download_dir):
    print('words file:{}'.format(data['file']))
    print('deck file:{}'.format(data['deck']))
    print('dict_source :{}'.format(data['dict_source']))
    print('card_type:{}'.format(data['card_type'] if 'card_type' in data else 'Basic'))

    inputFilePath = 'input/{}'.format(data['file'])

    if 'file' not in data or not os.path.exists(inputFilePath):
        print("No input file, Exit")
        return False

    input_file = "{}/{}".format(os.getcwd(), inputFilePath)

    card_type = data['card_type'] if 'card_type' in data else 'basic'

    dict_source = importlib.import_module('module.{}'.format(data['dict_source'].lower()))
    card_type = importlib.import_module('cardtype.{}'.format(card_type))
    deck = initAnkiModule(data, collection, card_type)
    with open(input_file , encoding='utf-8') as word_list:
        for word in word_list:
            result = dict_source.LookUp(word, data, download_dir)

            if result is None:
                continue
            elif result is False:
                deck.save()
                deck.close()
                return False
            card_data = card_type.MakeCard(result)

            if 0 == len(card_data):
                continue

            card = deck.newNote()
            for key in card_data:
                card[key] = card_data[key]
            try:
                deck.addNote(card)
            except(Exception, e):
                if hasattr(e, "data"):
                    sys.exit("ERROR: Cound not add {}:{}", e.data["field"], e.data['type'])
                else:
                    sys.exit(e)
    deck.save()
    deck.close()

def load_config(path):
    with open(path, encoding='utf-8') as data_file:
        return json.load(data_file)

if '__main__':
    if len(sys.argv) >= 2:
        config_path = sys.argv[1]
    else:
        config_path = 'config.json'
    data = load_config(config_path)
    collection = data['collection']
    download_dir = data['download_dir']
    for profile in data['profiles']:
        if handleProfile(profile, collection, download_dir) == False:
            break