Skip models that aren't in the configurations.

rlaverde · Mar 10, 2017 · eb5cca4 · eb5cca4
1 parent 025dd7e
commit eb5cca4
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/lektor_tipue_search.py b/lektor_tipue_search.py
@@ -50,14 +50,13 @@ def on_before_build_all(self, builder, **extra):
  def on_before_build(self, source, prog, **extra):
  if isinstance(source, Page):
 
- try:
+ if source.datamodel.id in self.models:
  model = self.models[source.datamodel.id]
- except Exception:
- return
- item = {key: source[field] for key,field in model.items()}
- item['url'] = source.url_path
 
- self.tipue_search[source.alt].append(item)
+ item = {key: source[field] for key, field in model.items()}
+ item['url'] = source.url_path
+
+ self.tipue_search[source.alt].append(item)
 
  @check_enabled
  def on_after_build_all(self, builder, **extra):