Remove pkg_resources

gee-community · Feb 2, 2025 · f6f5831 · f6f5831
1 parent c4de2a1
commit f6f5831
Show file tree

Hide file tree

Showing 8 changed files with 22 additions and 35 deletions.
diff --git a/geemap/common.py b/geemap/common.py
@@ -20,6 +20,7 @@
 import urllib.request
 import warnings
 import zipfile
+import importlib.resources
 
 import ee
 import ipywidgets as widgets
@@ -3454,12 +3455,11 @@ def create_colorbar(
     import decimal
 
     # import io
-    import pkg_resources
     from colour import Color
     from PIL import Image, ImageDraw, ImageFont
 
     warnings.simplefilter("ignore")
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
 
     if out_file is None:
         filename = "colorbar_" + random_string() + ".png"
@@ -4120,11 +4120,9 @@ def ee_api_to_csv(outfile=None, timeout=300, proxies=None):
         timeout (int, optional): Timeout in seconds. Defaults to 300.
         proxies (dict, optional): Proxy settings. Defaults to None.
     """
-    import pkg_resources
-
     from bs4 import BeautifulSoup
 
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     data_dir = os.path.join(pkg_dir, "data")
     template_dir = os.path.join(data_dir, "template")
     csv_file = os.path.join(template_dir, "ee_api_docs.csv")
@@ -4240,9 +4238,7 @@ def read_api_csv():
     """
     import copy
 
-    import pkg_resources
-
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     data_dir = os.path.join(pkg_dir, "data")
     template_dir = os.path.join(data_dir, "template")
     csv_file = os.path.join(template_dir, "ee_api_docs.csv")
@@ -5129,9 +5125,8 @@ def vis_to_qml(ee_class_table, out_qml):
         ee_class_table (str): An Earth Engine class table with triple quotes.
         out_qml (str): File path to the output QGIS Layer Style (.qml).
     """
-    import pkg_resources
 
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     data_dir = os.path.join(pkg_dir, "data")
     template_dir = os.path.join(data_dir, "template")
     qml_template = os.path.join(template_dir, "NLCD.qml")
@@ -5178,9 +5173,7 @@ def create_nlcd_qml(out_qml):
     Args:
         out_qml (str): File path to the output qml.
     """
-    import pkg_resources
-
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     data_dir = os.path.join(pkg_dir, "data")
     template_dir = os.path.join(data_dir, "template")
     qml_template = os.path.join(template_dir, "NLCD.qml")
@@ -10012,9 +10005,7 @@ def get_census_dict(reset=False):
     Returns:
         dict: A dictionary of Census data.
     """
-    import pkg_resources
-
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     census_data = os.path.join(pkg_dir, "data/census_data.json")
 
     if reset:
@@ -13729,10 +13720,9 @@ def create_legend(
         str: The HTML code of the legend.
     """
 
-    import pkg_resources
     from .legends import builtin_legends
 
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     legend_template = os.path.join(pkg_dir, "data/template/legend_style.html")
 
     if draggable:

diff --git a/geemap/datasets.py b/geemap/datasets.py
@@ -12,7 +12,7 @@
 from pathlib import Path
 
 import ipywidgets as widgets
-import pkg_resources
+import importlib.resources
 from box import Box
 from IPython.display import display
 
@@ -25,7 +25,7 @@ def get_data_csv() -> str:
     Returns:
         str: File path to the CSV file.
     """
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     template_dir = os.path.join(pkg_dir, "data/template")
     data_csv = os.path.join(template_dir, "ee_data_catalog.csv")
     return data_csv

diff --git a/geemap/examples/__init__.py b/geemap/examples/__init__.py
@@ -1,8 +1,8 @@
 import box
 import os
-import pkg_resources
+import importlib.resources
 
-_pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+_pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
 _datasets_path = os.path.join(_pkg_dir, "examples/datasets.txt")
 _baseurl = (
     "https://raw.githubusercontent.com/gee-community/geemap/master/examples/data/"

diff --git a/geemap/legends.py b/geemap/legends.py
@@ -493,8 +493,6 @@ def ee_table_to_legend(in_table, out_file):
         in_table (str): The input file path (*.txt) to the Earth Engine color table.
         out_file (str): The output file path (*.txt) to the legend dictionary.
     """
-    # pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
-    # ee_legend_table = os.path.join(pkg_dir, "data/template/ee_legend_table.txt")
 
     if not os.path.exists(in_table):
         print("The class table does not exist.")

diff --git a/geemap/map_widgets.py b/geemap/map_widgets.py
@@ -1437,10 +1437,10 @@ def _select_dataset(self, dataset_title):
 
     def get_ee_example(self, asset_id):
         try:
-            import pkg_resources
+            import importlib.resources
 
-            pkg_dir = os.path.dirname(
-                pkg_resources.resource_filename("geemap", "geemap.py")
+            pkg_dir = str(
+                importlib.resources.files("geemap").joinpath("geemap.py").parent
             )
             with open(os.path.join(pkg_dir, "data/gee_f.json"), encoding="utf-8") as f:
                 functions = json.load(f)

diff --git a/geemap/maplibregl.py b/geemap/maplibregl.py
@@ -2468,12 +2468,10 @@ def add_legend(
         Returns:
             None
         """
-        import pkg_resources
+        import importlib.resources
         from .legends import builtin_legends
 
-        pkg_dir = os.path.dirname(
-            pkg_resources.resource_filename("geemap", "geemap.py")
-        )
+        pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
         legend_template = os.path.join(pkg_dir, "data/template/legend.html")
 
         if not os.path.exists(legend_template):

diff --git a/geemap/timelapse.py b/geemap/timelapse.py
@@ -412,11 +412,12 @@ def add_text_to_gif(
     # import io
     import warnings
 
-    import pkg_resources
+    import importlib.resources
     from PIL import Image, ImageDraw, ImageFont, ImageSequence
 
     warnings.simplefilter("ignore")
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     default_font = os.path.join(pkg_dir, "data/fonts/arial.ttf")
 
     in_gif = os.path.abspath(in_gif)

diff --git a/geemap/toolbar.py b/geemap/toolbar.py
@@ -1574,9 +1574,9 @@ def set_properties(_, geometry):
 
 def get_tools_dict():
     import pandas as pd
-    import pkg_resources
+    import importlib.resources
 
-    pkg_dir = os.path.dirname(pkg_resources.resource_filename("geemap", "geemap.py"))
+    pkg_dir = str(importlib.resources.files("geemap").joinpath("geemap.py").parent)
     toolbox_csv = os.path.join(pkg_dir, "data/template/toolbox.csv")
 
     df = pd.read_csv(toolbox_csv).set_index("index")