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

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

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: ssb-konjunk
-Version: 0.1.16
+Version: 0.1.18
 Summary: SSB Konjunk
-Home-page: https://github.com/statisticsnorway/ssb-konjunk
 License: MIT
 Author: Edvard Garmannslund
 Author-email: ged@ssb.no
@@ -15,12 +14,13 @@ 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)
-Requires-Dist: pandas-stubs (>=2.2.2.240807,<3.0.0.0)
-Requires-Dist: pendulum (>=3.0.0,<4.0.0)
+Requires-Dist: dapla-toolbelt (>=2.0.20)
+Requires-Dist: pandas (>=2.2.0)
+Requires-Dist: pandas-stubs (>=2.2.2.240807)
+Requires-Dist: pendulum (>=3.0.0)
 Project-URL: Changelog, https://github.com/statisticsnorway/ssb-konjunk/releases
 Project-URL: Documentation, https://statisticsnorway.github.io/ssb-konjunk
+Project-URL: Homepage, https://github.com/statisticsnorway/ssb-konjunk
 Project-URL: Repository, https://github.com/statisticsnorway/ssb-konjunk
 Description-Content-Type: text/markdown
 

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ssb-konjunk"
-version = "0.1.16"
+version = "0.1.18"
 description = "SSB Konjunk"
 authors = ["Edvard Garmannslund <ged@ssb.no>"]
 license = "MIT"
@@ -14,12 +14,12 @@ classifiers = ["Development Status :: 1 - Planning"]
 Changelog = "https://github.com/statisticsnorway/ssb-konjunk/releases"
 
 [tool.poetry.dependencies]
-python = "^3.10"
+python = ">=3.10, <4.0"
 click = ">=8.0.1"
-pandas = "^2.2.0"
-pendulum = "^3.0.0"
-dapla-toolbelt = "^2.0.19"
-pandas-stubs = "^2.2.2.240807"
+pandas = ">=2.2.0"
+pendulum = ">=3.0.0"
+dapla-toolbelt = ">=2.0.20"
+pandas-stubs = ">=2.2.2.240807"
 
 [tool.poetry.group.dev.dependencies]
 pygments = ">=2.10.0"
@@ -71,7 +71,7 @@ show_error_context = true
 force-exclude = true  # Apply excludes to pre-commit
 show-fixes = true
 src = ["src", "tests"]
-target-version = "py310"  # Minimum Python version supported
+target-version = "py311"  # Minimum Python version supported
 include = ["*.py", "*.pyi", "**/pyproject.toml", "*.ipynb"]
 extend-exclude = [
     "__pycache__",
@@ -99,8 +99,11 @@ ignore = [
     "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.
+    "FBT001", # Allow boolean positional arguments in a function.
+    "FBT002", # Allow boolean default positional arguments in a function.
     "E402",   # Supress module-import-not-at-top-of-file, needed in jupyter notebooks.
     "E501",   # Supress line-too-long warnings: trust black's judgement on this one.
+    "PLR2004", # Allow to compare with unnamed numerical constants.
 ]
 
 [tool.ruff.lint.isort]
@@ -112,6 +115,9 @@ max-complexity = 15
 [tool.ruff.lint.pydocstyle]
 convention = "google"  # You can also use "numpy".
 
+[tool.ruff.lint.pylint]
+max-args = 8
+
 [tool.ruff.lint.pep8-naming]
 classmethod-decorators = ["classmethod", "validator", "root_validator", "pydantic.validator"]
 

{ssb_konjunk-0.1.16 → ssb_konjunk-0.1.18}/src/ssb_konjunk/prompts.py

@@ -272,6 +272,60 @@ def validate_day(day: int | str) -> str:
     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.