mindbridge-api-python-client 1.6.0__tar.gz → 1.6.1__tar.gz

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/PKG-INFO

@@ -1,32 +1,30 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: mindbridge-api-python-client
-Version: 1.6.0
+Version: 1.6.1
 Summary: Interact with the MindBridge API
-Home-page: https://www.mindbridge.ai
-License: Proprietary
+License: LicenseRef-Proprietary
+Keywords: MindBridge
 Author: MBAI Support
 Author-email: mbaisupport@mindbridge.ai
 Maintainer: MBAI Support
 Maintainer-email: mbaisupport@mindbridge.ai
-Requires-Python: >=3.9,<4.0
+Requires-Python: >= 3.9
 Classifier: Development Status :: 4 - Beta
-Classifier: Framework :: Flake8
 Classifier: Framework :: Pydantic
 Classifier: Framework :: Pydantic :: 2
 Classifier: Framework :: Pytest
 Classifier: License :: Other/Proprietary License
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Typing :: Typed
 Requires-Dist: pydantic (>=2.5.2,<3.0.0)
-Requires-Dist: urllib3 (>=2.0.7,<3.0.0)
+Requires-Dist: urllib3 (>=2.3.0,<3.0.0)
+Project-URL: Homepage, https://www.mindbridge.ai
 Description-Content-Type: text/markdown
 
 <h1 align="center">MindBridge API Python Client</h1>

mindbridge_api_python_client-1.6.1/pyproject.toml

@@ -0,0 +1,148 @@
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[project]
+name = "mindbridge-api-python-client"
+version = "1.6.1"
+description = "Interact with the MindBridge API"
+readme = "README.md"
+license = "LicenseRef-Proprietary"
+license-files = ["LICENSE.txt"]
+authors = [
+    {name = "MBAI Support", email = "mbaisupport@mindbridge.ai"}, # Must be first so that PyPI displays this
+    {name = "Edgar Silva", email = "edgar.silva@mindbridge.ai"},
+    {name = "Jordan Hatcher", email = "jordan.hatcher@mindbridge.ai"},
+    {name = "Kevin Paulson", email = "kevin.paulson@mindbridge.ai"},
+    {name = "Michael Smith", email = "msmith@mindbridge.ai"},
+    {name = "Owen Allen", email = "owen.allen@mindbridge.ai"},
+    {name = "Wing-Leung Chan", email = "wing-leung.chan@mindbridge.ai"},
+]
+maintainers = [
+    {name = "MBAI Support", email = "mbaisupport@mindbridge.ai"},
+]
+dependencies = [
+  "pydantic (>=2.5.2,<3.0.0)",
+  "urllib3 (>=2.3.0,<3.0.0)",
+]
+requires-python = ">= 3.9"
+keywords = [ "MindBridge" ]
+dynamic = [ "classifiers" ]
+
+[project.urls]
+homepage = "https://www.mindbridge.ai"
+
+[tool.poetry]
+requires-poetry = ">=2.0"
+classifiers = [ # Python classifiers are automatically added by poetry
+    "Development Status :: 4 - Beta",
+    "Framework :: Pydantic",
+    "Framework :: Pydantic :: 2",
+    "Framework :: Pytest",
+    "License :: Other/Proprietary License",
+    "Operating System :: OS Independent",
+    "Typing :: Typed",
+]
+packages = [
+    { include = "mindbridgeapi", from = "src" },
+]
+
+[tool.poetry.group.lint]
+optional = true
+
+[tool.poetry.group.lint.dependencies]
+ruff = "^0.9.6"
+
+[tool.poetry.group.test]
+optional = true
+
+[tool.poetry.group.test.dependencies]
+pytest = "^8.1.1"
+pytest-randomly = "^3.15.0"
+openpyxl = "^3.1.5"
+pandas = "^2.2.3"
+datamodel-code-generator = "^0.28.0"
+mypy = "^1.13.0"
+pytest-cov = "^6.0.0"
+pytest-html = "^4.1.1"
+
+[tool.ruff]
+extend-exclude = [
+    "src/mindbridgeapi/generated_pydantic_model/model.py",
+    ".cci_pycache",
+]
+
+[tool.ruff.format]
+line-ending = "lf"
+
+[tool.ruff.lint]
+select = [
+    "I", # isort
+    "F", # Pyflakes
+    "E", "W", # pycodestyle
+    "N", # pep8-naming
+    "B", # flake8-bugbear
+    "C4", # flake8-comprehensions
+    "ISC", # flake8-implicit-str-concat
+    "PT", # flake8-pytest-style
+    "RET", # flake8-return
+    "SIM", # flake8-simplify
+    "TID", # flake8-tidy-imports
+    "TRY", # tryceratops
+    "TC", # flake8-type-checking
+    "RUF100",
+    # flake8-clean-block = "^0.1.2"
+    # flake8-encodings = "^0.5.1" https://github.com/astral-sh/ruff/issues/3278
+]
+ignore = [
+    "E111", "E114", "E117", # Conflicting lint rules with formatter
+    "SIM105", # Did not seem useful
+    "TRY003", # Did not seem useful
+]
+
+[tool.ruff.lint.flake8-tidy-imports]
+ban-relative-imports = "all"
+
+[tool.ruff.lint.isort]
+combine-as-imports = true
+force-sort-within-sections = true
+no-lines-before = ["future", "standard-library", "first-party", "local-folder", "third-party"]
+
+[tool.pytest.ini_options]
+addopts = [
+    "--strict-config",
+    "--strict-markers",
+    "--cov=mindbridgeapi",
+    "--cov=generate_model",
+    "--cov=set_circleci_pytest_enviroment_variables",
+    "--cov-report=html",
+    "--cov-report=term",
+    "--cov-report=xml",
+]
+
+xfail_strict = true
+filterwarnings = [ "error" ]
+pythonpath = [ "tools" ]
+
+[tool.coverage.run]
+branch = true
+omit = [ "src/mindbridgeapi/generated_pydantic_model/*" ]
+
+[tool.coverage.report]
+exclude_also = [
+    "if TYPE_CHECKING:",
+    'if __name__ == "__main__":',
+]
+
+[tool.mypy]
+mypy_path = "$MYPY_CONFIG_FILE_DIR/src"
+packages = "mindbridgeapi,tools.generate_model,tools.set_circleci_pytest_enviroment_variables"
+plugins = [ "pydantic.mypy" ]
+strict = true
+no_implicit_optional = true
+show_error_codes = true
+
+[tool.pydantic-mypy]
+init_forbid_extra = true
+init_typed = true
+warn_required_dynamic_aliases = true

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/__init__.py

