Update search_engine.py

search_engine.py

@@ -4,13 +4,19 @@
 import os.path
 import time
 
-from googlesearch import search
+from googleapiclient.discovery import build
 import requests
 from bs4 import BeautifulSoup
 
 import constants
 
 
+def google_search_api(search_term, api_key, cse_id, **kwargs):
+    service = build("customsearch", "v1", developerKey=api_key)
+    res = service.cse().list(q=search_term, cx=cse_id, **kwargs).execute()
+    return res['items']
+
+    
 def search_google(subject, sites):
     try:
 
@@ -18,7 +24,7 @@ def search_google(subject, sites):
         for site in sites:
             print('    Buscando notícias no domínio: ' + site)
             query = f"{subject} site:{site}"
-            sites_searched = search(query, num_results=constants.num_sites)
+            sites_searched = google_search_api(query, os.environ['GOOGLE_KEY'], ['GOOGLE_SEARCH'], num=constants.num_sites)
             for s in sites_searched:
                 results.append(s)
                 #time.sleep(3)