ssb-sgis 1.2.9__py3-none-any.whl → 1.2.11__py3-none-any.whl

sgis/geopandas_tools/bounds.py

@@ -634,11 +634,13 @@ def bounds_to_polygon(
     1  POLYGON ((0.00000 0.00000, 0.00000 0.00000, 0....
 
     """
+    bounds = gdf.bounds.values
+    boxes = box(bounds[:, 0], bounds[:, 1], bounds[:, 2], bounds[:, 3])
     if isinstance(gdf, GeoSeries):
-        return GeoSeries([box(*arr) for arr in gdf.bounds.values], index=gdf.index)
+        return GeoSeries(boxes, index=gdf.index)
     if copy:
         gdf = gdf.copy()
-    gdf.geometry = [box(*arr) for arr in gdf.bounds.values]
+    gdf.geometry = boxes
     return gdf
 
 

sgis/geopandas_tools/neighbors.py

@@ -13,6 +13,7 @@ import shapely
 from geopandas import GeoDataFrame
 from geopandas import GeoSeries
 from pandas import DataFrame
+from pandas import Index
 from pandas import MultiIndex
 from pandas import Series
 
@@ -110,6 +111,8 @@ def get_neighbor_indices(
     if gdf.crs != neighbors.crs:
         raise ValueError(f"'crs' mismatch. Got {gdf.crs} and {neighbors.crs}")
 
+    index_name = gdf.index.name
+
     if rtree_runner is None:
         rtree_runner = _get_instance(config, "rtree_runner", n_jobs=n_jobs)
 
@@ -141,6 +144,8 @@ def get_neighbor_indices(
         )
     index_mapper1 = {i: x for i, x in enumerate(gdf.index)}
     left = np.array([index_mapper1[i] for i in left])
+    if index_name:
+        left = Index(left, name=index_name)
     index_mapper2 = {i: x for i, x in enumerate(neighbors.index)}
     right = np.array([index_mapper2[i] for i in right])
     return Series(right, index=left, name="neighbor_index")

sgis/io/dapla_functions.py

@@ -385,6 +385,9 @@ def get_bounds_series(
     """
     file_system = _get_file_system(file_system, {})
 
+    if isinstance(paths, (str | Path)):
+        paths = [paths]
+
     threads = (
         min(len(paths), int(multiprocessing.cpu_count())) or 1 if use_threads else 1
     )
@@ -457,7 +460,7 @@ def write_geopandas(
         )
 
     if not len(df) and get_child_paths(gcs_path, file_system):
-        # no need to write empty df
+        # no need to write empty df for partitioned parquet
         return
     elif not len(df):
         if pandas_fallback:
@@ -465,8 +468,13 @@ def write_geopandas(
             df.geometry = df.geometry.astype(str)
             df.geometry = None
         try:
+            file_format: str = Path(gcs_path).suffix.lstrip(".")
+            write_method: Callable = getattr(df, f"to_{file_format}")
+            if file_format == "parquet":
+                kwargs["engine"] = "pyarrow"
             with file_system.open(gcs_path, "wb") as file:
-                df.to_parquet(file, **kwargs)
+                write_method(file, **kwargs)
+
         except Exception as e:
             more_txt = PANDAS_FALLBACK_INFO if not pandas_fallback else ""
             raise e.__class__(
@@ -486,7 +494,7 @@ def write_geopandas(
                 **kwargs,
             )
         with file_system.open(gcs_path, mode="wb") as file:
-            df.to_parquet(file, **kwargs)
+            _to_geopandas(df, file, **kwargs)
         return
 
     layer = kwargs.pop("layer", None)

sgis/maps/explore.py

@@ -14,7 +14,6 @@ from statistics import mean
 from typing import Any
 from typing import ClassVar
 
-import branca as bc
 import folium
 import geopandas as gpd
 import joblib
@@ -723,9 +722,9 @@ class Explore(Map):
 
         bounds = self._get_bounds(gdf)
 
-        if bounds is None:
-            self.map = None
-            return
+        if bounds is None and self.mask is not None:
+            bounds = self._get_bounds(_to_or_set_crs(to_gdf(self.mask), 4326))
+
         self.map = self._make_folium_map(
             bounds=bounds,
             max_zoom=self.max_zoom,
@@ -790,6 +789,8 @@ class Explore(Map):
                 map_.add_child(tile)
 
     def _create_continous_map(self):
+        import branca as bc
+
         self._prepare_continous_map()
         if self.scheme:
             classified = self._classify_from_bins(self._gdf, bins=self.bins)
@@ -804,9 +805,8 @@ class Explore(Map):
 
         gdf = self._prepare_gdf_for_map(self._gdf)
         bounds = self._get_bounds(gdf)
-        if bounds is None:
-            self.map = None
-            return
+        if bounds is None and self.mask is not None:
+            bounds = self._get_bounds(_to_or_set_crs(to_gdf(self.mask), 4326))
         self.map = self._make_folium_map(
             bounds=bounds,
             max_zoom=self.max_zoom,
@@ -899,6 +899,8 @@ class Explore(Map):
         map_kwds: dict | None = None,
         **kwargs,
     ):
+        import branca as bc
+
         if not map_kwds:
             map_kwds = {}
 
@@ -908,7 +910,7 @@ class Explore(Map):
         # create folium.Map object
         # Get bounds to specify location and map extent
         location = kwargs.pop("location", None)
-        if location is None:
+        if location is None and bounds is not None:
             x = mean([bounds[0], bounds[2]])
             y = mean([bounds[1], bounds[3]])
             location = (y, x)
@@ -1130,8 +1132,9 @@ class Explore(Map):
             if gdf.index.name is not None:
                 gdf = gdf.reset_index()
             # specify fields to show in the tooltip
-            tooltip = _tooltip_popup("tooltip", tooltip, gdf, **tooltip_kwds)
-            popup = _tooltip_popup("popup", popup, gdf, **popup_kwds)
+            cols = list(gdf.columns.difference({df.geometry.name}))
+            tooltip = _tooltip_popup("tooltip", tooltip, cols, **tooltip_kwds)
+            popup = _tooltip_popup("popup", popup, cols, **popup_kwds)
         else:
             tooltip = None
             popup = None
@@ -1153,7 +1156,7 @@ class Explore(Map):
 
 
 def _tooltip_popup(
-    type_: str, fields: Any, gdf: GeoDataFrame, **kwargs
+    type_: str, fields: Any, columns: list[str], **kwargs
 ) -> folium.GeoJsonTooltip | folium.GeoJsonPopup:
     """Get tooltip or popup."""
     # specify fields to show in the tooltip
@@ -1161,9 +1164,9 @@ def _tooltip_popup(
         return None
     else:
         if fields is True:
-            fields = gdf.columns.drop(gdf.geometry.name).to_list()
+            fields = columns
         elif isinstance(fields, int):
-            fields = gdf.columns.drop(gdf.geometry.name).to_list()[:fields]
+            fields = columns[:fields]
         elif isinstance(fields, str):
             fields = [fields]
 
@@ -1526,3 +1529,10 @@ def _image_collection_to_background_map(
         out = sorted(out, key=lambda x: x["label"])
 
     return out
+
+
+def _to_or_set_crs(df, crs):
+    try:
+        return df.to_crs(crs)
+    except Exception:
+        return df.set_crs(crs)

sgis/maps/map.py

@@ -721,7 +721,7 @@ class Map:
 
         assert gdf.index.is_unique
 
-        # if equal lenght, convert to integer and check for equality
+        # if equal length, convert to integer and check for equality
         if len(bins) == len(self._unique_values):
             if gdf[self._column].isna().all():
                 return np.repeat(len(bins), len(gdf))

sgis/maps/wms.py

@@ -71,11 +71,10 @@ class NorgeIBilderWms(WmsLoader):
     not_contains: str | Iterable[str] | None = None
     show: bool | Iterable[int] | int = False
     _use_json: bool = True
+    url: str = "https://wms.geonorge.no/skwms1/wms.nib-prosjekter"
 
     def load_tiles(self, verbose: bool = False) -> None:
         """Load all Norge i bilder tiles into self.tiles."""
-        url = "https://wms.geonorge.no/skwms1/wms.nib-prosjekter?SERVICE=WMS&REQUEST=GetCapabilities"
-
         name_pattern = r"<Name>(.*?)</Name>"
         bbox_pattern = (
             r"<EX_GeographicBoundingBox>.*?"
@@ -86,7 +85,7 @@ class NorgeIBilderWms(WmsLoader):
         )
 
         all_tiles: list[dict] = []
-        with urlopen(url) as file:
+        with urlopen(self.url) as file:
             xml_data: str = file.read().decode("utf-8")
 
             for text in xml_data.split('<Layer queryable="1">')[1:]:
@@ -150,28 +149,6 @@ class NorgeIBilderWms(WmsLoader):
         url = "https://wms.geonorge.no/skwms1/wms.nib-mosaikk?SERVICE=WMS&REQUEST=GetCapabilities"
         wms = WebMapService(url, version="1.3.0")
         out = {}
-        # ttiles = {wms[layer].title: [] for layer in list(wms.contents)}
-        # for layer in list(wms.contents):
-        #     if wms[layer].title not in relevant_names:
-        #         continue
-        #     ttiles[wms[layer].title].append(layer)
-        # import pandas as pd
-
-        # df = pd.Series(ttiles).to_frame("title")
-        # df["n"] = df["title"].str.len()
-        # df = df.sort_values("n")
-        # for x in df["title"]:
-        #     if len(x) == 1:
-        #         continue
-        #     bounds = {tuple(wms[layer].boundingBoxWGS84) for layer in x}
-        #     if len(bounds) <= 1:
-        #         continue
-        #     print()
-        #     for layer in x:
-        #         print(layer)
-        #         print(wms[layer].title)
-        #         bbox = wms[layer].boundingBoxWGS84
-        #         print(bbox)
 
         for layer in list(wms.contents):
             title = wms[layer].title
@@ -267,7 +244,7 @@ class NorgeIBilderWms(WmsLoader):
         if not len(tile_layers):
             return tile_layers
 
-        if isinstance(self.show, int):
+        if isinstance(self.show, int) and not isinstance(self.show, bool):
             tile = tile_layers[list(tile_layers)[self.show]]
             tile.show = True
         elif isinstance(self.show, Iterable):
@@ -279,6 +256,8 @@ class NorgeIBilderWms(WmsLoader):
 
     def _filter_tiles(self, mask):
         """Filter relevant dates with pandas and geopandas because fast."""
+        if not self.tiles:
+            return pd.DataFrame(columns=["name", "year", "geometry", "bbox"])
         df = pd.DataFrame(self.tiles)
         filt = (df["name"].notna()) & (df["year"].str.contains("|".join(self.years)))
         if self.contains:

{ssb_sgis-1.2.9.dist-info → ssb_sgis-1.2.11.dist-info}/METADATA

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

{ssb_sgis-1.2.9.dist-info → ssb_sgis-1.2.11.dist-info}/RECORD

@@ -3,7 +3,7 @@ sgis/conf.py,sha256=pLqmvIKoKmXoW8chja3iQpbDUp9Z39vzl97MGH8ZHW0,2614
 sgis/debug_config.py,sha256=Tfr19kU46hSkkspsIJcrUWvlhaL4U3-f8xEPkujSCAQ,593
 sgis/exceptions.py,sha256=WNaEBPNNx0rmz-YDzlFX4vIE7ocJQruUTqS2RNAu2zU,660
 sgis/geopandas_tools/__init__.py,sha256=bo8lFMcltOz7TtWAi52_ekR2gd3mjfBfKeMDV5zuqFY,28
-sgis/geopandas_tools/bounds.py,sha256=YJyF0gp78hFAjLLZmDquRKCBAtbt7QouG3snTcJeNQs,23822
+sgis/geopandas_tools/bounds.py,sha256=PsA_rPEDIS0CopLtolWCB01UEdxc8gtsNbUqMd5AOLY,23855
 sgis/geopandas_tools/buffer_dissolve_explode.py,sha256=z9HvakazR_prXH862e8-gEe7UFbeI4rRTbUaBgPeMBk,19552
 sgis/geopandas_tools/centerlines.py,sha256=Q65Sx01SeAlulBEd9oaZkB2maBBNdLcJwAbTILg4SPU,11848
 sgis/geopandas_tools/cleaning.py,sha256=fST0xFztmyn-QUOAfvjZmu7aO_zPiolWK7gd7TR6ffI,24393
@@ -12,7 +12,7 @@ sgis/geopandas_tools/duplicates.py,sha256=TDDM4u1n7SIkyJrOfl1Lno92AmUPqtXBHsj1IU
 sgis/geopandas_tools/general.py,sha256=YRpNEdwTHyFdQOdAfbCmYXS7PxoDjXxoagwpteXkYdI,43937
 sgis/geopandas_tools/geocoding.py,sha256=sZjUW52ULhQWDLmU51C9_itBePkDuWkp8swvYaiYmJk,679
 sgis/geopandas_tools/geometry_types.py,sha256=ijQDbQaZPqPGjBl707H4yooNXpk21RXyatI7itnvqLk,7603
-sgis/geopandas_tools/neighbors.py,sha256=qTvnOyQyrcWQd0BDrmOvFQd8csMh005e_idJGmB4g7k,17485
+sgis/geopandas_tools/neighbors.py,sha256=VZGOwwC3-C6KpwLQ3j0K5cOVInmckxIXoGMqPGkemk4,17606
 sgis/geopandas_tools/overlay.py,sha256=5i9u8GgFuU0fCqzELsbIaoUPhw-E7eZHl_yKB0wEcGM,23464
 sgis/geopandas_tools/point_operations.py,sha256=JM4hvfIVxZaZdGNlGzcCurrKzkgC_b9hzbFYN42f9WY,6972
 sgis/geopandas_tools/polygon_operations.py,sha256=v-B9IgbFfm4dVHKPyzvmnNiqVCdtl9ddpCsQpZZ-9sU,49284
@@ -23,20 +23,20 @@ sgis/geopandas_tools/utils.py,sha256=X0pRvB1tWgV_0BCrRS1HU9LtLGnZCpvVPxyqM9JGb0Y
 sgis/helpers.py,sha256=4N6vFWQ3TYVzRHNcWY_fNa_GkFuaZB3vtCkkFND-qs0,9628
 sgis/io/__init__.py,sha256=uyBr20YDqB2bQttrd5q1JuGOvX32A-MSvS7Wmw5f5qg,177
 sgis/io/_is_dapla.py,sha256=wmfkSe98IrLhUg3dtXZusV6OVC8VlY1kbc5EQDf3P-Q,358
-sgis/io/dapla_functions.py,sha256=ISF1vaBZI0D9pMngWVvHDvQr9wFHagQaAGY3z5P3Iqo,31116
+sgis/io/dapla_functions.py,sha256=9NqkL67WdJJHczIWNdChHZp8T5hlXhcYucX7GWN1mCA,31430
 sgis/io/opener.py,sha256=HWO3G1NB6bpXKM94JadCD513vjat1o1TFjWGWzyVasg,898
 sgis/io/read_parquet.py,sha256=FvZYv1rLkUlrSaUY6QW6E1yntmntTeQuZ9ZRgCDO4IM,3776
 sgis/maps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sgis/maps/examine.py,sha256=Pb0dH8JazU5E2svfQrzHO1Bi-sjy5SeyY6zoeMO34jE,9369
-sgis/maps/explore.py,sha256=azfNNcEEiNp2lJxWN1Emf7kcmG60WL_ok_ilMw0-IL8,47914
+sgis/maps/explore.py,sha256=ANbhScthhwznLEUDDjRUBmniGmkdF9zftbzp9dYHu-U,48210
 sgis/maps/httpserver.py,sha256=TETSGOgLjKl3TquPGoIP0tCJCz7BIwmXrrzSCT7jhXE,2550
 sgis/maps/legend.py,sha256=qq2RkebuaNAdFztlXrDOWbN0voeK5w5VycmRKyx0NdM,26512
-sgis/maps/map.py,sha256=XWf3QJ6a4gZno2NziK1dKLRktJGGr-vn6eHudBlW9Uc,30758
+sgis/maps/map.py,sha256=07YK_oHa2z_yw_es06pd-80dV8YTI5mkTyJxyud707Y,30758
 sgis/maps/maps.py,sha256=fLK5WUlQ2YTm7t-8260lYxCFvpZN6j0Y-bVYCyv8NAY,23249
 sgis/maps/norge_i_bilder.json,sha256=G9DIN_2vyn-18UF5wUC-koZxFCbiNxMu0BbCJhMFJUk,15050340
 sgis/maps/thematicmap.py,sha256=Z3o_Bca0oty5Cn35pZfX5Qy52sXDVIMVSFD6IlZrovo,25111
 sgis/maps/tilesources.py,sha256=F4mFHxPwkiPJdVKzNkScTX6xbJAMIUtlTq4mQ83oguw,1746
-sgis/maps/wms.py,sha256=jQGF7na-IruG7FF3nHzvZB0BkdkhiWRRwfhkkx4aWkU,11970
+sgis/maps/wms.py,sha256=3ZPf-H3sirvHeLQdXGwE3j6w1sFQCkExsEWkvHl7a4g,11233
 sgis/networkanalysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sgis/networkanalysis/_get_route.py,sha256=dMX4Vm6O90ISIZPjQWuZMVMuEubkeSdC2osMCbFvrRU,7750
 sgis/networkanalysis/_od_cost_matrix.py,sha256=zkyPX7ObT996ahaFJ2oI0D0SqQWbWyfy_qLtXwValPg,3434
@@ -61,7 +61,7 @@ sgis/raster/indices.py,sha256=efJmgfPg_VuSzXFosXV661IendF8CwPFWtMhyP4TMUg,222
 sgis/raster/regex.py,sha256=4idTJ9vFtsGtbxcjJrx2VrpJJuDMP3bLdqF93Vc_cmY,3752
 sgis/raster/sentinel_config.py,sha256=nySDqn2R8M6W8jguoBeSAK_zzbAsqmaI59i32446FwY,1268
 sgis/raster/zonal.py,sha256=D4Gyptw-yOLTCO41peIuYbY-DANsJCG19xXDlf1QAz4,2299
-ssb_sgis-1.2.9.dist-info/LICENSE,sha256=np3IfD5m0ZUofn_kVzDZqliozuiO6wrktw3LRPjyEiI,1073
-ssb_sgis-1.2.9.dist-info/METADATA,sha256=yFVJtJkPtMRAxhQc10C0urtlsr6SNkjQkaBeFlCdZ88,11740
-ssb_sgis-1.2.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ssb_sgis-1.2.9.dist-info/RECORD,,
+ssb_sgis-1.2.11.dist-info/LICENSE,sha256=np3IfD5m0ZUofn_kVzDZqliozuiO6wrktw3LRPjyEiI,1073
+ssb_sgis-1.2.11.dist-info/METADATA,sha256=-U4ykZgkQ7azi_5WNdhpgOACp7lOHRcLYdYad-Ub2ks,11741
+ssb_sgis-1.2.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ssb_sgis-1.2.11.dist-info/RECORD,,