| import logging |
| import os |
| import os.path |
| import urllib.request, urllib.parse, urllib.error |
| |
| from edge.writer.solrtemplateresponsewriter import SolrTemplateResponseWriter |
| from edge.response.solrjsontemplateresponse import SolrJsonTemplateResponse |
| |
| class Writer(SolrTemplateResponseWriter): |
| def __init__(self, configFilePath): |
| super(Writer, self).__init__(configFilePath) |
| |
| self.contentType = 'application/json' |
| |
| templatePath = os.path.dirname(configFilePath) + os.sep |
| templatePath += self._configuration.get('service', 'template') |
| self.template = self._readTemplate(templatePath) |
| |
| def _generateOpenSearchResponse(self, solrResponse, searchText, searchUrl, searchParams, pretty): |
| response = SolrJsonTemplateResponse() |
| response.setTemplate(self.template) |
| |
| return response.generate(solrResponse, pretty=pretty) |
| |
| def _constructSolrQuery(self, startIndex, entriesPerPage, parameters, facets): |
| queries = [] |
| filterQueries = [] |
| filterQueries.append('status:1') |
| sort = None |
| |
| for key, value in parameters.items(): |
| if value != "": |
| if key == 'keyword': |
| #Special case keyword search on glossary_items only match title |
| if 'table' in parameters and parameters['table'] == 'glossary_items': |
| queries.append('title_t:('+urllib.parse.quote(value) + ')') |
| else: |
| queries.append(urllib.parse.quote(value)) |
| elif key == 'year': |
| start = value + "-01-01T00:00:00.000Z" |
| end = value + "-12-31T23:59:59.999Z" |
| filterQueries.append('created_at:['+start+'%20TO%20'+end+']') |
| elif key == 'table': |
| filterQueries.append('type:' + value) |
| elif key == 'glossary_title': |
| range = value.lower().split('-') |
| filterQueries.append('{!frange%20l=' + range[0] + '%20u=' + range[1] + 'z}' + 'title_lc') |
| elif key == 'sort': |
| sort = urllib.parse.quote(value) |
| elif key == 'topic_id': |
| filterQueries.append('categories_id:' + value) |
| elif key == 'mission_id': |
| filterQueries.append('mission_ids_array:' + value) |
| else: |
| if type(value) is list: |
| if 'table' in parameters and parameters['table'] == 'news_items': |
| filterQueries.append(key + ':(' + '+OR+'.join([self._urlEncodeSolrQueryValue(v) for v in value]) + ')') |
| else: |
| for v in value: |
| filterQueries.append(key + ':' + self._urlEncodeSolrQueryValue(v)) |
| else: |
| filterQueries.append(key + ':' + self._urlEncodeSolrQueryValue(value)) |
| if len(queries) == 0: |
| queries.append('*:*') |
| |
| query = 'q='+'+AND+'.join(queries)+'&version=2.2&indent=on&wt=json&start='+str(startIndex)+'&rows='+str(entriesPerPage) |
| |
| if len(filterQueries) > 0: |
| query += '&fq='+'+AND+'.join(filterQueries) |
| |
| if sort is not None: |
| query += '&sort=' + sort |
| |
| logging.debug('solr query: '+query) |
| |
| return query |