@@ -21,10 +21,10 @@ from mindbridgeapi.enumerations.analysis_type import AnalysisType
 from mindbridgeapi.enumerations.system_library import SystemLibrary
 from mindbridgeapi.file_manager_item import FileManagerItem, FileManagerType
 from mindbridgeapi.generated_pydantic_model.model import (
+    Frequency,
     PeriodType as AnalysisEffectiveDateMetricsPeriod,
+    TargetWorkflowState,
 )
-from mindbridgeapi.generated_pydantic_model.model import Frequency
-from mindbridgeapi.generated_pydantic_model.model import TargetWorkflowState
 from mindbridgeapi.library_item import LibraryItem
 from mindbridgeapi.organization_item import OrganizationItem
 from mindbridgeapi.server import Server

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/account_grouping_item.py

@@ -17,8 +17,6 @@ from mindbridgeapi.common_validators import (
 from mindbridgeapi.generated_pydantic_model.model import (
     ApiAccountGroupingRead,
     ApiAccountGroupingUpdate,
-)
-from mindbridgeapi.generated_pydantic_model.model import (
     Type7 as _AccountGroupingFileType,
 )
 

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/account_groupings.py

@@ -8,7 +8,6 @@
 
 from dataclasses import dataclass
 from functools import cached_property
-import logging
 from typing import TYPE_CHECKING, Any, Optional, Union
 from mindbridgeapi.account_grouping_item import (
     AccountGroupingFileType,
@@ -28,8 +27,6 @@ if TYPE_CHECKING:
     from os import PathLike
     from pathlib import Path
 
-logger = logging.getLogger(__name__)
-
 
 @dataclass
 class AccountGroupings(BaseSet):

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/account_groups.py

@@ -8,7 +8,6 @@
 
 from dataclasses import dataclass
 from functools import cached_property
-import logging
 from typing import TYPE_CHECKING, Any, Optional
 from mindbridgeapi.account_group_item import AccountGroupItem
 from mindbridgeapi.base_set import BaseSet
@@ -18,8 +17,6 @@ from mindbridgeapi.exceptions import ItemNotFoundError
 if TYPE_CHECKING:
     from collections.abc import Generator
 
-logger = logging.getLogger(__name__)
-
 
 @dataclass
 class AccountGroups(BaseSet):

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/account_mappings.py

@@ -8,7 +8,6 @@
 
 from dataclasses import dataclass
 from functools import cached_property
-import logging
 from typing import TYPE_CHECKING, Any, Optional
 from mindbridgeapi.account_mapping_item import AccountMappingItem
 from mindbridgeapi.base_set import BaseSet
@@ -18,8 +17,6 @@ from mindbridgeapi.exceptions import ItemNotFoundError
 if TYPE_CHECKING:
     from collections.abc import Generator
 
-logger = logging.getLogger(__name__)
-
 
 @dataclass
 class AccountMappings(BaseSet):

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/analysis_period.py

@@ -40,12 +40,9 @@ class AnalysisPeriod(ApiAnalysisPeriodRead):
         if not isinstance(other, AnalysisPeriod):
             return NotImplemented
 
-        if (self.start_date > other.end_date) or (
+        return (self.start_date > other.end_date) or (
             (self.end_date == other.end_date) and (self.start_date > other.start_date)
-        ):
-            return True
-
-        return False
+        )
 
     def __le__(self, other: object) -> bool:
         return NotImplemented

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/analysis_source_item.py

@@ -92,11 +92,11 @@ class _ApiAnalysisSourceCreateOnly(ApiAnalysisSourceCreateOnly):
         appropriate virtual column type.
     """
 
-    proposed_virtual_columns: Optional[
-        list[_VirtualColumn]
-    ] = Field().merge_field_infos(
-        ApiAnalysisSourceCreateOnly.model_fields["proposed_virtual_columns"]
-    )  # type: ignore[assignment]
+    proposed_virtual_columns: Optional[list[_VirtualColumn]] = (
+        Field().merge_field_infos(  # type: ignore[assignment]
+            ApiAnalysisSourceCreateOnly.model_fields["proposed_virtual_columns"]
+        )
+    )
 
 
 class _ApiAnalysisSourceUpdate(ApiAnalysisSourceUpdate):
@@ -106,11 +106,11 @@ class _ApiAnalysisSourceUpdate(ApiAnalysisSourceUpdate):
         determine the appropriate virtual column type.
     """
 
-    proposed_virtual_columns: Optional[
-        list[_VirtualColumn]
-    ] = Field().merge_field_infos(
-        ApiAnalysisSourceUpdate.model_fields["proposed_virtual_columns"]
-    )  # type: ignore[assignment]
+    proposed_virtual_columns: Optional[list[_VirtualColumn]] = (
+        Field().merge_field_infos(  # type: ignore[assignment]
+            ApiAnalysisSourceUpdate.model_fields["proposed_virtual_columns"]
+        )
+    )
     virtual_columns: Optional[list[_VirtualColumn]] = Field().merge_field_infos(
         ApiAnalysisSourceUpdate.model_fields["virtual_columns"]
     )  # type: ignore[assignment]
@@ -127,11 +127,11 @@ class AnalysisSourceItem(ApiAnalysisSourceRead):
         ApiAnalysisSourceRead.model_fields["analysis_source_type_id"],
         default=AnalysisSourceTypeItem.GENERAL_LEDGER_JOURNAL,
     )
-    proposed_virtual_columns: Optional[
-        list[_VirtualColumn]
-    ] = Field().merge_field_infos(
-        ApiAnalysisSourceRead.model_fields["proposed_virtual_columns"]
-    )  # type: ignore[assignment]
+    proposed_virtual_columns: Optional[list[_VirtualColumn]] = (
+        Field().merge_field_infos(  # type: ignore[assignment]
+            ApiAnalysisSourceRead.model_fields["proposed_virtual_columns"]
+        )
+    )
     virtual_columns: Optional[list[_VirtualColumn]] = Field().merge_field_infos(
         ApiAnalysisSourceRead.model_fields["virtual_columns"]
     )  # type: ignore[assignment]

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/analysis_sources.py

@@ -21,8 +21,6 @@ from mindbridgeapi.exceptions import (
 )
 from mindbridgeapi.generated_pydantic_model.model import (
     ApiEffectiveDateMetricsRead as AnalysisEffectiveDateMetrics,
-)
-from mindbridgeapi.generated_pydantic_model.model import (
     PeriodType as AnalysisEffectiveDateMetricsPeriod,
 )
 
@@ -80,7 +78,9 @@ class AnalysisSources(BaseSet):
     def effective_date_metrics(
         self,
         item: AnalysisSourceItem,
-        period_type: AnalysisEffectiveDateMetricsPeriod = AnalysisEffectiveDateMetricsPeriod.MONTH,  # noqa:B950
+        period_type: AnalysisEffectiveDateMetricsPeriod = (
+            AnalysisEffectiveDateMetricsPeriod.MONTH
+        ),
     ) -> AnalysisEffectiveDateMetrics:
         if getattr(item, "id", None) is None:
             raise ItemNotFoundError

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/async_result_item.py

@@ -13,9 +13,11 @@ from mindbridgeapi.common_validators import (
     _warning_if_extra_fields,
 )
 from mindbridgeapi.exceptions import ItemError, ItemNotFoundError, ValidationError
-from mindbridgeapi.generated_pydantic_model.model import ApiAsyncResult
-from mindbridgeapi.generated_pydantic_model.model import Status3 as _AsyncResultStatus
-from mindbridgeapi.generated_pydantic_model.model import Type1 as _AsyncResultType
+from mindbridgeapi.generated_pydantic_model.model import (
+    ApiAsyncResult,
+    Status3 as _AsyncResultStatus,
+    Type1 as _AsyncResultType,
+)
 
 logger = logging.getLogger(__name__)
 

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/base_set.py

@@ -13,6 +13,7 @@ import logging
 import shutil
 from typing import TYPE_CHECKING, Any, Optional
 from urllib.parse import urlencode
+from urllib3.util import Retry
 from mindbridgeapi.exceptions import UnexpectedServerError, ValidationError
 
 if TYPE_CHECKING:
@@ -61,7 +62,14 @@ class BaseSet:
         params = {"page": page_number}
         request_url = f"{url}?{urlencode(params)}"
 
-        resp = self.server.http.request("POST", request_url, json=json)
+        # Same as set for the PoolManager (http) in Server, but getting a page will
+        # always be considered to be idempotent so we can allow repeat post requests
+        method = "POST"
+        retries = Retry(
+            connect=3, read=3, redirect=0, other=0, allowed_methods={method}
+        )
+
+        resp = self.server.http.request(method, request_url, retries=retries, json=json)
         self._check_response(resp)
         resp_dict = self._response_as_dict(resp)
 

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/common_validators.py

@@ -11,9 +11,12 @@ from typing import Any, Optional, Union
 import warnings
 from pydantic import RootModel
 from mindbridgeapi.exceptions import ParameterError
-from mindbridgeapi.generated_pydantic_model.model import ApiUserInfo, ApiUserInfoRead
-from mindbridgeapi.generated_pydantic_model.model import ApiUserRead as UserItem
-from mindbridgeapi.generated_pydantic_model.model import Role
+from mindbridgeapi.generated_pydantic_model.model import (
+    ApiUserInfo,
+    ApiUserInfoRead,
+    ApiUserRead as UserItem,
+    Role,
+)
 
 MindBridgeQuery = RootModel[dict[str, Any]]
 

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/data_tables.py

@@ -8,7 +8,6 @@
 
 from dataclasses import dataclass
 from functools import cached_property
-import logging
 from typing import TYPE_CHECKING, Any, Optional
 import warnings
 from mindbridgeapi.async_result_item import AsyncResultItem, AsyncResultType
@@ -21,15 +20,13 @@ from mindbridgeapi.generated_pydantic_model.model import (
     ApiDataTableQuerySortOrder,
     ApiDataTableRead,
     Direction,
+    Type4 as DataTableColumnType,
 )
-from mindbridgeapi.generated_pydantic_model.model import Type4 as DataTableColumnType
 
 if TYPE_CHECKING:
     from collections.abc import Generator
     from pathlib import Path
 
-logger = logging.getLogger(__name__)
-
 
 @dataclass
 class DataTables(BaseSet):
@@ -83,7 +80,20 @@ class DataTables(BaseSet):
         input_item: ApiDataTableRead,
         sort_direction: Optional[Direction] = None,
         sort_field: Optional[str] = None,
-    ) -> ApiDataTableQuerySortOrder:
+    ) -> Optional[ApiDataTableQuerySortOrder]:
+        if sort_field is not None and not isinstance(sort_field, str):
+            raise ParameterError(
+                parameter_name="sort_field", details="Not provided as str."
+            )
+
+        if sort_field is None and input_item.logical_name == "gl_journal_lines":
+            sort_field = "rowid"
+        elif sort_field is None and input_item.logical_name == "gl_journal_tx":
+            sort_field = "txid"
+
+        if sort_field is None or sort_field == "":
+            return None
+
         if sort_direction is None:
             sort_direction = Direction.ASC
 
