from flask import current_app, request from itertools import islice import os.path import json import math import user_agents import re from datetime import date from num2words import num2words metres_per_mile = 1609.344 feet_per_metre = 3.28084 feet_per_mile = 5280 def chunk(it, size): it = iter(it) return iter(lambda: tuple(islice(it, size)), ()) def flatten(l): return [item for sublist in l for item in sublist] def drop_start(s, start): assert s.startswith(start) return s[len(start) :] def remove_start(s, start): return s[len(start) :] if s.startswith(start) else s def normalize_url(url): for start in "http://", "https://", "www.": url = remove_start(url, start) return url.rstrip("/") def contains_digit(s): return any(c.isdigit() for c in s) def cache_dir(): return current_app.config["CACHE_DIR"] def cache_filename(filename): return os.path.join(cache_dir(), filename) def load_from_cache(filename): return json.load(open(cache_filename(filename))) def get_radius(default=1000): arg_radius = request.args.get("radius") return int(arg_radius) if arg_radius and arg_radius.isdigit() else default def get_int_arg(name): if name in request.args and request.args[name].isdigit(): return int(request.args[name]) def calc_chunk_size(area_in_sq_km, size=22): side = math.sqrt(area_in_sq_km) return max(1, math.ceil(side / size)) def file_missing_or_empty(filename): return os.path.exists(filename) or os.stat(filename).st_size == 0 def is_bot(): """ Is the current request from a web robot? """ ua = request.headers.get("User-Agent") return ua and user_agents.parse(ua).is_bot def log_location(): return current_app.config["LOG_DIR"] def good_location(): return os.path.join(log_location(), "complete") def capfirst(value): """ Uppercase first letter of string, leave rest as is. """ return value[0].upper() + value[1:] if value else value def any_upper(value): return any(c.isupper() for c in value) def find_log_file(place): start = f"{place.place_id}_" for f in os.scandir(good_location()): if f.name.startswith(start): return f.path def get_free_space(config): s = os.statvfs(config["FREE_SPACE_PATH"]) return s.f_bsize * s.f_bavail def display_distance(units, dist): if units in ("miles_and_feet", "miles_and_yards"): total_feet = dist * feet_per_metre miles = total_feet / feet_per_mile if miles > 0.5: return f"{miles:,.2f} miles" else: return { "miles_and_feet": f"{total_feet:,.0f} feet", "miles_and_yards": f"{total_feet / 3:,.0f} yards", }[units] if units == "miles_and_metres": miles = dist / metres_per_mile return f"{miles:,.2f} miles" if miles > 0.5 else f"{dist:,.0f} metres" if units == "km_and_metres": units = "km" if dist > 500 else "metres" if units == "metres": return f"{dist:,.0f} m" if units == "km": return f"{dist / 1000:,.2f} km" re_range = re.compile(r"\b(\d+) ?(?:to|-) ?(\d+)\b", re.I) re_number_list = re.compile(r"\b([\d, ]+) (?:and|&) (\d+)\b", re.I) re_number = re.compile(r"^(?:No\.?|Number)? ?(\d+)\b") def is_in_range(address_range, address): m_number = re_number.match(address) if not m_number: return False m_range = re_range.search(address_range) if m_range: start, end = int(m_range.group(1)), int(m_range.group(2)) if re_range.search(address): return False return start <= int(m_number.group(1)) <= end m_list = re_number_list.search(address_range) if m_list: numbers = {n.strip() for n in m_list.group(1).split(",")} | {m_list.group(2)} if re_number_list.search(address): return False return m_number.group(1) in numbers return False def format_wikibase_time(v): p = v["precision"] t = v["time"] # TODO: handle dates with century precision (7) # example: https://www.wikidata.org/wiki/Q108266998 if p == 11: return date.fromisoformat(t[1:11]).strftime("%-d %B %Y") if p == 10: return date.fromisoformat(t[1:8] + "-01").strftime("%B %Y") if p == 9: return t[1:5] if p == 7: century = ((int(t[:5]) - 1) // 100) + 1 end = " BC" if century < 0 else "" return num2words(abs(century), to="ordinal_num") + " century" + end