"""
Set the datadir path to the local data directory
"""
import os
from pyproj.exceptions import DataDirError
_USER_PROJ_DATA = None
[文档]def set_data_dir(proj_data_dir):
"""
Set the data directory for PROJ.4 to use.
Parameters
----------
proj_data_dir: str
The path to rhe PROJ.4 data directory.
"""
global _USER_PROJ_DATA
_USER_PROJ_DATA = proj_data_dir
[文档]def get_data_dir():
"""
The order of preference for the data directory is:
1. The one set by pyproj.datadir.set_data_dir (if exists & valid)
2. The internal proj directory (if exists & valid)
3. The directory in PROJ_LIB
Returns
-------
str: The valid data directory.
"""
global _USER_PROJ_DATA
internal_datadir = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "proj_dir", "share", "proj"
)
def valid_data_dir(potential_data_dir):
if potential_data_dir is not None and os.path.exists(
os.path.join(potential_data_dir, "proj.db")
):
return True
return False
proj_data_dir = None
if valid_data_dir(_USER_PROJ_DATA):
proj_data_dir = _USER_PROJ_DATA
elif valid_data_dir(internal_datadir):
proj_data_dir = internal_datadir
else:
proj_lib_dirs = os.environ.get("PROJ_LIB", "")
for proj_lib_dir in proj_lib_dirs.split(";"):
if valid_data_dir(proj_lib_dir):
proj_data_dir = proj_lib_dir
break
if proj_data_dir is None:
raise DataDirError(
"Valid PROJ.4 data directory not found."
"Either set the path using the environmental variable PROJ_LIB or "
"with `pyproj.datadir.set_data_dir`."
)
return proj_data_dir