import urllib2

from flask import json
import pysolr

from DLIObject import DLIObject


class QueryResolver(object):
    def __init__(self, solr_url):
        self.solr_url = solr_url
        self.solr_server = pysolr.Solr(self.solr_url)
        self.facet_renaming = {'entitytype': "Typology", 'provenance': "Content Provider",
                               "Content Provider": 'provenance', "Typology": 'entitytype'}

    def get_facet(self, facet, query=None):
        data = []

        f = 'entitytype'
        if query is not None:
            data.append({'facet': self.facet_renaming[f], 'query_name': f, 'values': [
                {'key': facet[f][i], 'value': facet[f][i + 1]} for i in range(0, len(facet[f]), 2)]})
        else:
            data.append({'facet': self.facet_renaming[f], 'query_name': f,
                         'values': [{'key': facet[f][i], 'value': facet[f][i + 1]} for i in
                                    range(0, len(facet[f]), 2)]})
        f = 'provenance'
        data.append({'facet': self.facet_renaming[f], 'query_name': f,
                     'values': [{'key': facet[f][i], 'value': facet[f][i + 1]} for i in range(0, len(facet[f]), 2)]})

        return data

    def get_item(self, persistent_identifier):
        query = 'objidentifier:"%s"' % persistent_identifier
        response = self.solr_server.search(q=query)
        query_response = []
        for hit in response:
            result = hit['__result'][0]
            query_response.append(DLIObject(result.encode('utf8')))

        return {'objects': query_response}

    def get_stats_detail(self):
        data = urllib2.urlopen("http://node0-p-dli.d4science.org:8080/dli/mvc/ui/stats/getLastStats")
        result = []
        d = json.loads(data.read())
        for k in d:
            if 'Deduction' not in k['datasource']:
                result.append(k)
        return result

    def get_stats(self):

        data = urllib2.urlopen("http://node0-p-dli.d4science.org:8080/dli/mvc/ui/stats/getAllStats")
        result = {}
        d = json.loads(data.read())
        for item in d:
            result[item] = []
            for k in d[item]:
                if 'Deduction' not in k['datasource']:
                    result[item].append(k)
        return result

    def query(self, q, browse_fields, start=None, page_size=10, only_result=False, filter_query=None):
        params = {}
        if browse_fields is not None:
            params['facet'] = 'true',
            params['facet.field'] = browse_fields
            params['facet.mincount'] = 1

        if filter_query is not None:
            params['fq'] = filter_query
        if start is not None:
            params['start'] = start
        response = self.solr_server.search(q, rows=page_size, **params)
        query_response = []
        for hit in response:
            result = hit['__result'][0]
            query_response.append(DLIObject(result.encode('utf8')))
        if not only_result:
            facet = self.get_facet(response.facets['facet_fields'], q)
        else:
            facet = self.get_facet(response.facets['facet_fields'])
        return {'objects': query_response, 'facet': facet, 'total': response.hits}