estella 0.2.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bfebaabb0c030f132e17e046fcfea05c28d2fa9d
-  data.tar.gz: 9dbe818723ccfa698c3ad7e6f2775aaef20ec26e
+  metadata.gz: d881bd248152b83758455a9116ea0c56e382d279
+  data.tar.gz: 31d2a0d826ab272ad92627ddd68b10b6ab74d07b
 SHA512:
-  metadata.gz: 3ddf9ea6b35d7e9e3a7a9b2f35c908a3ff3d02b65410fd4a7fad89adbf376946c99930126c78e8a6884c67793cca77018d22b7a9e59742ad80600de576ef4ebd
-  data.tar.gz: 5ff181eaed10c332c34fa21aafc78e415269515c5d0eaf7dfec77eecfe15079d6c0463ae9627f5253443b783abad53170c07be3fe8e888aaac70f71f05b76c34
+  metadata.gz: 8c60f3f41f1b4baae0fb677034f8edd524e5f14d40761da4f5484f51e9497ff304c6092e2c6d121c129c06462f48e6511ba35314380c21b6708a27fcdc4e6a88
+  data.tar.gz: 1389d0798d01f143d7a9640dc4eba8d367b2fe11e14329d1aa03b90127a2f697b69067c6bfaab946198e4f3a9fa1068100b1d5e2fd257496575a20d2e7f8246c

data/.rubocop_todo.yml

@@ -1,21 +1,21 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2017-01-25 11:50:14 -0500 using RuboCop version 0.47.1.
+# on 2017-01-27 12:52:56 -0500 using RuboCop version 0.47.1.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
 # versions of RuboCop, may require this file to be generated again.
 
-# Offense count: 2
+# Offense count: 1
 Metrics/AbcSize:
-  Max: 24
+  Max: 18
 
 # Offense count: 3
 # Configuration parameters: CountComments, ExcludedMethods.
 Metrics/BlockLength:
-  Max: 117
+  Max: 134
 
-# Offense count: 28
+# Offense count: 30
 # Configuration parameters: AllowHeredoc, AllowURI, URISchemes, IgnoreCopDirectives, IgnoredPatterns.
 # URISchemes: http, https
 Metrics/LineLength:
@@ -24,7 +24,7 @@ Metrics/LineLength:
 # Offense count: 1
 # Configuration parameters: CountComments.
 Metrics/MethodLength:
-  Max: 14
+  Max: 13
 
 # Offense count: 3
 # Cop supports --auto-correct.
@@ -34,11 +34,6 @@ Performance/RedundantMerge:
     - 'lib/estella/parser.rb'
     - 'lib/estella/searchable.rb'
 
-# Offense count: 1
-Style/AccessorMethodName:
-  Exclude:
-    - 'lib/estella/helpers.rb'
-
 # Offense count: 1
 Style/ClassVars:
   Exclude:
@@ -55,11 +50,10 @@ Style/Documentation:
     - 'lib/estella/query.rb'
     - 'lib/estella/searchable.rb'
 
-# Offense count: 3
+# Offense count: 1
 # Configuration parameters: MinBodyLength.
 Style/GuardClause:
   Exclude:
-    - 'lib/estella/query.rb'
     - 'lib/estella/searchable.rb'
 
 # Offense count: 9

data/CHANGELOG.md

@@ -1,10 +1,11 @@
 ## Changelog
 
