ssb-konjunk 0.1.14__tar.gz → 0.1.16__tar.gz

{ssb_konjunk-0.1.14 → ssb_konjunk-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ssb-konjunk
-Version: 0.1.14
+Version: 0.1.16
 Summary: SSB Konjunk
 Home-page: https://github.com/statisticsnorway/ssb-konjunk
 License: MIT
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (>=8.0.1)
 Requires-Dist: dapla-toolbelt (>=2.0.19,<3.0.0)
 Requires-Dist: pandas (>=2.2.0,<3.0.0)

{ssb_konjunk-0.1.14 → ssb_konjunk-0.1.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ssb-konjunk"
-version = "0.1.14"
+version = "0.1.16"
 description = "SSB Konjunk"
 authors = ["Edvard Garmannslund <ged@ssb.no>"]
 license = "MIT"
@@ -96,8 +96,6 @@ select = [
     "RUF",  # the ruff developer's own rules
 ]
 ignore = [
-    "ANN101", # Supress missing-type-self.
-    "ANN102", # Supress missing-type-cls.
     "ANN202", # Don't requiere return type annotation for private functions.
     "ANN401", # Allow type annotation with type Any.
     "D100",   # Supress undocumented-public-module. Only doc of public api required.

{ssb_konjunk-0.1.14 → ssb_konjunk-0.1.16}/src/ssb_konjunk/statbank_format.py

@@ -1,8 +1,14 @@
 import pandas as pd
 
+from ssb_konjunk import prompts
+
 
 def format_time_period(
-    df: pd.DataFrame, year: int, quarter: int | str = "", col_name: str = "periode"
+    df: pd.DataFrame,
+    year: int,
+    quarter: int | str = "",
+    col_name: str = "periode",
+    month: int | str = "",
 ) -> pd.DataFrame:
     """Add column with time period.
 
@@ -11,12 +17,15 @@ def format_time_period(
         year: the year.
         quarter: optional, default ''.
         col_name: optional, default 'periode'. The name of the column for the time period.
+        month: optional, default ''.
 
     Returns:
         pd.DataFrame: dataframe with a column with time period.
     """
-    if quarter != "":
+    if quarter != "" and month == "":
         df[col_name] = f"{year}K{int(quarter)}"
+    elif quarter == "" and month != "":
+        df[col_name] = f"{year}M{prompts.validate_month(month)}"
     else:
         df[col_name] = f"{year}"