mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 10:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

reformat with black

Browse Source
This commit is contained in:
fscottfoti
2022-04-09 13:35:21 -07:00
parent ed40933563
commit ba6a620709
3 changed files with 51 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
python/bin/pmtiles-serve
View File

@@ -13,51 +13,58 @@ from pmtiles.reader import read
class ThreadingSimpleServer(ThreadingMixIn, http.server.HTTPServer):
pass
parser = argparse.ArgumentParser(description='Convert between PMTiles and other archive formats.')
parser.add_argument('pmtiles_file',help='PMTiles archive to serve')
parser.add_argument('port',help='Port to bind to')
parser.add_argument('--bind', help='Address to bind server to: default localhost')
parser.add_argument('--cors-allow-all', help='Return Access-Control-Allow-Origin:* header',action='store_true')
parser = argparse.ArgumentParser(
description="Convert between PMTiles and other archive formats."
)
parser.add_argument("pmtiles_file", help="PMTiles archive to serve")
parser.add_argument("port", help="Port to bind to")
parser.add_argument("--bind", help="Address to bind server to: default localhost")
parser.add_argument(
"--cors-allow-all",
help="Return Access-Control-Allow-Origin:* header",
action="store_true",
)
args = parser.parse_args()
with read(args.pmtiles_file) as reader:
fmt = reader.metadata['format']
fmt = reader.metadata["format"]
class Handler(http.server.SimpleHTTPRequestHandler):
def do_GET(self):
if self.path == "/metadata":
self.send_response(200)
if args.cors_allow_all:
self.send_header('Access-Control-Allow-Origin','*')
self.send_header("Access-Control-Allow-Origin", "*")
self.end_headers()
self.wfile.write(json.dumps(reader.metadata).encode('utf-8'))
self.wfile.write(json.dumps(reader.metadata).encode("utf-8"))
return
match = re.match("/(\d+)/(\d+)/(\d+)." + fmt, self.path)
if not match:
self.send_response(400)
self.end_headers()
self.wfile.write("bad request".encode('utf-8'))
self.wfile.write("bad request".encode("utf-8"))
return
z = int(match.group(1))
x = int(match.group(2))
y = int(match.group(3))
data = reader.get(z,x,y)
data = reader.get(z, x, y)
if not data:
self.send_response(404)
self.end_headers()
self.wfile.write("tile not found".encode('utf-8'))
self.wfile.write("tile not found".encode("utf-8"))
return
self.send_response(200)
if args.cors_allow_all:
self.send_header('Access-Control-Allow-Origin','*')
if fmt == 'pbf':
self.send_header('Content-Type','application/x-protobuf')
self.send_header("Access-Control-Allow-Origin", "*")
if fmt == "pbf":
self.send_header("Content-Type", "application/x-protobuf")
else:
self.send_header('Content-Type','image/' + fmt)
self.send_header("Content-Type", "image/" + fmt)
self.end_headers()
self.wfile.write(data)
bind = args.bind or 'localhost'
print(f'serving {bind}:{args.port}/{{z}}/{{x}}/{{y}}.{fmt}, for development only')
httpd = ThreadingSimpleServer((args.bind or '', int(args.port)), Handler)
bind = args.bind or "localhost"
print(f"serving {bind}:{args.port}/{{z}}/{{x}}/{{y}}.{fmt}, for development only")
httpd = ThreadingSimpleServer((args.bind or "", int(args.port)), Handler)
httpd.serve_forever()