-### 0.2.2 (Next)
+### 0.3.0 (Next)
 
 * [#15](https://github.com/artsy/estella/pull/15): Removed undocumented `set_index_alias!` - [@dblock](https://github.com/dblock).
 * [#15](https://github.com/artsy/estella/pull/15): Added `delete_index!` and `create_index!` - [@dblock](https://github.com/dblock).
 * [#14](https://github.com/artsy/estella/pull/14): Fix: destroying documents fails to remove them from the index - [@dblock](https://github.com/dblock).
+* [#17](https://github.com/artsy/estella/pull/17): Added `exclude` to `estella_search` - [@dblock](https://github.com/dblock).
 * [#13](https://github.com/artsy/estella/pull/13): Added code coverage w/ Coveralls - [@dblock](https://github.com/dblock).
 * Your contribution here.
 

data/README.md

@@ -27,18 +27,17 @@ Just include the `Estella::Searchable` module and add a `searchable` block in yo
 
 ```ruby
 class Artist < ActiveRecord::Base
-    include Estella::Searchable
-
-    searchable do
-      field :name, type: :string, analysis: Estella::Analysis::FULLTEXT_ANALYSIS, factor: 1.0
-      field :keywords, type: :string, analysis: ['snowball', 'shingle'], factor: 0.5
-      field :bio, using: :biography, type: :string, index: :not_analyzed
-      field :birth_date, type: :date
-      field :follows, type: :integer
-      field :published, type: :boolean, filter: true
-      boost :follows, modifier: 'log1p', factor: 1E-3
-    end
-    ...
+  include Estella::Searchable
+
+  searchable do
+    field :name, type: :string, analysis: Estella::Analysis::FULLTEXT_ANALYSIS, factor: 1.0
+    field :keywords, type: :string, analysis: ['snowball', 'shingle'], factor: 0.5
+    field :bio, using: :biography, type: :string, index: :not_analyzed
+    field :birth_date, type: :date
+    field :follows, type: :integer
+    field :published, type: :boolean, filter: true
+    boost :follows, modifier: 'log1p', factor: 1E-3
+  end
 end
 ```
 
@@ -173,7 +172,13 @@ Artist.estella_search(term: 'frank', published: true)
 Artist.estella_search(term: 'frank', size: 10, from: 5)
 ```
 
-If you'd like to customize your query further, you can extend `Estella::Query` and override the `query_definition`:
+You can exclude records.
+
+```ruby
+Artist.estella_search(term: 'frank', exclude: { keywords: 'sinatra' })
+```
+
+If you'd like to customize your query further, you can extend `Estella::Query` and override `query_definition` and `field_factors`:
 
 ```ruby
 class MyQuery < Estella::Query
@@ -184,6 +189,30 @@ class MyQuery < Estella::Query
       }
     }
   end
+
+  def field_factors
+    {
+      default: 5,
+      ngram: 5,
+      snowball: 2,
+      shingle: 1,
+      search: 1
+    }
+  end
+end
+```
+
+Or manipulate the query in the initializer directly via `query` or using built-in helpers `must` and `exclude`.
+
+```ruby
+class MyQuery < Estella::Query
+  def initialize(params)
+    super
+    # same as query[:filter][:bool][:must] = { keywords: 'frank' }
+    must term: { keywords: 'frank' }
+    # same as query[:filter][:bool][:must_not] = { keywords: 'sinatra' }
+    exclude term: { keywords: 'sinatra' }
+  end
 end
 ```
 

data/lib/estella/query.rb

@@ -16,17 +16,10 @@ module Estella
       }
       add_query
       add_filters
+      add_excludes
       add_pagination
-      add_aggregations if params[:aggregations]
-      add_sort
     end
 
-    # override if needed
-    def add_aggregations; end
-
-    # override if needed
-    def add_sort; end
-
     def must(filter)
       query[:filter][:bool][:must] << filter
     end
@@ -35,6 +28,22 @@ module Estella
       query[:filter][:bool][:must_not] << filter
     end
 
+    def query_definition
+      {
+        multi_match: {
+          type: 'most_fields',
+          fields: term_search_fields,
+          query: params[:term]
+        }
+      }
+    end
+
+    def field_factors
+      Estella::Analysis::DEFAULT_FIELD_FACTORS
+    end
+
+    private
+
     def add_pagination
       query[:size] = params[:size] if params[:size]
       query[:from] = params[:from] if params[:from]
@@ -59,32 +68,19 @@ module Estella
       add_field_boost
     end
 
-    def query_definition
-      {
-        multi_match: {
-          type: 'most_fields',
-          fields: term_search_fields,
-          query: params[:term]
-        }
-      }
-    end
-
     def add_field_boost
-      if params[:boost]
-        query[:query][:function_score][:field_value_factor] = {
-          field: params[:boost][:field],
-          modifier: params[:boost][:modifier],
-          factor: params[:boost][:factor]
-        }
+      boost = params[:boost]
+      return unless boost
 
-        if params[:boost][:max]
-          query[:query][:function_score][:max_boost] = params[:boost][:max]
-        end
-      end
-    end
+      query[:query][:function_score][:field_value_factor] = {
+        field: boost[:field],
+        modifier: boost[:modifier],
+        factor: boost[:factor]
+      }
 
-    def field_factors
-      Estella::Analysis::DEFAULT_FIELD_FACTORS
+      max = boost[:max]
+      return unless max
+      query[:query][:function_score][:max_boost] = max
     end
 
     # search all analysed string fields by default
@@ -103,10 +99,19 @@ module Estella
     end
 
     def add_filters
-      if params[:indexed_fields]
-        params[:indexed_fields].each do |field, opts|
-          must term: { field => params[field] } if opts[:filter] && params[field]
-        end
+      indexed_fields = params[:indexed_fields]
+      return unless indexed_fields
+      indexed_fields.each do |field, opts|
+        next unless opts[:filter] && params[field]
+        must term: { field => params[field] }
+      end
+    end
+
+    def add_excludes
+      exclude = params[:exclude]
+      return unless exclude
+      exclude.each do |k, v|
+        exclude(term: { k => v })
       end
     end
   end

data/lib/estella/version.rb

@@ -1,3 +1,3 @@
 module Estella
-  VERSION = '0.2.2'
+  VERSION = '0.3.0'
 end

data/spec/searchable_spec.rb

@@ -55,6 +55,9 @@ describe Estella::Searchable, type: :model do
     it 'searches all text fields by default' do
       expect(SearchableModel.estella_search(term: 'jez')).to eq([@jez])
     end
+    it 'can exclude an instance' do
+      expect(SearchableModel.estella_search(term: 'jez tez fab', exclude: { keywords: 'jez' })).to eq([@fab, @tez])
+    end
     it 'boosts on follows_count' do
       popular_jeremy = SearchableModel.create(title: 'jeremy corban', follows_count: 20_000)
       SearchableModel.refresh_index!
@@ -105,6 +108,22 @@ describe Estella::Searchable, type: :model do
         expect(SearchableModel.estella_search(term: 'david')).to eq([@fab])
       end
     end
+    context 'with query customization' do
+      before do
+        class CustomQuery < Estella::Query
+          def initialize(params)
+            super
+            exclude term: { keywords: 'jez' }
+          end
+        end
+
+        allow(SearchableModel).to receive(:estella_search_query).and_return(CustomQuery)
+      end
+
+      it 'uses the custom query' do
+        expect(SearchableModel.estella_search(term: 'jez tez fab')).to eq([@fab, @tez])
+      end
+    end
   end
 
   describe 'configuration errors' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: estella
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.0
 platform: ruby
 authors:
 - Anil Bawa-Cavia