mirror of https://github.com/chubin/wttr.in
parent
aa3736bb4c
commit
918a026422
@ -0,0 +1,114 @@
|
||||
|
||||
import gevent
|
||||
from gevent.pywsgi import WSGIServer
|
||||
from gevent.queue import Queue
|
||||
from gevent.monkey import patch_all
|
||||
from gevent.subprocess import Popen, PIPE, STDOUT
|
||||
patch_all()
|
||||
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
|
||||
from flask import Flask, request, render_template, send_from_directory, send_file, make_response, jsonify, Response
|
||||
app = Flask(__name__)
|
||||
|
||||
MYDIR = os.path.abspath(os.path.dirname('__file__'))
|
||||
sys.path.append(os.path.join(MYDIR, 'lib'))
|
||||
|
||||
CACHEDIR = os.path.join(MYDIR, 'cache')
|
||||
|
||||
from geopy.geocoders import Nominatim #, Mapzen
|
||||
#geomapzen = Mapzen("mapzen-RBNbmcZ") # Nominatim()
|
||||
geoosm = Nominatim(timeout=7, user_agent="wttrin-geo/0.0.2")
|
||||
|
||||
import airports
|
||||
|
||||
# from tzwhere import tzwhere
|
||||
import timezonefinder
|
||||
tf = timezonefinder.TimezoneFinder()
|
||||
|
||||
def load_cache(location_string):
|
||||
try:
|
||||
location_string = location_string.replace('/', '_')
|
||||
cachefile = os.path.join(CACHEDIR, location_string)
|
||||
|
||||
return json.loads(open(cachefile, 'r').read())
|
||||
except:
|
||||
return None
|
||||
|
||||
def shorten_full_address(address):
|
||||
parts = address.split(',')
|
||||
if len(parts) > 6:
|
||||
parts = parts[:2] + [x for x in parts[-4:] if len(x) < 20]
|
||||
return ','.join(parts)
|
||||
return address
|
||||
|
||||
|
||||
def save_cache(location_string, answer):
|
||||
location_string = location_string.replace('/', '_')
|
||||
cachefile = os.path.join(CACHEDIR, location_string)
|
||||
open(cachefile, 'w').write(json.dumps(answer))
|
||||
|
||||
def query_osm(location_string):
|
||||
try:
|
||||
location = geoosm.geocode(location_string)
|
||||
return {
|
||||
'address': location.address,
|
||||
'latitude': location.latitude,
|
||||
'longitude':location.longitude,
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return None
|
||||
|
||||
def add_timezone_information(geo_data):
|
||||
# tzwhere_ = tzwhere.tzwhere()
|
||||
# timezone_str = tzwhere_.tzNameAt(geo_data["latitude"], geo_data["longitude"])
|
||||
timezone_str = tf.certain_timezone_at(lat=geo_data["latitude"], lng=geo_data["longitude"])
|
||||
|
||||
answer = geo_data.copy()
|
||||
answer["timezone"] = timezone_str
|
||||
|
||||
return answer
|
||||
|
||||
@app.route("/<string:location>")
|
||||
def find_location(location):
|
||||
|
||||
airport_gps_location = airports.get_airport_gps_location(location.upper().lstrip('~'))
|
||||
is_airport = False
|
||||
if airport_gps_location is not None:
|
||||
location = airport_gps_location
|
||||
is_airport = True
|
||||
|
||||
location = location.replace('+', ' ')
|
||||
answer = load_cache(location)
|
||||
loaded_answer = None
|
||||
|
||||
if answer is not None:
|
||||
loaded_answer = answer.copy()
|
||||
print("cache found: %s" % location)
|
||||
|
||||
if answer is None:
|
||||
answer = query_osm(location)
|
||||
|
||||
if is_airport:
|
||||
answer['address'] = shorten_full_address(answer['address'])
|
||||
|
||||
if "timezone" not in answer:
|
||||
answer = add_timezone_information(answer)
|
||||
|
||||
if answer is not None and loaded_answer != answer:
|
||||
save_cache(location, answer)
|
||||
|
||||
if answer is None:
|
||||
return ""
|
||||
else:
|
||||
r = Response(json.dumps(answer))
|
||||
r.headers["Content-Type"] = "text/json; charset=utf-8"
|
||||
return r
|
||||
|
||||
app.config['JSON_AS_ASCII'] = False
|
||||
server = WSGIServer(("127.0.0.1", 8004), app)
|
||||
server.serve_forever()
|
Loading…
Reference in new issue