def export_all_habitats(
info_role,
export_format="csv",
):
"""
Download all stations
The route is in post to avoid a too large query string
.. :quickref: Occhab;
"""
data = request.get_json()
export_view = GenericTableGeo(
tableName="v_export_sinp",
schemaName="pr_occhab",
engine=DB.engine,
geometry_field="geom_local",
srid=current_app.config["LOCAL_SRID"],
)
file_name = datetime.datetime.now().strftime("%Y_%m_%d_%Hh%Mm%S")
file_name = filemanager.removeDisallowedFilenameChars(file_name)
db_cols_for_shape = []
columns_to_serialize = []
for db_col in export_view.db_cols:
if db_col.key in blueprint.config["EXPORT_COLUMS"]:
if db_col.key != "geometry":
db_cols_for_shape.append(db_col)
columns_to_serialize.append(db_col.key)
results = (
DB.session.query(export_view.tableDef)
.filter(export_view.tableDef.columns.id_station.in_(data["idsStation"]))
.limit(blueprint.config["NB_MAX_EXPORT"])
)
if export_format == "csv":
formated_data = [export_view.as_dict(d, fields=[]) for d in results]
return to_csv_resp(
file_name, formated_data, separator=";", columns=columns_to_serialize
)
elif export_format == "geojson":
features = []
for r in results:
features.append(
Feature(
geometry=json.loads(r.geojson),
properties=export_view.as_dict(r, fields=columns_to_serialize),
)
)
return to_json_resp(
FeatureCollection(features), as_file=True, filename=file_name, indent=4
)
else:
try:
dir_name, file_name = export_as_geo_file(
export_format=export_format,
export_view=export_view,
db_cols=db_cols_for_shape,
geojson_col=None,
data=results,
file_name=file_name,
)
return send_from_directory(dir_name, file_name, as_attachment=True)
except GeonatureApiError as e:
message = str(e)
return render_template(
"error.html",
error=message,
redirect=current_app.config["URL_APPLICATION"]
+ "/#/"
+ blueprint.config["MODULE_URL"],
)
def export(info_role):
"""Export data from pr_occtax.v_export_occtax view (parameter)
.. :quickref: Occtax; Export data from pr_occtax.v_export_occtax
:query str format: format of the export ('csv', 'geojson', 'shapefile', 'gpkg')
"""
export_view_name = blueprint.config["export_view_name"]
export_geom_column = blueprint.config["export_geom_columns_name"]
export_columns = blueprint.config["export_columns"]
export_srid = blueprint.config["export_srid"]
export_format = request.args[
"format"] if "format" in request.args else "geojson"
export_col_name_additional_data = blueprint.config[
"export_col_name_additional_data"]
export_view = GenericTableGeo(
tableName=export_view_name,
schemaName="pr_occtax",
engine=DB.engine,
geometry_field=export_geom_column,
srid=export_srid,
)
columns = (export_columns if len(export_columns) > 0 else
[db_col.key for db_col in export_view.db_cols])
releve_repository = ReleveRepository(export_view)
q = releve_repository.get_filtered_query(info_role,
from_generic_table=True)
q = get_query_occtax_filters(
request.args,
export_view,
q,
from_generic_table=True,
obs_txt_column=blueprint.config["export_observer_txt_column"],
)
if current_app.config["OCCTAX"]["ADD_MEDIA_IN_EXPORT"]:
q, columns = releve_repository.add_media_in_export(q, columns)
data = q.all()
file_name = datetime.datetime.now().strftime("%Y_%m_%d_%Hh%Mm%S")
file_name = filemanager.removeDisallowedFilenameChars(file_name)
#Ajout des colonnes additionnels
additional_col_names = []
query_add_fields = DB.session.query(TAdditionalFields).filter(
TAdditionalFields.modules.any(module_code="OCCTAX")).filter(
TAdditionalFields.exportable == True)
global_add_fields = query_add_fields.filter(
~TAdditionalFields.datasets.any()).all()
if "id_dataset" in request.args:
dataset_add_fields = query_add_fields.filter(
TAdditionalFields.datasets.any(
id_dataset=request.args['id_dataset'])).all()
global_add_fields = [*global_add_fields, *dataset_add_fields]
additional_col_names = [field.field_name for field in global_add_fields]
if export_format == "csv":
# set additional data col at the end (remove it and inset it ...)
columns.remove(export_col_name_additional_data)
columns = columns + additional_col_names
columns.append(export_col_name_additional_data)
if additional_col_names:
serialize_result = [
as_dict_with_add_cols(export_view, row,
export_col_name_additional_data,
additional_col_names) for row in data
]
else:
serialize_result = [export_view.as_dict(row) for row in data]
return to_csv_resp(file_name, serialize_result, columns, ";")
elif export_format == "geojson":
if additional_col_names:
features = []
for row in data:
properties = as_dict_with_add_cols(
export_view, row, export_col_name_additional_data,
additional_col_names)
feature = Feature(properties=properties,
geometry=to_shape(
getattr(row, export_geom_column)))
features.append(feature)
serialize_result = FeatureCollection(features)
else:
serialize_result = FeatureCollection([
export_view.as_geofeature(d, fields=export_columns)
for d in data
])
return to_json_resp(serialize_result,
as_file=True,
filename=file_name,
indent=4,
extension="geojson")
else:
try:
db_cols = [
db_col for db_col in export_view.db_cols
if db_col.key in export_columns
]
dir_name, file_name = export_as_geo_file(
export_format=export_format,
export_view=export_view,
db_cols=db_cols,
geojson_col=None,
data=data,
file_name=file_name,
)
db_cols = [
db_col for db_col in export_view.db_cols
if db_col.key in export_columns
]
return send_from_directory(dir_name, file_name, as_attachment=True)
except GeonatureApiError as e:
message = str(e)
return render_template(
"error.html",
error=message,
redirect=current_app.config["URL_APPLICATION"] + "/#/occtax",
)
def export(info_role):
"""Export data from pr_occtax.v_export_occtax view (parameter)
.. :quickref: Occtax; Export data from pr_occtax.v_export_occtax
:query str format: format of the export ('csv', 'geojson', 'shapefile')
"""
export_view_name = blueprint.config["export_view_name"]
export_geom_column = blueprint.config["export_geom_columns_name"]
export_columns = blueprint.config["export_columns"]
export_srid = blueprint.config["export_srid"]
export_view = GenericTableGeo(
tableName=export_view_name,
schemaName="pr_occtax",
engine=DB.engine,
geometry_field=export_geom_column,
srid=export_srid,
)
releve_repository = ReleveRepository(export_view)
q = releve_repository.get_filtered_query(info_role, from_generic_table=True)
q = get_query_occtax_filters(
request.args,
export_view,
q,
from_generic_table=True,
obs_txt_column=blueprint.config["export_observer_txt_column"],
)
data = q.all()
file_name = datetime.datetime.now().strftime("%Y_%m_%d_%Hh%Mm%S")
file_name = filemanager.removeDisallowedFilenameChars(file_name)
export_format = request.args["format"] if "format" in request.args else "geojson"
if export_format == "csv":
columns = (
export_columns
if len(export_columns) > 0
else [db_col.key for db_col in export_view.db_cols]
)
return to_csv_resp(
file_name, [export_view.as_dict(d) for d in data], columns, ";"
)
elif export_format == "geojson":
results = FeatureCollection(
[export_view.as_geofeature(d, columns=export_columns) for d in data]
)
return to_json_resp(
results, as_file=True, filename=file_name, indent=4, extension="geojson"
)
else:
try:
db_cols = [
db_col for db_col in export_view.db_cols if db_col.key in export_columns
]
dir_name, file_name = export_as_geo_file(
export_format=export_format,
export_view=export_view,
db_cols=db_cols,
geojson_col=None,
data=data,
file_name=file_name,
)
return send_from_directory(dir_name, file_name, as_attachment=True)
except GeonatureApiError as e:
message = str(e)
return render_template(
"error.html",
error=message,
redirect=current_app.config["URL_APPLICATION"] + "/#/occtax",
)
def export_observations_web(info_role):
"""Optimized route for observations web export.
.. :quickref: Synthese;
This view is customisable by the administrator
Some columns are mandatory: id_synthese, geojson and geojson_local to generate the exported files
POST parameters: Use a list of id_synthese (in POST parameters) to filter the v_synthese_for_export_view
:query str export_format: str<'csv', 'geojson', 'shapefiles', 'gpkg'>
"""
params = request.args
export_format = params.get("export_format", "csv")
# Test export_format
if not export_format in current_app.config["SYNTHESE"]["EXPORT_FORMAT"]:
raise BadRequest("Unsupported format")
# set default to csv
export_view = GenericTableGeo(
tableName="v_synthese_for_export",
schemaName="gn_synthese",
engine=DB.engine,
geometry_field=None,
srid=current_app.config["LOCAL_SRID"],
)
# get list of id synthese from POST
id_list = request.get_json()
db_cols_for_shape = []
columns_to_serialize = []
# loop over synthese config to get the columns for export
for db_col in export_view.db_cols:
if db_col.key in current_app.config["SYNTHESE"]["EXPORT_COLUMNS"]:
db_cols_for_shape.append(db_col)
columns_to_serialize.append(db_col.key)
query = select([export_view.tableDef]).where(
export_view.tableDef.columns[current_app.config["SYNTHESE"]["EXPORT_ID_SYNTHESE_COL"]].in_(
id_list
)
)
synthese_query_class = SyntheseQuery(
export_view.tableDef,
query,
{},
id_synthese_column=current_app.config["SYNTHESE"]["EXPORT_ID_SYNTHESE_COL"],
id_dataset_column=current_app.config["SYNTHESE"]["EXPORT_ID_DATASET_COL"],
observers_column=current_app.config["SYNTHESE"]["EXPORT_OBSERVERS_COL"],
id_digitiser_column=current_app.config["SYNTHESE"]["EXPORT_ID_DIGITISER_COL"],
with_generic_table=True,
)
# check R and E CRUVED to know if we filter with cruved
cruved = cruved_scope_for_user_in_module(info_role.id_role, module_code="SYNTHESE")[0]
if cruved["R"] > cruved["E"]:
synthese_query_class.filter_query_with_cruved(info_role)
results = DB.session.execute(synthese_query_class.query.limit(
current_app.config["SYNTHESE"]["NB_MAX_OBS_EXPORT"])
)
file_name = datetime.datetime.now().strftime("%Y_%m_%d_%Hh%Mm%S")
file_name = filemanager.removeDisallowedFilenameChars(file_name)
if export_format == "csv":
formated_data = [export_view.as_dict(d, columns=columns_to_serialize) for d in results]
return to_csv_resp(file_name, formated_data, separator=";", columns=columns_to_serialize)
elif export_format == "geojson":
features = []
for r in results:
geometry = ast.literal_eval(
getattr(r, current_app.config["SYNTHESE"]["EXPORT_GEOJSON_4326_COL"])
)
feature = Feature(
geometry=geometry,
properties=export_view.as_dict(r, columns=columns_to_serialize),
)
features.append(feature)
results = FeatureCollection(features)
return to_json_resp(results, as_file=True, filename=file_name, indent=4)
else:
try:
dir_name, file_name = export_as_geo_file(
export_format=export_format,
export_view=export_view,
db_cols=db_cols_for_shape,
geojson_col=current_app.config["SYNTHESE"]["EXPORT_GEOJSON_LOCAL_COL"],
data=results,
file_name=file_name,
)
return send_from_directory(dir_name, file_name, as_attachment=True)
except GeonatureApiError as e:
message = str(e)
return render_template(
"error.html",
error=message,
redirect=current_app.config["URL_APPLICATION"] + "/#/synthese",
)