get_coordinates_by_zipcode.py

"""
For a given German Zipcode, returns the corresponding WKT Polygon or Multipolygon.
If pyperclip is installed, the WKT gets copied to the clipboard,
e.g. for RegionChooser or download_files_from_LGL_BW.py.

Also accepts multiple Zipcodes, or Zipcode prefix.

    python get_coordinates_by_zipcode.py --help
    usage: get_coordinates_by_zipcode.py [-h] PLZ [PLZ ...]

    Get WKT geometry for desired PLZs

    positional arguments:
    PLZ         desired PLZs

    options:
    -h, --help  show this help message and exit

> python get_coordinates_by_zipcode.py 70174
> python get_coordinates_by_zipcode.py 70567 70569
> python get_coordinates_by_zipcode.py 70
"""

# TODO: Write tests
# TODO: Use caching for shapes
# TODO: Rename functions

import argparse
import json
import re
from pathlib import Path
from shapely.geometry import shape
from shapely.ops import unary_union

INPUT_FOLDER = Path('plz')
PLZ_FILENAME = 'plz-5stellig.geojson'
PLZ_SHAPE_FILE = INPUT_FOLDER / PLZ_FILENAME


def _download_plz_shapes_if_needed():
    if not PLZ_SHAPE_FILE.exists():
        from tqdm import tqdm
        import requests
        URL = "https://downloads.suche-postleitzahl.org/v2/public/" + PLZ_FILENAME
        response = requests.get(URL, stream=True)
        INPUT_FOLDER.mkdir(exist_ok=True)
        with open(PLZ_SHAPE_FILE, "wb") as handle:
            for data in tqdm(response.iter_content(chunk_size=1024), unit='kB'):
                handle.write(data)


def _get_plz_shapes():
    print("Parsing %s..." % PLZ_FILENAME)
    _download_plz_shapes_if_needed()
    try:
        with open(PLZ_SHAPE_FILE) as f:
            print('  Done')
            return json.load(f)
    except json.decoder.JSONDecodeError:
        PLZ_SHAPE_FILE.unlink()
        raise AttributeError(f"{PLZ_FILENAME} seems to be damaged. Removing it. Please try again!")


def _get_coordinates(data, plz_patterns):
    geometries = []
    for plz_pattern in plz_patterns:
        found = False
        for plz_geojson in data['features']:
            if re.match(plz_pattern, plz_geojson['properties']['plz']):
                found = True

                properties = plz_geojson['properties']

                print('## %s' % properties['note'])
                print('Population : %d' % properties['einwohner'])
                print('Area : %.2f km²' % properties['qkm'])
                # NOTE : Geometry can be either a polygon,
                # a MultiPolygon : 98694 Ilmenau
                # or a polygon with holes : 31860 Emmerthal
                print('WKT Polygon : ')

                geometries.append(shape(plz_geojson['geometry']))

        if not found:
            raise AttributeError(f"Sorry, no information could be found for PLZ={plz_pattern}")

    merged = unary_union(geometries)
    wkt_polygon = merged.simplify(0.0001).wkt
    print(wkt_polygon)
    try:
        import pyperclip
        pyperclip.copy(wkt_polygon)
        print("WKT Polygon copied to clipboard.")
    except ModuleNotFoundError:
        pass

    print()
    print("Done!")
    return wkt_polygon


def get_coordinates_by_zipcode(plzs):
    plz_shapes = _get_plz_shapes()
    return _get_coordinates(plz_shapes, plzs)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Get WKT geometry for desired PLZs')
    parser.add_argument('plzs', metavar='PLZ', type=str, nargs='+',
                        help='desired PLZs')
    args = parser.parse_args()
    get_coordinates_by_zipcode(args.plzs)