mirror of https://github.com/chubin/wttr.in
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.2 KiB
137 lines
3.2 KiB
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()
|