From 500942cb9949d6122862b0bbdb3040feccd65630 Mon Sep 17 00:00:00 2001
From: DUO Labs <dvdugo333@gmail.com>
Date: Wed, 2 Feb 2022 14:57:05 -0500
Subject: [PATCH] Update minimal mode for new Google formatting (#637)

Google's latest formatting changes broke the modifications made when enabling
`WHOOGLE_MINIMAL`. This updates the result filtering to work with the new
changes.

Fixes #634
---
 app/filter.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/filter.py b/app/filter.py
index df8f243..a5fe928 100644
--- a/app/filter.py
+++ b/app/filter.py
@@ -12,6 +12,8 @@ import urllib.parse as urlparse
 from urllib.parse import parse_qs
 import os
 
+minimal_mode_sections = ['Top stories', 'Images', 'People also ask']
+
 
 def extract_q(q_str: str, href: str) -> str:
     """Extracts the 'q' element from a result link. This is typically
@@ -169,7 +171,11 @@ class Filter:
         for result in self.main_divs:
             result_children = pull_child_divs(result)
             if minimal_mode:
-                if len(result_children) in (1, 3):
+                if any(f">{x}</span" in str(s) for s in result_children
+                   for x in minimal_mode_sections):
+                    result.decompose()
+                    continue
+                if len(result_children) < self.RESULT_CHILD_LIMIT:
                     continue
             else:
                 if len(result_children) < self.RESULT_CHILD_LIMIT: