ssb-konjunk 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

ssb_konjunk/prompts.py

@@ -271,6 +271,59 @@ def validate_day(day: int | str) -> str:
         day = "0" + str(int(day))
     return str(day)
 
+  
+def quarter_for_month(month: str | int) -> int:
+    """Find corresponding quarter for a month.
+
+    Args:
+        month: Month to find corresponding quarter for.
+
+    Returns:
+        int: The corresponding quarter.
+
+    Raises:
+        ValueError: If invalid month
+    """
+    month = int(month)
+
+    if month < 1 or month > 12:
+        raise ValueError(f"Invalid month: {month}")
+
+    if month < 4:
+        return 1
+    elif month < 7:
+        return 2
+    elif month < 10:
+        return 3
+    else:
+        return 4
+
+
+def months_in_quarter(quarter: int | str) -> list[int]:
+    """Return the three months in the quarter.
+
+    Args:
+        quarter: the relevant quarter.
+
+    Returns:
+        list: a list with the months in the quarter.
+
+    Raises:
+        ValueError: If invalid quarter.
+    """
+    quarter = int(quarter)
+
+    if quarter < 1 or quarter > 4:
+        raise ValueError(f"Invalid quarter: {quarter}")
+
+    if quarter == 1:
+        return [1, 2, 3]
+    elif quarter == 2:
+        return [4, 5, 6]
+    elif quarter == 3:
+        return [7, 8, 9]
+    else:
+        return [10, 11, 12]
 
 def set_publishing_date() -> str:
     """Set the date for publication of tables.
@@ -355,4 +408,4 @@ def get_previous_month(year: str | int, month: str | int) -> list[int]:
         prev_month = 12
         prev_year = int(year) - 1
 
-    return [prev_year, prev_month]
+    return [prev_year, prev_month]

ssb_konjunk/statbank_format.py

@@ -1,5 +1,7 @@
 import pandas as pd
 
+from ssb_konjunk import prompts
+
 
 def format_time_period(
     df: pd.DataFrame,
@@ -23,7 +25,7 @@ def format_time_period(
     if quarter != "" and month == "":
         df[col_name] = f"{year}K{int(quarter)}"
     elif quarter == "" and month != "":
-        df[col_name] = f"{year}M{int(quarter)}"
+        df[col_name] = f"{year}M{prompts.validate_month(month)}"
     else:
         df[col_name] = f"{year}"
 

{ssb_konjunk-0.1.15.dist-info → ssb_konjunk-0.1.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ssb-konjunk
-Version: 0.1.15
+Version: 0.1.17
 Summary: SSB Konjunk
 Home-page: https://github.com/statisticsnorway/ssb-konjunk
 License: MIT

{ssb_konjunk-0.1.15.dist-info → ssb_konjunk-0.1.17.dist-info}/RECORD

@@ -3,15 +3,15 @@ ssb_konjunk/__main__.py,sha256=bqGEjbMKlKza1kQeyZ5NcjV5p5vsGpoQ81YhO8cGPvk,213
 ssb_konjunk/_functions.py,sha256=Dp39rEMqR0M43o4mCujoRzlSyT_rHYcK6VHHc9wSY8E,470
 ssb_konjunk/data_formating.py,sha256=7GcDS_wsHWzCNJg4wpj-zFRWYX0c1KAB4FcbxZ9Ez4A,627
 ssb_konjunk/fame.py,sha256=6Yw0D1kY20yOu49DHgwfSIAmiHVCERuMeIANg1lYu64,1619
-ssb_konjunk/prompts.py,sha256=DsXmM4JWY3-WGb6ebj819zhaJ01JT8IL1226yadk724,9164
+ssb_konjunk/prompts.py,sha256=htj9zQxr2vgAr3pkke9lGiNM3pkbOQ1W7FA4JJ0o6Tk,10269
 ssb_konjunk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ssb_konjunk/rounding.py,sha256=nbxYMvvFzoJOxwJBCA80OyaZGhhOHCxbndeYB271Uzk,1269
 ssb_konjunk/saving.py,sha256=3lwYNnwh5Z-wbWJfiqaXwvsvya3zwBrEEE_3XejjtNI,14567
-ssb_konjunk/statbank_format.py,sha256=FfYgpEx4gvEbR2UPdqW5B_pkyCrn09QLSwjbTepb7Vs,1419
+ssb_konjunk/statbank_format.py,sha256=cVXrq2qSuMk-FAiMdbIS8Cc8YgZUyTCKnsMdknMbfkM,1469
 ssb_konjunk/timestamp.py,sha256=XPB5JhZQdDgUPYYNVAFiMZGe8dYL8seb_k98G9CNtjY,9960
 ssb_konjunk/xml_handling.py,sha256=kvAZ699iOSxHCi2ksEGroFk60-ufsRW7b_m3j47A2gY,1726
-ssb_konjunk-0.1.15.dist-info/LICENSE,sha256=np3IfD5m0ZUofn_kVzDZqliozuiO6wrktw3LRPjyEiI,1073
-ssb_konjunk-0.1.15.dist-info/METADATA,sha256=LpXgcRtXxJqZgE-rtk0gXnLgBpEk98ClkzRFAs42kU4,3966
-ssb_konjunk-0.1.15.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-ssb_konjunk-0.1.15.dist-info/entry_points.txt,sha256=qU7y58sZiFKYonuEJTbV0MhhbAJtposy46crLp4TjAM,57
-ssb_konjunk-0.1.15.dist-info/RECORD,,
+ssb_konjunk-0.1.17.dist-info/LICENSE,sha256=np3IfD5m0ZUofn_kVzDZqliozuiO6wrktw3LRPjyEiI,1073
+ssb_konjunk-0.1.17.dist-info/METADATA,sha256=xSofBjnYyeK75w_2c1k4L427LjGvmAqp1BBdtzOu-UI,3966
+ssb_konjunk-0.1.17.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+ssb_konjunk-0.1.17.dist-info/entry_points.txt,sha256=qU7y58sZiFKYonuEJTbV0MhhbAJtposy46crLp4TjAM,57
+ssb_konjunk-0.1.17.dist-info/RECORD,,