Python solr.QueryBuilderSearch类代码示例

def facets():
    """
    returns facet sets for a search query
    """
    form = QueryForm.init_with_defaults(request.values)
    if form.validate():
        query_components = QueryBuilderSearch.build(form, request.values, facets_components=True)
        resp = solr.query(**query_components)
        return render_template('facets_sublevel.html', resp=resp, facet_field_interf_id=query_components['facet_field_interf_id'] )

    def test_only_query(self):
        """basic case"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano")'
            expected['ui_q'] = u' author:"civano"'

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_13(self):
        """with article only property"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=astronomy&article=y'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND NOT property:NONARTICLE AND database:"astronomy"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'database:"astronomy"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_02(self):
        """including database filter (uppercase)"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=ASTRONOMY'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND database:"ASTRONOMY"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'database:"ASTRONOMY"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_11(self):
        """with journal abbr fitler (wrong syntax)"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"+bibstem%3A"ApJ%3B+AJ"&db_f=astronomy'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano" bibstem:"ApJ; AJ") AND database:"astronomy"'
            expected['ui_q'] = u' author:"civano" bibstem:"ApJ; AJ"'
            expected['ui_filters'] = [u'database:"astronomy"',]

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_12(self):
        """with refereed property"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=astronomy&prop_f=refereed'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND property:"refereed" AND database:"astronomy"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'property:"refereed"', u'database:"astronomy"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_08(self):
        """with date range"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=astronomy&month_from=09&year_from=2009&year_to=2010'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND pubdate:[2009-09-00 TO 2010-12-00] AND database:"astronomy"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'pubdate:[2009-09-00 TO 2010-12-00]', u'database:"astronomy"',]

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_04(self):
        """2nd order query"""
        with self.app.test_request_context('/search/?q=hot(galaxy+clusters)&db_f=astronomy'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'(hot(galaxy clusters)) AND database:"astronomy"'
            expected['ui_q'] = u'hot(galaxy clusters)'
            expected['ui_filters'] = [u'database:"astronomy"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_03(self):
        """non-default database filter"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=physics'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND database:"physics"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'database:"physics"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_19(self):
        """with facet selection"""
        with self.app.test_request_context('/search/?q=*&db_f=astronomy&year_f=[2000 TO 2010]'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'(*) AND year:[2000 TO 2010] AND database:"astronomy"'
            expected['ui_q'] = u'*'
            expected['ui_filters'] = [u'year:[2000 TO 2010]', 
                                   u'database:"astronomy"',]

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_16(self):
        """with facet selection"""
        with self.app.test_request_context('/search/?q=author%3A"Civano"&db_f=astronomy&aut_f=-("1%2FCivano%2C+F%2FCivano%2C Francesca M."+OR+"1%2FElvis%2C M%2FElvis%2C Martin")'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'(author:"Civano") AND NOT author_facet_hier:("1/Civano, F/Civano, Francesca M." OR "1/Elvis, M/Elvis, Martin") AND database:"astronomy"'
            expected['ui_q'] = u'author:"Civano"'
            expected['ui_filters'] = [u'-author_facet_hier:("1/Civano, F/Civano, Francesca M." OR "1/Elvis, M/Elvis, Martin")',
                                    u'database:"astronomy"',]

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)

    def test_query_with_default_params_26(self):
        """test no fulltext"""
        with self.app.test_request_context('/search/?q=civano&no_ft=1'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'(civano)'
            expected['ui_q'] = u'civano'
            expected['ui_filters'] = []
            expected['query_fields'] = config.SOLR_SEARCH_DEFAULT_QUERY_FIELDS_METADATA_ONLY

            form = QueryForm.init_with_defaults(request.values)
            actual = QueryBuilderSearch.build(form, request.values)
            self.assertEqual(actual, expected)

    def test_query_with_default_params_25(self):
        """test topn operator"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=astronomy&re_sort_type=POPULARITY&re_sort_dir=asc&topn=1000'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'topn(1000, (( author:"civano") AND database:"astronomy"), "read_count asc")'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'database:"astronomy"']
            expected['sort'] = None

            form = QueryForm.init_with_defaults(request.values)
            actual = QueryBuilderSearch.build(form, request.values)
            self.assertEqual(actual, expected)

    def test_query_with_default_params_22(self):
        """with different row count"""
        with self.app.test_request_context('/search/?q=+author%3A"civano"&db_f=astronomy&nr=33'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'( author:"civano") AND database:"astronomy"'
            expected['ui_q'] = u' author:"civano"'
            expected['ui_filters'] = [u'database:"astronomy"']
            expected['rows'] = "33"

            form = QueryForm.init_with_defaults(request.values)
            actual = QueryBuilderSearch.build(form, request.values)
            self.assertEqual(actual, expected)

    def test_query_with_default_params_14(self):
        """with facet selection"""
        with self.app.test_request_context('/search/?q=author%3A"Civano"&aut_f=0%2FComastri%2C+A&db_f=astronomy&grant_f=0%2FNASA-HQ'):
            expected = deepcopy(QueryBuilderSearch.DEFAULT_COMPONENTS)
            expected['q'] = u'(author:"Civano") AND grant_facet_hier:"0/NASA-HQ" AND author_facet_hier:"0/Comastri, A" AND database:"astronomy"'
            expected['ui_q'] = u'author:"Civano"'
            expected['ui_filters'] = [u'grant_facet_hier:"0/NASA-HQ"',
                                    u'author_facet_hier:"0/Comastri, A"',
                                    u'database:"astronomy"']

            form = QueryForm.init_with_defaults(request.values)
            self.assertEqual(QueryBuilderSearch.build(form, request.values), expected)