aboutsummaryrefslogtreecommitdiff
path: root/upload/close.py
diff options
context:
space:
mode:
Diffstat (limited to 'upload/close.py')
-rwxr-xr-xupload/close.py225
1 files changed, 225 insertions, 0 deletions
diff --git a/upload/close.py b/upload/close.py
new file mode 100755
index 0000000..c3ee3e3
--- /dev/null
+++ b/upload/close.py
@@ -0,0 +1,225 @@
+#! /usr/bin/python3
+# vim: fileencoding=utf-8 encoding=utf-8 et sw=4
+
+# Copyright (C) 2009 Jacek Konieczny <jajcus@jajcus.net>
+# Copyright (C) 2009 Andrzej Zaborowski <balrogg@gmail.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+
+"""
+Closes a changeset, given id.
+"""
+
+__version__ = "$Revision: 21 $"
+
+import os
+import subprocess
+import sys
+import traceback
+import base64
+
+import http.client as httplib
+import xml.etree.cElementTree as ElementTree
+import urllib.parse as urlparse
+
+class HTTPError(Exception):
+ pass
+
+class OSM_API(object):
+ url = 'https://master.apis.dev.openstreetmap.org/'
+ def __init__(self, username = None, password = None):
+ if username and password:
+ self.username = username
+ self.password = password
+ else:
+ self.username = ""
+ self.password = ""
+ self.changeset = None
+ self.progress_msg = None
+
+ def __del__(self):
+ if self.changeset is not None:
+ self.close_changeset()
+
+ def msg(self, mesg):
+ sys.stderr.write("\r%s… " % (self.progress_msg))
+ sys.stderr.write("\r%s… %s" % (self.progress_msg, mesg))
+ sys.stderr.flush()
+
+ def request(self, conn, method, url, body, headers, progress):
+ if progress:
+ self.msg("making request")
+ conn.putrequest(method, url)
+ self.msg("sending headers")
+ if body:
+ conn.putheader('Content-Length', str(len(body)))
+ for hdr, value in headers.iteritems():
+ conn.putheader(hdr, value)
+ self.msg("end of headers")
+ conn.endheaders()
+ self.msg(" 0%")
+ if body:
+ start = 0
+ size = len(body)
+ chunk = size / 100
+ if chunk < 16384:
+ chunk = 16384
+ while start < size:
+ end = min(size, start + chunk)
+ conn.send(body[start:end])
+ start = end
+ self.msg("%2i%%" % (start * 100 / size))
+ else:
+ self.msg(" ")
+ conn.request(method, url, body, headers)
+
+ def _run_request(self, method, url, body = None, progress = 0, content_type = "text/xml"):
+ url = urlparse.urljoin(self.url, url)
+ purl = urlparse.urlparse(url)
+ if purl.scheme != "https":
+ raise ValueError("Unsupported url scheme: %r" % (purl.scheme,))
+ if ":" in purl.netloc:
+ host, port = purl.netloc.split(":", 1)
+ port = int(port)
+ else:
+ host = purl.netloc
+ port = None
+ url = purl.path
+ if purl.query:
+ url += "?" + query
+ headers = {}
+ if body:
+ headers["Content-Type"] = content_type
+
+ try_no_auth = 0
+
+ if not try_no_auth and not self.username:
+ raise HTTPError("Need a username")
+
+ try:
+ self.msg("connecting")
+ conn = httplib.HTTPSConnection(host, port)
+# conn.set_debuglevel(10)
+
+ if try_no_auth:
+ self.request(conn, method, url, body, headers, progress)
+ self.msg("waiting for status")
+ response = conn.getresponse()
+
+ if not try_no_auth or (response.status == httplib.UNAUTHORIZED and
+ self.username):
+ if try_no_auth:
+ conn.close()
+ self.msg("re-connecting")
+ conn = httplib.HTTPSConnection(host, port)
+# conn.set_debuglevel(10)
+
+ creds = self.username + ":" + self.password
+ headers["Authorization"] = "Basic " + \
+ base64.b64encode(bytes(creds, "utf8")).decode("utf8")
+ # ^ Seems to be broken in python3 (even the raw
+ # documentation examples don't run for base64)
+ self.request(conn, method, url, body, headers, progress)
+ self.msg("waiting for status")
+ response = conn.getresponse()
+
+ if response.status == httplib.OK:
+ self.msg("reading response")
+ sys.stderr.flush()
+ response_body = response.read()
+ else:
+ raise HTTPError("%02i: %s (%s)" % (response.status,
+ response.reason, response.read()))
+ finally:
+ conn.close()
+ return response_body
+
+ def create_changeset(self, created_by, comment):
+ if self.changeset is not None:
+ raise RuntimeError("Changeset already opened")
+ self.progress_msg = "I'm creating the changeset"
+ self.msg("")
+ sys.stderr.flush()
+ root = ElementTree.Element("osm")
+ tree = ElementTree.ElementTree(root)
+ element = ElementTree.SubElement(root, "changeset")
+ ElementTree.SubElement(element, "tag", {"k": "created_by", "v": created_by})
+ ElementTree.SubElement(element, "tag", {"k": "comment", "v": comment})
+ body = ElementTree.tostring(root, "utf-8")
+ reply = self._run_request("PUT", "/api/0.6/changeset/create", body)
+ changeset = int(reply.strip())
+ self.msg("done. Id: %i" % (changeset,))
+ sys.stderr("\n")
+ self.changeset = changeset
+
+ def upload(self, change):
+ if self.changeset is None:
+ raise RuntimeError("Changeset not opened")
+ self.progress_msg = "Now I'm sending changes"
+ self.msg("")
+ sys.stderr.flush()
+ for operation in change:
+ if operation.tag not in ("create", "modify", "delete"):
+ continue
+ for element in operation:
+ element.attrib["changeset"] = str(self.changeset)
+ body = ElementTree.tostring(change, "utf-8")
+ reply = self._run_request("POST", "/api/0.6/changeset/%i/upload"
+ % (self.changeset,), body, 1)
+ self.msg("done.")
+ sys.stderr.write("\n")
+ return reply
+
+ def close_changeset(self):
+ if self.changeset is None:
+ raise RuntimeError("Changeset not opened")
+ self.progress_msg = "Closing"
+ self.msg("")
+ sys.stderr.flush()
+ sys.stderr.flush()
+ reply = self._run_request("PUT", "/api/0.6/changeset/%i/close"
+ % (self.changeset,))
+ self.changeset = None
+ self.msg("done, too.")
+ sys.stderr.write("\n")
+
+try:
+ this_dir = os.path.dirname(__file__)
+ try:
+ version = int(subprocess.Popen(["svnversion", this_dir], stdout = subprocess.PIPE).communicate()[0].strip())
+ except:
+ version = 1
+ if len(sys.argv) != 2:
+ sys.stderr.write("Synopsis:\n")
+ sys.stderr.write(" %s <changeset-id>\n", sys.argv[0])
+ sys.exit(1)
+
+ login = input("OSM login: ")
+ if not login:
+ sys.exit(1)
+ password = input("OSM password: ")
+ if not login:
+ sys.exit(1)
+
+ api = OSM_API(login, password)
+ api.changeset = int(sys.argv[1])
+ api.close_changeset()
+except HTTPError as err:
+ sys.stderr.write(str(err) + "\n")
+ sys.exit(1)
+except Exception as err:
+ sys.stderr.write(str(err) + "\n")
+ traceback.print_exc(file=sys.stderr)
+ sys.exit(1)
generated by cgit v1.2.3 (git 2.39.1) at 2025-07-03 23:36:15 +0000