def test_export_mbtiles(tmpdir):
with TPK("tests/data/states_filled.tpk") as tpk:
mbtiles_filename = str(tmpdir.join("test.mbtiles"))
tpk.to_mbtiles(mbtiles_filename)
tpk.close()
assert os.path.exists(mbtiles_filename)
with sqlite3.connect(mbtiles_filename) as db:
cursor = db.cursor()
# Verify tile data (note xyz scheme for select)
cursor.execute("select tile_data from tiles where zoom_level=2 "
"and tile_column=0 and tile_row=2")
data = cursor.fetchone()
assert len(data) == 1 # make sure we got one tile back
assert hashlib.sha1(data[0]).hexdigest() == TILE_2_0_2_SHA1
# Verify zoom levels present
cursor.execute(
"select distinct zoom_level from tiles order by zoom_level")
zoom_levels = {x[0] for x in cursor.fetchall()}
assert zoom_levels == {0, 1, 2, 3, 4}
# Verify bounds in metadata
cursor.execute('select value from metadata where name="bounds"')
bounds = cursor.fetchone()
assert bounds
assert bounds[0] == ",".join("{0:4f}".format(v) for v in tpk.bounds)
cursor.close()
def test_alt_root_name(tmpdir):
tpk = TPK('tests/data/alt_root_name.tpk')
mbtiles_filename = str(tmpdir.join('test'))
tpk.to_mbtiles(mbtiles_filename)
tpk.close()
assert os.path.exists('{0}.mbtiles'.format(mbtiles_filename))
def test_alt_root_name(tmpdir):
tpk = TPK("tests/data/alt_root_name.tpk")
mbtiles_filename = str(tmpdir.join("test"))
tpk.to_mbtiles(mbtiles_filename)
tpk.close()
assert os.path.exists("{0}.mbtiles".format(mbtiles_filename))
def test_export_mbtiles_add_suffix(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
mbtiles_filename = str(tmpdir.join("test"))
tpk.to_mbtiles(mbtiles_filename)
tpk.close()
assert os.path.exists("{0}.mbtiles".format(mbtiles_filename))
def test_export_disk_invalid_scheme(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
path = str(tmpdir.join("tiles"))
with pytest.raises(ValueError):
tpk.to_disk(path, scheme="bad")
tpk.close()
def test_export_mbtiles_int_zoom(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
mbtiles_filename = str(tmpdir.join("test.mbtiles"))
tpk.to_mbtiles(mbtiles_filename, zoom=1)
tpk.close()
assert os.path.exists(mbtiles_filename)
def test_export_mbtiles_mixed_format(tmpdir):
tpk = TPK('tests/data/states_filled.tpk')
tpk.format = 'mixed' # this is a hack to make test fail, need a test file for this
mbtiles_filename = str(tmpdir.join('test.mbtiles'))
with pytest.raises(ValueError):
tpk.to_mbtiles(mbtiles_filename)
tpk.close()
def test_export_disk_int_zoom(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
path = str(tmpdir.join("tiles"))
tpk.to_disk(path, zoom=1)
tpk.close()
assert os.path.exists(path)
assert os.path.exists(os.path.join(path, "1/0/0.png"))
assert not os.path.exists(os.path.join(path, "0/0/0.png"))
def test_export_disk_mixed_format(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
tpk.format = "mixed" # this is a hack to make test fail, need a test file for this
path = str(tmpdir.join("tiles"))
with pytest.raises(ValueError):
tpk.to_disk(path)
tpk.close()
def test_export_disk_existing_output(tmpdir):
tpk = TPK('tests/data/states_filled.tpk')
path = str(tmpdir.join('tiles'))
os.makedirs(path)
with open(os.path.join(path, 'test.txt'), 'w') as outfile:
outfile.write('Foo')
with pytest.raises(IOError):
tpk.to_disk(path)
tpk.close()
def test_read_tile():
with TPK("tests/data/states_filled.tpk") as tpk:
tile = next(tpk.read_tiles(zoom=[2]))
assert tile.z == 2
assert tile.x == 0
assert tile.y == 1
assert hashlib.sha1(tile.data).hexdigest() == TILE_2_0_2_SHA1
tile2 = next(tpk.read_tiles(zoom=2))
assert tile2 == tile
def test_export_disk_existing_output(tmpdir):
tpk = TPK("tests/data/states_filled.tpk")
path = str(tmpdir.join("tiles"))
os.makedirs(path)
with open(os.path.join(path, "test.txt"), "w") as outfile:
outfile.write("Foo")
with pytest.raises(IOError):
tpk.to_disk(path)
tpk.close()
def disk(tpk_filename, path, zoom, scheme, drop_empty, path_format, preview,
verbose):
"""Export the tile package to disk: z/x/y.<ext> or pattern specified using
--path-format option.
Will use the 'arcgis' tile scheme by default. If using with an XYZ tile
server or client, use the 'xyz' tile scheme.
Not recommended for higher zoom levels as this will produce large
directory trees."""
configure_logging(verbose)
if os.path.exists(path) and len(os.listdir(path)) > 0:
raise click.ClickException("Output directory must be empty.")
start = time.time()
if zoom is not None:
zoom = [int(v) for v in zoom.split(",")]
with TPK(tpk_filename) as tpk:
tpk.to_disk(path, zoom, scheme, drop_empty, path_format)
if preview:
template_filename = os.path.join(
pkg_resources.resource_filename(__name__,
"preview_template.html"))
with open(template_filename) as infile:
template = infile.read()
template = (template.replace(
"{{BOUNDS}}",
"[[{1}, {0}], [{3}, {2}]]".format(*tpk.bounds)).replace(
"{{MINZOOM}}", str(tpk.zoom_levels[0])).replace(
"{{MAXZOOM}}", str(tpk.zoom_levels[-1])).replace(
"{{URL}}",
os.path.join(
os.path.abspath(path),
path_format.replace(
"{ext}",
tpk.format.lower().replace("jpeg",
"jpg")[:3]),
),
))
outfilename = os.path.join(path, "preview.html")
with open(outfilename, "w") as outfile:
outfile.write(template)
webbrowser.open(outfilename)
print("Exported tiles in {0:2f} seconds".format(time.time() - start))
def test_tile_bounds(tmpdir):
mbtiles_filename = str(tmpdir.join("test.mbtiles"))
with TPK("tests/data/states_filled.tpk") as tpk:
tpk.to_mbtiles(mbtiles_filename, zoom=[0], tile_bounds=True)
with MBtiles(mbtiles_filename) as mbtiles:
assert mbtiles.zoom_range() == (0, 0)
# bounds calculated from tile 0 should be world bounds in web mercator coordinates
assert mbtiles.meta[
"bounds"] == "-180.000000,-85.051129,180.000000,85.051129"
def test_metadata():
with TPK('tests/data/states_filled.tpk') as tpk:
assert tpk.version == '1.0.0'
assert tpk.attribution == ''
assert tpk.name == 'states_filled'
assert tpk.summary == 'states'
assert tpk.tags == 'states'
assert tpk.description == ''
assert tpk.credits == 'US Census Bureau'
assert tpk.use_constraints == ''
assert tpk.bounds == [
-179.23108600000003, -14.601813000000014, 179.859681, 71.441059
]
def mbtiles(tpk_filename, mbtiles_filename, zoom, overwrite, verbose):
"""Export the tile package to mbtiles format"""
configure_logging(verbose)
if os.path.exists(mbtiles_filename) and not overwrite:
raise click.ClickException('Output exists and overwrite is false. '
'Use --overwrite option to overwrite')
start = time.time()
if zoom is not None:
zoom = [int(v) for v in zoom.split(',')]
tpk = TPK(tpk_filename)
tpk.to_mbtiles(mbtiles_filename, zoom)
tpk.close()
print('Exported tiles in {0:2f} seconds'.format(time.time() - start))
def test_metadata():
with TPK("tests/data/states_filled.tpk") as tpk:
assert tpk.version == "1.0.0"
assert tpk.attribution == ""
assert tpk.name == "states_filled"
assert tpk.summary == "states"
assert tpk.tags == "states"
assert tpk.description == ""
assert tpk.credits == "US Census Bureau"
assert tpk.use_constraints == ""
assert tpk.bounds == [
-179.23108600000003,
-14.601813000000014,
179.859681,
71.441059,
]
assert tpk.tile_size == 256
assert tpk.lods == [0, 1, 2, 3, 4]
assert tpk.zoom_levels == [0, 1, 2, 3, 4]
def mbtiles(tpk_filename, mbtiles_filename, zoom, overwrite, drop_empty,
tile_bounds, verbose):
"""Export the tile package to mbtiles format"""
configure_logging(verbose)
if os.path.exists(mbtiles_filename) and not overwrite:
raise click.ClickException("Output exists and overwrite is false. "
"Use --overwrite option to overwrite")
start = time.time()
if zoom is not None:
zoom = [int(v) for v in zoom.split(",")]
tpk = TPK(tpk_filename)
tpk.to_mbtiles(mbtiles_filename,
zoom=zoom,
tile_bounds=tile_bounds,
drop_empty=drop_empty)
tpk.close()
print("Exported tiles in {0:2f} seconds".format(time.time() - start))
def test_nonstandard_zoom_levels(tmpdir):
"""Not all TPK file are created with zoom levels that start at 0.
Make sure that we handle those where zoom level != level of detail ordinal value.
Also verify that it writes a proper mbtiles file.
"""
mbtiles_filename = str(tmpdir.join("test.mbtiles"))
with TPK("tests/data/nonstandard_zoom_levels.tpk") as tpk:
assert tpk.lods == [0, 1, 2]
assert tpk.zoom_levels == [2, 3, 4]
assert len(list(tpk.read_tiles())) == 22
tpk.to_mbtiles(mbtiles_filename)
with MBtiles(mbtiles_filename) as mbtiles:
assert mbtiles.zoom_range() == (2, 4)
assert mbtiles.row_range(2) == (2, 2)
assert mbtiles.col_range(2) == (0, 1)
assert mbtiles.row_range(3) == (4, 5)
assert mbtiles.col_range(3) == (1, 2)
assert mbtiles.row_range(4) == (8, 11)
assert mbtiles.col_range(4) == (2, 5)