toolsos 0.2.6__py3-none-any.whl → 0.2.8__py3-none-any.whl

toolsos/geo.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from typing import Any, Union
 
+import geopandas as gpd
 import requests
 
 
@@ -17,15 +18,20 @@ def get_geo_json(
     Returns:
         dict[str, str]: geo json containg of the desired level and year
     """
-    base_url = "https://gitlab.com/os-amsterdam/datavisualisatie-onderzoek-en-statistiek/-/raw/main/public/geo/"
+    base_url = "https://gitlab.com/os-amsterdam/datavisualisatie-onderzoek-en-statistiek/-/raw/main/public/geo"
+
+    if level not in ["buurten", "wijken", "gebieden", "stadsdelen"]:
+        raise ValueError(
+            "level should be 'buurten', 'wijken', 'gebieden' or 'stadsdelen'"
+        )
 
     if mra:
         level = f"{level}-mra"
-        base_url = f"{base_url}mra/"
+        base_url = f"{base_url}/mra"
     else:
-        base_url = f"{base_url}amsterdam/"
+        base_url = f"{base_url}/amsterdam"
 
-    if (year <= 2020) & ~mra:
+    if (year <= 2020) and not mra:
         year = "2015-2020"
 
     if with_water:
@@ -33,11 +39,19 @@ def get_geo_json(
     else:
         url = f"{base_url}/{year}/{level}-{year}-zw-geo.json"
 
-    print(url)
     json = requests.get(url).json()
+
     return json
 
 
+def get_geo_dataframe(
+    level: str, year: Union[int, Any], with_water: bool = False, mra: bool = False
+):
+    return gpd.read_file(
+        get_geo_json(level=level, year=year, with_water=with_water, mra=mra)
+    )
+
+
 def extract_name_code_table(geo_json: dict[str, str]) -> dict[str, str]:
     """_summary_
 

toolsos/huisstijl/tables/tables.py

@@ -1,6 +1,7 @@
 import functools
 import json
 import re
+import warnings
 from itertools import groupby
 from typing import Any, Callable, Dict
 
@@ -277,7 +278,14 @@ def get_cols_id_with_pattern(df: pd.DataFrame, pattern: str) -> list[int]:
     Returns:
         list[int]: list with column indices matching pattern
     """
-    return [idx for idx, col in enumerate(df.columns) if re.findall(pattern, col)]
+
+    if isinstance(df.columns, pd.MultiIndex):
+        # Use the lowest level in the multi-index column
+        return [
+            idx for idx, col in enumerate(df.columns) if re.findall(pattern, col[-1])
+        ]
+    else:
+        return [idx for idx, col in enumerate(df.columns) if re.findall(pattern, col)]
 
 
 def get_string_cols_ids(df: pd.DataFrame) -> list[int]:
@@ -502,7 +510,8 @@ def write_table(
     blue_border: bool | None = True,
     blue_border_row_ids: int | list[int] | None = None,
     number_format: str = "0.0",
-    autofit_columns: str | None = "column_names",
+    autofit_columns: str | None = None,
+    column_widths: list[int] | None = None,
     min_column_width: int | None = None,
     col_filter: bool | None = False,
     style: str = "old",
@@ -542,6 +551,7 @@ def write_table(
             blue_border_row_ids=blue_border_row_ids,
             number_format=number_format,
             autofit_columns=autofit_columns,
+            column_widths=column_widths,
             min_column_width=min_column_width,
             col_filter=col_filter,
             combine_multiindex=combine_multiindex,
@@ -590,7 +600,8 @@ def format_worksheet(
     blue_border: bool | None = True,
     blue_border_row_ids: int | list[int] | None = None,
     number_format: str = "0.0",
-    autofit_columns: str | None = "column_names",
+    autofit_columns: str | None = None,
+    column_widths: list[int] | None = None,
     min_column_width: int | None = None,
     col_filter: bool | None = False,
     combine_multiindex: bool | int = False,
@@ -692,7 +703,14 @@ def format_worksheet(
     if combine_multiindex:
         cells_to_merge = get_cells_to_merge(df)
 
-    if autofit_columns == "column_names":
+    if column_widths:
+        if len(arr[0]) != len(column_widths):
+            raise Warning(
+                "The number of widths defined in column_widths should be equal to the number of columsn in the dataframe"
+            )
+        col_widths = column_widths
+
+    elif (autofit_columns == "column_names") or (autofit_columns is None):
         col_widths = get_max_col_widths(df)
     elif autofit_columns == "all_data":
         col_widths = get_max_col_widths(arr)

{toolsos-0.2.6.dist-info → toolsos-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: toolsos
-Version: 0.2.6
+Version: 0.2.8
 Summary: OS tools
 Author-email: OS <d.schmitz@amsterdam.nl>
 Keywords: tools,Onderzoek & Statistiek

{toolsos-0.2.6.dist-info → toolsos-0.2.8.dist-info}/RECORD

@@ -2,7 +2,7 @@ toolsos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 toolsos/cbs_tools.py,sha256=361cogk0aIU4D4BKHaa7YSOBh64t5C3zrHlqtWx0iIc,3465
 toolsos/create_tables.py,sha256=43FHK3EERjumBtnGhngIdtthZzcc_Qi37lJ1MgATzBg,908
 toolsos/download.py,sha256=88hehmPL5m5d1nrcJjltuh4xrCItF5EYHaZdHOcSt-g,2652
-toolsos/geo.py,sha256=arapy_ol6_so8KZ5gJk9ywXysSz4W8ah-cjrJ3DuxAo,2419
+toolsos/geo.py,sha256=AUPwGCsIjS_JktiyK4D8LQUq1y-kXx_PUMWzScgkBc8,2851
 toolsos/helpers.py,sha256=VeOl-fLgePCbjEmAQdVmYe7z8OE1pISeDDuP1t5QSxM,997
 toolsos/polars_helpers.py,sha256=P3RHLQFeDL7-9U_Q1n4ma_NSkdYAiker4pnc57uluHw,770
 toolsos/database/database_connection.py,sha256=HCP8H_-iE2BOOQDp9v1HOgfUSX45XS-aqw7Nzf8drAU,4359
@@ -15,11 +15,12 @@ toolsos/huisstijl/graphs/graph_styles.py,sha256=Z9LLH7j8ODTsYMYK0rslacphuiRDcq5_
 toolsos/huisstijl/graphs/linegraph.py,sha256=dMUarRe31SXaY78OCXLy-PgnU8LlVJ9KkzKaHhDtuuI,698
 toolsos/huisstijl/graphs/piegraph.py,sha256=aEFiEM-9QuhBOjKHSXVuE5bTh-8uucq4FP6O8Vk1vZI,703
 toolsos/huisstijl/graphs/styler.py,sha256=-uZ7pjY1G39XvmaGHQd31gPRxjxmJGhYZk8xhy2JUWc,6623
+toolsos/huisstijl/maps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 toolsos/huisstijl/tables/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 toolsos/huisstijl/tables/table_helpers.py,sha256=jsQ6lw93sxtGJGrUn8X2_LyA2vYYnytngpUI5A_wpWQ,2037
 toolsos/huisstijl/tables/table_styles.py,sha256=oYU6GJcfqlKpZof5PUjPsA7woJ3Tew78CHPyT0_jY6w,1343
-toolsos/huisstijl/tables/tables.py,sha256=2FO-ByLjgs-DbNgem3cDfYJbLbIDzRDqXtjL75WN7kY,24054
-toolsos-0.2.6.dist-info/METADATA,sha256=Fm7Ng8JDYMiDD3Ccm5AYGlGrl3QPZGmRMHC6oNA5o5o,2683
-toolsos-0.2.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-toolsos-0.2.6.dist-info/top_level.txt,sha256=2ClEjUBbtfDQ8oPwvWRy1Sz2nrkLCXlg0mHaMdCWia0,8
-toolsos-0.2.6.dist-info/RECORD,,
+toolsos/huisstijl/tables/tables.py,sha256=yeHiZd4EuOoeb_CeRE74xM1r1j6gCMNM1-I3Lpd2tyQ,24728
+toolsos-0.2.8.dist-info/METADATA,sha256=1copLfCg6pko0yR08KsSLxsVKiXmxS6iLEV-oMghV7I,2683
+toolsos-0.2.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+toolsos-0.2.8.dist-info/top_level.txt,sha256=2ClEjUBbtfDQ8oPwvWRy1Sz2nrkLCXlg0mHaMdCWia0,8
+toolsos-0.2.8.dist-info/RECORD,,