1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import os
__all__ = ["available", "get_path"]
_module_path = os.path.dirname(__file__)
_available_dir = [p for p in next(os.walk(_module_path))[1] if not p.startswith("__")]
_available_zip = {"nybb": "nybb_16a.zip"}
available = _available_dir + list(_available_zip.keys())
def get_path(dataset):
"""
Get the path to the data file.
Parameters
----------
dataset : str
The name of the dataset. See ``geopandas.datasets.available`` for
all options.
Examples
--------
>>> geopandas.datasets.get_path("naturalearth_lowres") # doctest: +SKIP
'.../python3.8/site-packages/geopandas/datasets/\
naturalearth_lowres/naturalearth_lowres.shp'
"""
if dataset in _available_dir:
return os.path.abspath(os.path.join(_module_path, dataset, dataset + ".shp"))
elif dataset in _available_zip:
fpath = os.path.abspath(os.path.join(_module_path, _available_zip[dataset]))
return "zip://" + fpath
else:
msg = "The dataset '{data}' is not available. ".format(data=dataset)
msg += "Available datasets are {}".format(", ".join(available))
raise ValueError(msg)
|
