from __future__ import print_function
from future import standard_library
standard_library.install_aliases()
from builtins import map
from builtins import range
from vsm import *
from vsm.viewer.wrappers import doc_label_name

import os.path
from collections import defaultdict

# load in the configuration file
from configparser import ConfigParser as ConfigParser
config_file = r"$config_file" 
config = ConfigParser({
        'topic_range': None,
        'topics': None,
        'sentences' : 'false'})
config.read(config_file)

# load the corpus
if config.getboolean('main','sentences'):
    from vsm.extensions.ldasentences import CorpusSent
    c = CorpusSent.load(config.get('main', 'corpus_file'))
else:
    c = Corpus.load(config.get('main', 'corpus_file'))
context_type = config.get('main', 'context_type')
ctx_metadata = c.view_metadata(context_type)
all_ids = ctx_metadata[doc_label_name(context_type)]

# create topic model patterns
pattern = config.get('main', 'model_pattern')
if config.get('main', 'topic_range'):
    topic_range = list(map(int, config.get('main', 'topic_range').split(',')))
    topic_range = list(range(*topic_range))
if config.get('main', 'topics'):
    topic_range = eval(config.get('main', 'topics'))

# load the topic models
class keydefaultdict(defaultdict):
    """ Solution from: http://stackoverflow.com/a/2912455 """
    def __missing__(self, key):
        if self.default_factory is None:
            raise KeyError( key )
        else:
            ret = self[key] = self.default_factory(key)
            return ret
def load_model(k):
    if k in topic_range:
        return LdaCgsSeq.load(pattern.format(k))
    else:
        raise KeyError("No model trained for k={}.".format(k))
def load_viewer(k):
    """ Function to dynamically load the LdaCgsViewer. 
        Failure handling for missing keys is handled by `load_model`"""
    return LdaCgsViewer(c,lda_m[k])

lda_m = keydefaultdict(load_model)
lda_v = keydefaultdict(load_viewer)
print(topic_range)
print(pattern)

from topicexplorer.lib.color import get_topic_colors
