konova/konova/utils/schneider/fetcher.py
mpeltriaux c7aa90aa5b #280 Schneider capability
* refactors update_parcels() method in Geometry model to work on Schneider
* old WFS based logic still exists as update_parcels_wfs() in Geometry model to have a fallback. Can be deleted in the future
2022-12-14 12:18:18 +01:00

59 lines
1.6 KiB
Python

"""
Author: Michel Peltriaux
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
Contact: ksp-servicestelle@sgdnord.rlp.de
Created on: 14.12.22
"""
import json
from json import JSONDecodeError
import requests
from konova.sub_settings import schneider_settings
class ParcelFetcher:
base_url = schneider_settings.base_url
auth_header = schneider_settings.auth_header
auth_header_token = schneider_settings.auth_header_token
geometry = None
geojson = None
results = None
def __init__(self, geometry):
if geometry is None:
raise AssertionError("Geometry must not be none")
self.geometry = geometry
# Reduce size of geometry to avoid "intersections" because of exact border matching
geom = geometry.geom.buffer(-0.001)
self.geojson = geom.ewkt
self.results = []
def get_parcels(self, url: str = None):
post_url = url
if post_url is None:
post_url = f"{self.base_url}/parcel/intersect/"
response = requests.post(
url=post_url,
data=self.geojson,
headers={
self.auth_header: self.auth_header_token
}
)
try:
content = json.loads(response.content.decode("utf-8"))
except JSONDecodeError:
content = {}
next = content.get("next", None)
fetched_parcels = content.get("results", [])
self.results += fetched_parcels
if next:
self.get_parcels(next)
return self.results