lazy load ecoscope

atmorling · May 28, 2024 · ce65132 · ce65132
1 parent 86809dd
commit ce65132
Show file tree

Hide file tree

Showing 7 changed files with 80 additions and 35 deletions.
diff --git a/ecoscope/__init__.py b/ecoscope/__init__.py
@@ -1,4 +1,9 @@
-from ecoscope import base, io, mapping, plotting, analysis
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submodules=["base", "io", "mapping", "plotting", "analysis"],
+)
 
 ASCII = """\
  _____
@@ -49,16 +54,17 @@ def init(silent=False, selenium=False, force=False):
     warnings.filterwarnings("ignore", message=".*initial implementation of Parquet.*")
 
     import geopandas as gpd
+    from ecoscope.mapping.map import EcoMap
 
     def explore(data, *args, **kwargs):
         """
         Monkey-patched `geopandas.explore._explore` to use EcoMap instead.
         """
         initialized = "m" in kwargs
         if not initialized:
-            kwargs["m"] = mapping.EcoMap()
+            kwargs["m"] = EcoMap()
 
-        if isinstance(kwargs["m"], mapping.EcoMap):
+        if isinstance(kwargs["m"], EcoMap):
             m = kwargs.pop("m")
             m.add_gdf(data, *args, **kwargs)
             if not initialized:

diff --git a/ecoscope/analysis/UD/__init__.py b/ecoscope/analysis/UD/__init__.py
@@ -1,4 +1,9 @@
-from ecoscope.analysis.UD.etd_range import calculate_etd_range
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submod_attrs={"etd_range": ["calculate_etd_range"]},
+)
 
 __all__ = [
     "calculate_etd_range",

diff --git a/ecoscope/analysis/__init__.py b/ecoscope/analysis/__init__.py
@@ -1,7 +1,13 @@
-from ecoscope.analysis import UD, seasons
-from ecoscope.analysis.ecograph import Ecograph, get_feature_gdf
-from ecoscope.analysis.percentile import get_percentile_area
-from ecoscope.analysis.speed import SpeedDataFrame
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submodules=["UD", "seasons", "speed"],
+    submod_attrs={
+        "ecograph": ["Ecograph", "get_feature_gdf"],
+        "percentile": ["get_percentile_area"],
+    },
+)
 
 __all__ = [
     "Ecograph",

diff --git a/ecoscope/base/__init__.py b/ecoscope/base/__init__.py
@@ -1,12 +1,19 @@
-from ecoscope.base._dataclasses import (
-    RelocsCoordinateFilter,
-    RelocsDateRangeFilter,
-    RelocsDistFilter,
-    RelocsSpeedFilter,
-    TrajSegFilter,
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submod_attrs={
+        "_dataclasses": [
+            "RelocsCoordinateFilter",
+            "RelocsDateRangeFilter",
+            "RelocsDistFilter",
+            "RelocsSpeedFilter",
+            "TrajSegFilter",
+        ],
+        "base": ["EcoDataFrame", "Relocations", "Trajectory"],
+        "utils": ["cachedproperty", "create_meshgrid", "groupby_intervals", "to_EarthLocation", "is_night"],
+    },
 )
-from ecoscope.base.base import EcoDataFrame, Relocations, Trajectory
-from ecoscope.base.utils import cachedproperty, create_meshgrid, groupby_intervals, to_EarthLocation, is_night
 
 __all__ = [
     "EcoDataFrame",

diff --git a/ecoscope/io/__init__.py b/ecoscope/io/__init__.py
@@ -1,6 +1,13 @@
-from ecoscope.io import earthranger, eetools, raster, utils
-from ecoscope.io.earthranger import EarthRangerIO
-from ecoscope.io.utils import download_file
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submodules=["earthranger", "eetools", "raster", "utils"],
+    submod_attrs={
+        "earthranger": ["EarthRangerIO"],
+        "utils": ["download_file"],
+    },
+)
 
 __all__ = [
     "earthranger",

diff --git a/ecoscope/mapping/__init__.py b/ecoscope/mapping/__init__.py
@@ -1,11 +1,18 @@
-from ecoscope.mapping.map import (
-    ControlElement,
-    EcoMap,
-    FloatElement,
-    NorthArrowElement,
-    ScaleElement,
-    GeoTIFFElement,
-    PrintControl,
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submod_attrs={
+        "map": [
+            "ControlElement",
+            "EcoMap",
+            "FloatElement",
+            "NorthArrowElement",
+            "ScaleElement",
+            "GeoTIFFElement",
+            "PrintControl",
+        ],
+    },
 )
 
 __all__ = [

diff --git a/ecoscope/plotting/__init__.py b/ecoscope/plotting/__init__.py
@@ -1,11 +1,18 @@
-from ecoscope.plotting.plot import (
-    EcoPlotData,
-    add_seasons,
-    ecoplot,
-    mcp,
-    nsd,
-    plot_seasonal_dist,
-    speed,
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submod_attrs={
+        "plot": [
+            "EcoPlotData",
+            "add_seasons",
+            "ecoplot",
+            "mcp",
+            "nsd",
+            "plot_seasonal_dist",
+            "speed",
+        ],
+    },
 )
 
 __all__ = [