gaur3009 commited on
Commit
0c5878f
Β·
verified Β·
1 Parent(s): 1b58f37

Update search.py

Browse files
Files changed (1) hide show
  1. search.py +16 -2
search.py CHANGED
@@ -1,5 +1,19 @@
1
  from googlesearch import search
 
 
 
 
 
 
 
2
 
3
  def search_google(query, num_results=5):
4
- """Search Google and return list of URLs."""
5
- return list(search(query, num_results=num_results))
 
 
 
 
 
 
 
 
1
  from googlesearch import search
2
+ import random
3
+
4
+ USER_AGENTS = [
5
+ "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
6
+ "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.3 Safari/605.1.15",
7
+ "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"
8
+ ]
9
 
10
  def search_google(query, num_results=5):
11
+ try:
12
+ return list(search(
13
+ query,
14
+ num_results=num_results,
15
+ advanced=True,
16
+ user_agent=random.choice(USER_AGENTS)
17
+ ))[:num_results]
18
+ except:
19
+ return []