|
14 | 14 | from landcoverpy.mongo import MongoConnection |
15 | 15 | from landcoverpy.minio import MinioConnection |
16 | 16 | from landcoverpy.utilities.aoi_tiles import get_list_of_tiles_in_iberian_peninsula |
17 | | -from landcoverpy.utilities.geometries import _group_polygons_by_tile, _kmz_to_geojson |
| 17 | +from landcoverpy.utilities.geometries import _group_validated_data_points_by_tile, _kmz_to_geojson, _csv_to_geojson |
18 | 18 | from landcoverpy.utilities.utils import get_products_by_tile_and_date, get_season_dict |
19 | 19 | from landcoverpy.workflow import _process_tile |
20 | 20 |
|
@@ -66,23 +66,20 @@ def time_training_dataset(client: Client = None): |
66 | 66 | tiles = get_list_of_tiles_in_iberian_peninsula() |
67 | 67 | tile = random.choice(tiles) |
68 | 68 |
|
69 | | - geojson_files = [] |
70 | | - for data_class in glob(join(settings.DB_DIR, "*.kmz")): |
71 | | - if not Path.exists(Path(data_class.replace("kmz","geojson"))): |
72 | | - print(f"Parsing database to geojson: {data_class}") |
73 | | - _kmz_to_geojson(data_class) |
| 69 | + data_file = settings.DB_FILE |
| 70 | + if data_file.endswith(".kmz"): |
| 71 | + data_file = _kmz_to_geojson(data_file) |
| 72 | + if data_file.endswith(".csv"): |
| 73 | + data_file = _csv_to_geojson(data_file, sep=',') |
74 | 74 |
|
75 | | - for data_class in glob(join(settings.DB_DIR, "*.geojson")): |
76 | | - print(f"Working with database {data_class}") |
77 | | - geojson_files.append(data_class) |
78 | | - polygons_per_tile = _group_polygons_by_tile(*geojson_files) |
| 75 | + polygons_per_tile = _group_validated_data_points_by_tile(data_file) |
79 | 76 |
|
80 | 77 | metadata_filename = "metadata.json" |
81 | | - metadata_filepath = join(settings.TMP_DIR, settings.LAND_COVER_MODEL_FOLDER, metadata_filename) |
| 78 | + metadata_filepath = join(settings.TMP_DIR, "land-cover", metadata_filename) |
82 | 79 |
|
83 | 80 | minio.fget_object( |
84 | 81 | bucket_name=settings.MINIO_BUCKET_MODELS, |
85 | | - object_name=join(settings.MINIO_DATA_FOLDER_NAME, settings.LAND_COVER_MODEL_FOLDER, metadata_filename), |
| 82 | + object_name=join(settings.MINIO_DATA_FOLDER_NAME, "land-cover", metadata_filename), |
86 | 83 | file_path=metadata_filepath, |
87 | 84 | ) |
88 | 85 |
|
@@ -129,11 +126,11 @@ def time_predicting_tile(client: Client = None): |
129 | 126 |
|
130 | 127 | # For predictions, read the rasters used in "metadata.json". |
131 | 128 | metadata_filename = "metadata.json" |
132 | | - metadata_filepath = join(settings.TMP_DIR, settings.LAND_COVER_MODEL_FOLDER, metadata_filename) |
| 129 | + metadata_filepath = join(settings.TMP_DIR, "land-cover", metadata_filename) |
133 | 130 |
|
134 | 131 | minio.fget_object( |
135 | 132 | bucket_name=settings.MINIO_BUCKET_MODELS, |
136 | | - object_name=join(settings.MINIO_DATA_FOLDER_NAME, settings.LAND_COVER_MODEL_FOLDER, metadata_filename), |
| 133 | + object_name=join(settings.MINIO_DATA_FOLDER_NAME, "land-cover", metadata_filename), |
137 | 134 | file_path=metadata_filepath, |
138 | 135 | ) |
139 | 136 |
|
|
0 commit comments