From a922b42cbdff1e9836b343b15188efe400efa4ea Mon Sep 17 00:00:00 2001
From: Ben Busby <ben.busby7@gmail.com>
Date: Wed, 22 Jan 2020 05:51:02 +0000
Subject: [PATCH] Added desktop/mobile agent switching, updated gitignore

---
 .gitignore    |  2 ++
 app/routes.py | 11 ++++++++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index d0ee3b1..1c1abdd 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,5 @@
 venv/
 __pycache__/
 *.pyc
+*.pem
+run.sh
diff --git a/app/routes.py b/app/routes.py
index 6ac5175..507a578 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -5,6 +5,9 @@ import pycurl
 from .url import url_parse
 from io import BytesIO
 
+MOBILE_UA = 'Mozilla/5.0 (Linux; Android 8.0.0; SM-G960F Build/R16NW) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/62.0.3202.84 Mobile Safari/537.36'
+DESKTOP_UA = 'Brozilla/5.0 (Windows NT 6.1; Win64; x64; rv:59.0) Gecko/20100101 Mobile LizzieMcGuirefox/59.0'
+
 
 @app.route('/', methods=['GET'])
 def index():
@@ -21,10 +24,16 @@ def search():
     if 'tbm' in request.args:
         tbm = '&tbm=' + request.args.get('tbm')
 
+    user_agent = request.headers.get('User-Agent')
+
+    google_ua = DESKTOP_UA
+    if 'Android' in user_agent or 'iPhone' in user_agent:
+        google_ua = MOBILE_UA
+
     b_obj = BytesIO()
     crl = pycurl.Curl()
     crl.setopt(crl.URL, 'https://www.google.com/search?q=' + url_parse(q) + tbm)
-    crl.setopt(crl.USERAGENT, 'Brozilla/5.0 (Windows NT 6.1; Win64; x64; rv:59.0) Gecko/20100101 LizzieMcGuirefox/59.0')
+    crl.setopt(crl.USERAGENT, google_ua)
     crl.setopt(crl.WRITEDATA, b_obj)
     crl.perform()
     crl.close()