def _input_fiona_info(input_):
with fiona.open(input_) as src:
return dict(output_params=dict(schema=src.schema,
format=src.driver if src.driver
in available_input_formats() else None),
pyramid=None,
crs=src.crs,
zoom_levels=None,
input_type="vector",
bounds=src.bounds)
def _input_rasterio_info(input_):
with rasterio.open(input_) as src:
return dict(output_params=dict(bands=src.meta["count"],
dtype=src.meta["dtype"],
format=src.driver if src.driver
in available_input_formats() else None),
pyramid=None,
crs=src.crs,
zoom_levels=None,
pixel_size=src.transform[0],
input_type="raster",
bounds=src.bounds)
def list_formats(args):
"""List input and/or output formats."""
if args.input_formats == args.output_formats:
show_inputs, show_outputs = True, True
else:
show_inputs, show_outputs = args.input_formats, args.output_formats
if show_inputs:
print "input formats:"
for driver in available_input_formats():
print "-", driver
if show_outputs:
print "output formats:"
for driver in available_output_formats():
print "-", driver
def formats(input_formats, output_formats):
"""List input and/or output formats."""
if input_formats == output_formats:
show_inputs, show_outputs = True, True
else:
show_inputs, show_outputs = input_formats, output_formats
if show_inputs:
click.echo("input formats:")
for driver in available_input_formats():
click.echo("- %s" % driver)
if show_outputs:
click.echo("output formats:")
for driver in available_output_formats():
click.echo("- %s" % driver)
def test_available_input_formats():
"""Check if default input formats can be listed."""
assert set(['Mapchete', 'raster_file',
'vector_file']).issubset(set(available_input_formats()))
def test_format_available():
"""Format can be listed."""
assert "SAFE" in available_input_formats()
def test_driver_available():
"""Driver is correctly registered."""
assert "TileDirectory" in available_input_formats()
