def mbtiles_to_disk(mbtiles_file, directory_path, **kwargs):
logger.info("Exporting database to disk: %s --> %s" % (mbtiles_file, directory_path))
delete_after_export = kwargs.get('delete_after_export', False)
no_overwrite = kwargs.get('no_overwrite', False)
zoom = kwargs.get('zoom', -1)
min_zoom = kwargs.get('min_zoom', 0)
max_zoom = kwargs.get('max_zoom', 255)
if zoom >= 0:
min_zoom = max_zoom = zoom
con = mbtiles_connect(mbtiles_file)
cur = con.cursor()
optimize_connection(cur)
if not os.path.isdir(directory_path):
os.mkdir(directory_path)
base_path = os.path.join(directory_path, "tiles")
if not os.path.isdir(base_path):
os.makedirs(base_path)
metadata = dict(con.execute('SELECT name, value FROM metadata').fetchall())
json.dump(metadata, open(os.path.join(directory_path, 'metadata.json'), 'w'), indent=4)
count = 0
start_time = time.time()
image_format = metadata.get('format', 'png')
total_tiles = con.execute("""SELECT count(zoom_level) FROM tiles WHERE zoom_level>=? AND zoom_level<=?""",
(min_zoom, max_zoom)).fetchone()[0]
sending_mbtiles_is_compacted = (con.execute("SELECT count(name) FROM sqlite_master WHERE type='table' AND name='images'").fetchone()[0] > 0)
tiles = cur.execute("""SELECT zoom_level, tile_column, tile_row, tile_data FROM tiles WHERE zoom_level>=? AND zoom_level<=?""",
(min_zoom, max_zoom))
t = tiles.fetchone()
while t:
z = t[0]
x = t[1]
y = t[2]
tile_data = t[3]
# Execute commands
if kwargs.get('command_list'):
tile_data = execute_commands_on_tile(kwargs['command_list'], image_format, tile_data)
if kwargs.get('flip_y', False) == True:
y = flip_y(z, y)
tile_dir = os.path.join(base_path, str(z), str(x))
if not os.path.isdir(tile_dir):
os.makedirs(tile_dir)
tile_file = os.path.join(tile_dir, '%s.%s' % (y, metadata.get('format', 'png')))
if no_overwrite == False or not os.path.isfile(tile_file):
f = open(tile_file, 'wb')
f.write(tile_data)
f.close()
count = count + 1
if (count % 100) == 0:
logger.debug("%s / %s tiles exported (%.1f%%, %.1f tiles/sec)" %
(count, total_tiles, (float(count) / float(total_tiles)) * 100.0, count / (time.time() - start_time)))
t = tiles.fetchone()
logger.info("%s / %s tiles exported (100.0%%, %.1f tiles/sec)" % (count, total_tiles, count / (time.time() - start_time)))
if delete_after_export:
logger.debug("WARNING: Removing exported tiles from %s" % (mbtiles_file))
if sending_mbtiles_is_compacted:
cur.execute("""DELETE FROM images WHERE tile_id IN (SELECT tile_id FROM map WHERE zoom_level>=? AND zoom_level<=?)""",
(min_zoom, max_zoom))
cur.execute("""DELETE FROM map WHERE zoom_level>=? AND zoom_level<=?""", (min_zoom, max_zoom))
else:
cur.execute("""DELETE FROM tiles WHERE zoom_level>=? AND zoom_level<=?""", (min_zoom, max_zoom))
optimize_database(cur, kwargs.get('skip_analyze', False), kwargs.get('skip_vacuum', False))
con.commit()
con.close()
