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()