from scrapy import log
from scrapy.utils.project import get_project_settings
import ConfigParser

class Configurator:
    """
    A helper class in the fourmi class. This class is used to process the settings as set
    from one of the Fourmi applications.
    """

    def __init__(self):
        self.scrapy_settings = get_project_settings()


    def set_output(self, filename, fileformat):
        """
        This function manipulates the Scrapy output file settings that normally would be set in the settings file.
        In the Fourmi project these are command line arguments.
        :param filename: The filename of the file where the output will be put.
        :param fileformat: The format in which the output will be.
        """

        if filename != 'results.*format*':
            self.scrapy_settings.overrides["FEED_URI"] = filename
        elif fileformat == "jsonlines":
            self.scrapy_settings.overrides["FEED_URI"] = "results.json"
        elif fileformat is not None:
            self.scrapy_settings.overrides["FEED_URI"] = "results." + fileformat

        if fileformat is not None:
            self.scrapy_settings.overrides["FEED_FORMAT"] = fileformat


    def start_log(self, logfile, verbose):
        """
        This function starts the logging functionality of Scrapy using the settings given by the CLI.
        :param logfile: The location where the logfile will be saved.
        :param verbose: A boolean value to switch between loglevels.
        """
        if logfile is not None:
            if verbose:
                log.start(logfile=logfile, logstdout=False, loglevel=log.DEBUG)
            else:
                log.start(logfile=logfile, logstdout=True, loglevel=log.WARNING)
        else:
            if verbose:
                log.start(logstdout=False, loglevel=log.DEBUG)
            else:
                log.start(logstdout=True, loglevel=log.WARNING)

    @staticmethod
    def read_sourceconfiguration():
        """
        This function reads sources.cfg in the main folder for configuration
        variables for sources
        :return a ConfigParser object of sources.cfg
        """
        config = ConfigParser.ConfigParser()
        config.read('sources.cfg') # [TODO]: should be softcoded eventually
        return config

    @staticmethod
    def get_section(config, sourcename):
        """
        This function reads a config section labeled in variable sourcename and
        tests whether the reliability variable is set else set to empty string.
        Return the default section if the labeled config section does not exist
        :param config: a ConfigParser object
        :param sourcename: the name of the section to be read
        :return a dictionary of the section in the config labeled in sourcename
        """
        section = dict()
        if config.has_section(sourcename):
            section = dict(config.items(sourcename))
        elif config.defaults():
            section = config.defaults()
        if 'reliability' not in section:
            log.msg('Reliability not set for %s' % sourcename,
                    level=log.WARNING)
            section['reliability'] = ''
        return section