lobo/wttr
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Remove PLAIN_TEXT_AGENTS from globals scope and add python-requests

Browse source
This commit is contained in:
Santiago Basulto 2017-02-21 20:16:07 -03:00 • committed by GitHub
parent 124034eb76
commit 32c1686878
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bin/srv.py
View file

@ -40,6 +40,13 @@ We were unable to find your location,
so we have brought you to Oymyakon, so we have brought you to Oymyakon,
one of the coldest permanently inhabited locales on the planet. one of the coldest permanently inhabited locales on the planet.
""" """
PLAIN_TEXT_AGENTS = [
"curl",
"httpie",
"lwp-request",
"wget",
"python-requests"
]
if not os.path.exists(os.path.dirname( LOG_FILE )): if not os.path.exists(os.path.dirname( LOG_FILE )):
os.makedirs( os.path.dirname( LOG_FILE ) ) os.makedirs( os.path.dirname( LOG_FILE ) )
@ -218,16 +225,10 @@ def send_favicon():
@app.route("/") @app.route("/")
@app.route("/<string:location>") @app.route("/<string:location>")
def wttr(location = None): def wttr(location = None):
plain_text_agents = [
"curl",
"httpie",
"lwp-request",
"wget",
]
user_agent = request.headers.get('User-Agent').lower() user_agent = request.headers.get('User-Agent').lower()
if any(agent in user_agent for agent in plain_text_agents): if any(agent in user_agent for agent in PLAIN_TEXT_AGENTS):
html_output = False html_output = False
else: else:
html_output = True html_output = True