owl-map/matcher/edit.py

import html

import requests
from flask import g

from . import database, mail, osm_oauth, user_agent_headers
from .model import Changeset

really_save = True
osm_api_base = "https://api.openstreetmap.org/api/0.6"


def new_changeset(comment: str) -> str:
    """XML for a new changeset."""
    return f"""
<osm>
  <changeset>
    <tag k="created_by" v="https://map.osm.wikidata.link/"/>
    <tag k="comment" v="{html.escape(comment)}"/>
  </changeset>
</osm>"""


def osm_request(path, **kwargs) -> requests.Response:
    return osm_oauth.api_put_request(path, **kwargs)


def create_changeset(changeset: str) -> requests.Response:
    """Create new changeset."""
    try:
        return osm_request("/changeset/create", data=changeset.encode("utf-8"))
    except requests.exceptions.HTTPError as r:
        print(changeset)
        print(r.response.text)
        raise


def close_changeset(changeset_id: int) -> requests.Response:
    """Close changeset."""
    return osm_request(f"/changeset/{changeset_id}/close")


def save_element(
    osm_type: str, osm_id: int, element_data: str
) -> requests.Response | None:
    """Upload new version of object to OSM map."""
    osm_path = f"/{osm_type}/{osm_id}"
    r = osm_request(osm_path, data=element_data)
    reply = r.text.strip()
    if reply.isdigit():
        return r

    subject = f"matcher error saving element: {osm_path}"
    username = g.user.username
    body = f"""
https://www.openstreetmap.org{osm_path}

user: {username}
message user: https://www.openstreetmap.org/message/new/{username}

error:
{reply}
"""

    mail.send_mail(subject, body)

    return None


def record_changeset(**kwargs: str) -> Changeset:
    """Record changeset in the database."""
    change: Changeset = Changeset(created=database.now_utc(), **kwargs)

    database.session.add(change)
    database.session.commit()

    return change


def get_existing(osm_type: str, osm_id: int) -> requests.Response:
    """Get existing OSM object."""
    url = f"{osm_api_base}/{osm_type}/{osm_id}"
    return requests.get(url, headers=user_agent_headers())