ssb-sgis 1.1.0__tar.gz → 1.1.2__tar.gz

{ssb_sgis-1.1.0 → ssb_sgis-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ssb-sgis
-Version: 1.1.0
+Version: 1.1.2
 Summary: GIS functions used at Statistics Norway.
 Home-page: https://github.com/statisticsnorway/ssb-sgis
 License: MIT

{ssb_sgis-1.1.0 → ssb_sgis-1.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ssb-sgis"
-version = "1.1.0"
+version = "1.1.2"
 description = "GIS functions used at Statistics Norway."
 authors = ["Morten Letnes <morten.letnes@ssb.no>"]
 license = "MIT"

{ssb_sgis-1.1.0 → ssb_sgis-1.1.2}/src/sgis/__init__.py

@@ -1,10 +1,6 @@
-config = {
-    "n_jobs": 1,
-}
-
-
 import sgis.raster.indices as indices
 
+from .conf import config
 from .geopandas_tools.bounds import Gridlooper
 from .geopandas_tools.bounds import bounds_to_points
 from .geopandas_tools.bounds import bounds_to_polygon
@@ -89,6 +85,7 @@ from .io.opener import opener
 from .io.read_parquet import read_parquet_url
 from .maps.examine import Examine
 from .maps.explore import Explore
+from .maps.explore import HtmlViewer
 from .maps.httpserver import run_html_server
 from .maps.legend import Legend
 from .maps.maps import clipmap

ssb_sgis-1.1.2/src/sgis/conf.py

@@ -0,0 +1,72 @@
+try:
+    from gcsfs import GCSFileSystem
+
+    file_system = GCSFileSystem
+except ImportError:
+    import datetime
+    import glob
+    import io
+    import os
+    import pathlib
+    from concurrent.futures import ThreadPoolExecutor
+    from typing import Any
+
+    class LocalFileSystem:
+        """Mimicks GCS's FileSystem but using standard library (os, glob)."""
+
+        @staticmethod
+        def glob(
+            path: str,
+            detail: bool = False,
+            recursive: bool = True,
+            include_hidden: bool = True,
+            **kwargs,
+        ) -> list[dict] | list[str]:
+            """Like GCSFileSystem.glob."""
+            relevant_paths = glob.iglob(
+                path, recursive=recursive, include_hidden=include_hidden, **kwargs
+            )
+
+            if not detail:
+                return list(relevant_paths)
+            with ThreadPoolExecutor() as executor:
+                return list(executor.map(get_file_info, relevant_paths))
+
+        @classmethod
+        def ls(
+            cls, path: str, detail: bool = False, **kwargs
+        ) -> list[dict] | list[str]:
+            """Like GCSFileSystem.ls."""
+            return cls().glob(
+                str(pathlib.Path(path) / "**"), detail=detail, recursive=False, **kwargs
+            )
+
+        @staticmethod
+        def info(path) -> dict[str, Any]:
+            """Like GCSFileSystem.info."""
+            return get_file_info(path)
+
+        @staticmethod
+        def open(path: str, *args, **kwargs) -> io.TextIOWrapper:
+            """Built in open."""
+            return open(path, *args, **kwargs)
+
+        @staticmethod
+        def exists(path: str) -> bool:
+            """os.path.exists."""
+            return os.path.exists(path)
+
+    def get_file_info(path) -> dict[str, str | float]:
+        return {
+            "updated": datetime.datetime.fromtimestamp(os.path.getmtime(path)),
+            "size": os.path.getsize(path),
+            "name": path,
+            "type": "directory" if os.path.isdir(path) else "file",
+        }
+
+    file_system = LocalFileSystem
+
+config = {
+    "n_jobs": 1,
+    "file_system": file_system,
+}