@@ -94,22 +104,6 @@ class DataTables(BaseSet):
                 parameter_name="sort_direction", details="Not a valid Direction."
             ) from err
 
-        if sort_field is not None and not isinstance(sort_field, str):
-            raise ParameterError(
-                parameter_name="sort_field", details="Not provided as str."
-            )
-
-        if sort_field is None and input_item.type == "flows_compact":
-            sort_field = "flow_id"
-        elif sort_field is None and input_item.logical_name == "gl_journal_lines":
-            sort_field = "rowid"
-        elif sort_field is None and input_item.logical_name == "gl_journal_tx":
-            sort_field = "txid"
-        elif sort_field is None or sort_field == "":
-            # Don't send a sort
-            sort_field = None
-            sort_direction = None
-
         return ApiDataTableQuerySortOrder(direction=sort_direction, field=sort_field)
 
     def export(
@@ -126,13 +120,15 @@ class DataTables(BaseSet):
 
         fields = self._export_get_fields(input_item=input_item, fields=fields)
 
-        sort = self._export_get_sort(
-            input_item=input_item, sort_direction=sort_direction, sort_field=sort_field
-        )
-
         url = f"{self.base_url}/{input_item.id}/export"
         data_table_export_request = ApiDataTableExportRequest(
-            fields=fields, limit=limit, sort=sort
+            fields=fields,
+            limit=limit,
+            sort=self._export_get_sort(
+                input_item=input_item,
+                sort_direction=sort_direction,
+                sort_field=sort_field,
+            ),
         )
 
         json = data_table_export_request.model_dump(

mindbridge_api_python_client-1.6.1/src/mindbridgeapi/engagement_account_group_item.py

@@ -0,0 +1,21 @@
+#
+#  Copyright MindBridge Analytics Inc. all rights reserved.
+#
+#  This material is confidential and may not be copied, distributed,
+#  reversed engineered, decompiled or otherwise disseminated without
+#  the prior written consent of MindBridge Analytics Inc.
+#
+
+from pydantic import ConfigDict, model_validator
+from mindbridgeapi.common_validators import _warning_if_extra_fields
+from mindbridgeapi.generated_pydantic_model.model import ApiEngagementAccountGroupRead
+
+
+class EngagementAccountGroupItem(ApiEngagementAccountGroupRead):
+    model_config = ConfigDict(
+        extra="allow",
+        validate_assignment=True,
+        validate_default=True,
+        validate_return=True,
+    )
+    _ = model_validator(mode="after")(_warning_if_extra_fields)

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/engagement_account_grouping_item.py

@@ -6,17 +6,34 @@
 #  the prior written consent of MindBridge Analytics Inc.
 #
 
-from pydantic import ConfigDict, field_validator, model_validator
+from collections.abc import Generator
+from pydantic import ConfigDict, Field, field_validator, model_validator
 from mindbridgeapi.common_validators import (
     _convert_userinfo_to_useritem,
     _warning_if_extra_fields,
 )
+from mindbridgeapi.engagement_account_group_item import EngagementAccountGroupItem
 from mindbridgeapi.generated_pydantic_model.model import (
     ApiEngagementAccountGroupingRead,
 )
 
 
+def _empty_engagement_account_groups() -> Generator[
+    EngagementAccountGroupItem, None, None
+]:
+    """Empty generator function
+
+    This returns an empty generator function, it's use is to ensure
+    engagement_account_groups is not None for the EngagementAccountGroupingItem class
+    """
+    yield from ()
+
+
 class EngagementAccountGroupingItem(ApiEngagementAccountGroupingRead):
+    engagement_account_groups: Generator[EngagementAccountGroupItem, None, None] = (
+        Field(default_factory=_empty_engagement_account_groups, exclude=True)
+    )
+
     model_config = ConfigDict(
         extra="allow",
         validate_assignment=True,

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/engagement_account_groupings.py

@@ -28,7 +28,11 @@ class EngagementAccountGroupings(BaseSet):
     def get_by_id(self, id: str) -> EngagementAccountGroupingItem:
         url = f"{self.base_url}/{id}"
         resp_dict = super()._get_by_id(url=url)
-        return EngagementAccountGroupingItem.model_validate(resp_dict)
+        engagement_account_grouping = EngagementAccountGroupingItem.model_validate(
+            resp_dict
+        )
+        self.restart_engagement_account_groups(engagement_account_grouping)
+        return engagement_account_grouping
 
     def get(
         self, json: Optional[dict[str, Any]] = None
@@ -38,7 +42,11 @@ class EngagementAccountGroupings(BaseSet):
 
         url = f"{self.base_url}/query"
         for resp_dict in super()._get(url=url, json=json):
-            yield EngagementAccountGroupingItem.model_validate(resp_dict)
+            engagement_account_grouping = EngagementAccountGroupingItem.model_validate(
+                resp_dict
+            )
+            self.restart_engagement_account_groups(engagement_account_grouping)
+            yield engagement_account_grouping
 
     def export(self, input_item: EngagementAccountGroupingItem) -> AsyncResultItem:
         if getattr(input_item, "id", None) is None:
@@ -90,3 +98,19 @@ class EngagementAccountGroupings(BaseSet):
         return self.server.file_results.export(
             file_result=file_result, output_file_path=output_file_path
         )
+
+    def restart_engagement_account_groups(
+        self, engagement_account_grouping: EngagementAccountGroupingItem
+    ) -> None:
+        if getattr(engagement_account_grouping, "id", None) is None:
+            raise ItemNotFoundError
+
+        engagement_account_grouping.engagement_account_groups = (
+            self.server.engagement_account_groups.get(
+                json={
+                    "engagementAccountGroupingId": {
+                        "$eq": engagement_account_grouping.id
+                    }
+                }
+            )
+        )

mindbridge_api_python_client-1.6.1/src/mindbridgeapi/engagement_account_groups.py

@@ -0,0 +1,41 @@
+#
+#  Copyright MindBridge Analytics Inc. all rights reserved.
+#
+#  This material is confidential and may not be copied, distributed,
+#  reversed engineered, decompiled or otherwise disseminated without
+#  the prior written consent of MindBridge Analytics Inc.
+#
+
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING, Any, Optional
+from mindbridgeapi.base_set import BaseSet
+from mindbridgeapi.common_validators import _convert_json_query
+from mindbridgeapi.engagement_account_group_item import EngagementAccountGroupItem
+
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
+
+@dataclass
+class EngagementAccountGroups(BaseSet):
+    base_url: str = field(init=False)
+
+    def __post_init__(self) -> None:
+        self.base_url = f"{self.server.base_url}/engagement-account-groups"
+
+    def get_by_id(self, id: str) -> EngagementAccountGroupItem:
+        url = f"{self.base_url}/{id}"
+        resp_dict = super()._get_by_id(url=url)
+
+        return EngagementAccountGroupItem.model_validate(resp_dict)
+
+    def get(
+        self, json: Optional[dict[str, Any]] = None
+    ) -> "Generator[EngagementAccountGroupItem, None, None]":
+        mb_query_dict = _convert_json_query(
+            json, required_key="engagementAccountGroupingId"
+        )
+
+        url = f"{self.base_url}/query"
+        for resp_dict in super()._get(url=url, json=mb_query_dict):
+            yield EngagementAccountGroupItem.model_validate(resp_dict)

{mindbridge_api_python_client-1.6.0 → mindbridge_api_python_client-1.6.1}/src/mindbridgeapi/engagement_item.py

@@ -45,9 +45,9 @@ def _empty_file_manager_items() -> Generator[FileManagerItem, None, None]:
     yield from ()
 
 
-def _empty_engagement_account_groupings() -> (
-    Generator[EngagementAccountGroupingItem, None, None]
-):
+def _empty_engagement_account_groupings() -> Generator[
+    EngagementAccountGroupingItem, None, None
+]:
     """Empty generator function
 
     This returns an empty generator function, it's use is to ensure