feat: helpers to calculate prior (local) month range

by default use the America/Los_Angeles time zone

override with TZ_NAME env var

Author: thekaveman

.env.sample

@@ -8,3 +8,5 @@ TOGGL_USER_INFO=data/toggl-user-info-sample.json
 TOGGL_API_TOKEN=token
 TOGGL_CLIENT_ID=client
 TOGGL_WORKSPACE_ID=workspace
+
+TZ_NAME=America/Los_Angeles

compiler_admin/main.py

@@ -1,5 +1,8 @@
 from argparse import ArgumentParser, _SubParsersAction
+from datetime import datetime, timedelta
+import os
 import sys
+from zoneinfo import ZoneInfo
 
 from compiler_admin import __version__ as version
 from compiler_admin.commands.info import info
@@ -9,6 +12,24 @@
 from compiler_admin.commands.user.convert import ACCOUNT_TYPE_OU
 
 
+TZINFO = ZoneInfo(os.environ.get("TZ_NAME", "America/Los_Angeles"))
+
+
+def local_now():
+    return datetime.now(tz=TZINFO)
+
+
+def prior_month_end():
+    now = local_now()
+    first = now.replace(day=1)
+    return first - timedelta(days=1)
+
+
+def prior_month_start():
+    end = prior_month_end()
+    return end.replace(day=1)
+
+
 def add_sub_cmd_parser(parser: ArgumentParser, dest="subcommand", help=None):
     """Helper adds a subparser for the given dest."""
     return parser.add_subparsers(dest=dest, help=help)

pyproject.toml

@@ -13,6 +13,7 @@ requires-python = ">=3.11"
 dependencies = [
     "advanced-gam-for-google-workspace @ git+https://github.com/taers232c/[email protected]#subdirectory=src",
     "pandas==2.2.3",
+    "tzdata",
 ]
 
 [project.urls]

tests/test_main.py

@@ -1,14 +1,32 @@
 from argparse import Namespace
+from datetime import datetime
 import subprocess
 import sys
 
 import pytest
 
 import compiler_admin.main
-from compiler_admin.main import main, __name__ as MODULE
+from compiler_admin.main import main, prior_month_start, prior_month_end, TZINFO, __name__ as MODULE
 from compiler_admin.services.google import DOMAIN
 
 
+@pytest.fixture
+def mock_local_now(mocker):
+    dt = datetime(2024, 9, 25, tzinfo=TZINFO)
+    mocker.patch(f"{MODULE}.local_now", return_value=dt)
+    return dt
+
+
+@pytest.fixture
+def mock_start(mock_local_now):
+    return datetime(2024, 8, 1, tzinfo=TZINFO)
+
+
+@pytest.fixture
+def mock_end(mock_local_now):
+    return datetime(2024, 8, 31, tzinfo=TZINFO)
+
+
 @pytest.fixture
 def mock_commands_info(mock_commands_info):
     return mock_commands_info(MODULE)
@@ -29,6 +47,18 @@ def mock_commands_user(mock_commands_user):
     return mock_commands_user(MODULE)
 
 
+def test_prior_month_start(mock_start):
+    start = prior_month_start()
+
+    assert start == mock_start
+
+
+def test_prior_month_end(mock_end):
+    end = prior_month_end()
+
+    assert end == mock_end
+
+
 def test_main_info(mock_commands_info):
     main(argv=["info"])