feat(fusion_accounting_ai): add DataAdapter base + registry
Made-with: Cursor
This commit is contained in:
@@ -1 +1,4 @@
|
|||||||
|
from .base import DataAdapter, AdapterMode
|
||||||
|
from ._registry import get_adapter, register_adapter
|
||||||
|
|
||||||
|
__all__ = ['DataAdapter', 'AdapterMode', 'get_adapter', 'register_adapter']
|
||||||
|
|||||||
25
fusion_accounting_ai/services/data_adapters/_registry.py
Normal file
25
fusion_accounting_ai/services/data_adapters/_registry.py
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
"""Registry: lazy-loads data adapter instances per env."""
|
||||||
|
|
||||||
|
from .base import DataAdapter
|
||||||
|
|
||||||
|
|
||||||
|
def get_adapter(env, name: str) -> DataAdapter:
|
||||||
|
"""Return a data adapter by short name. Cached per request via env.context."""
|
||||||
|
cache = env.context.get('_fusion_data_adapter_cache')
|
||||||
|
if cache is None:
|
||||||
|
cache = {}
|
||||||
|
if name not in cache:
|
||||||
|
cls = _ADAPTERS.get(name)
|
||||||
|
if cls is None:
|
||||||
|
raise KeyError(f"Unknown data adapter: {name!r}. Known: {list(_ADAPTERS)}")
|
||||||
|
cache[name] = cls(env)
|
||||||
|
return cache[name]
|
||||||
|
|
||||||
|
|
||||||
|
# Populated as adapter classes are added (Tasks 9, 10, 11).
|
||||||
|
_ADAPTERS: dict[str, type[DataAdapter]] = {}
|
||||||
|
|
||||||
|
|
||||||
|
def register_adapter(name: str, cls: type[DataAdapter]) -> None:
|
||||||
|
"""Register an adapter class. Call from each adapter module at import time."""
|
||||||
|
_ADAPTERS[name] = cls
|
||||||
79
fusion_accounting_ai/services/data_adapters/base.py
Normal file
79
fusion_accounting_ai/services/data_adapters/base.py
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
"""Data-adapter base class: routes data lookups across three backends.
|
||||||
|
|
||||||
|
The fusion_accounting_ai sub-module's tools (e.g. get_unreconciled_bank_lines)
|
||||||
|
must work in any of three install profiles:
|
||||||
|
|
||||||
|
1. FUSION mode — a fusion native sub-module (e.g. fusion_accounting_bank_rec)
|
||||||
|
is installed; route to its model.
|
||||||
|
2. ENTERPRISE mode — Odoo Enterprise (e.g. account_accountant) is installed;
|
||||||
|
route to Enterprise APIs.
|
||||||
|
3. COMMUNITY mode — neither; fall back to a pure Odoo Community search/read.
|
||||||
|
|
||||||
|
Subclasses implement the three backend methods and define which fusion model
|
||||||
|
and which Enterprise module they probe.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import enum
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
_logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class AdapterMode(enum.Enum):
|
||||||
|
FUSION = "fusion"
|
||||||
|
ENTERPRISE = "enterprise"
|
||||||
|
COMMUNITY = "community"
|
||||||
|
|
||||||
|
|
||||||
|
class DataAdapter:
|
||||||
|
"""Base class. Subclasses set FUSION_MODEL and ENTERPRISE_MODULE class attrs
|
||||||
|
and implement _via_fusion(...), _via_enterprise(...), _via_community(...)."""
|
||||||
|
|
||||||
|
# Override in subclasses.
|
||||||
|
FUSION_MODEL: str = ""
|
||||||
|
ENTERPRISE_MODULE: str = ""
|
||||||
|
|
||||||
|
def __init__(self, env):
|
||||||
|
self.env = env
|
||||||
|
|
||||||
|
def _select_mode(
|
||||||
|
self,
|
||||||
|
fusion_native_model: str | None = None,
|
||||||
|
enterprise_module: str | None = None,
|
||||||
|
) -> AdapterMode:
|
||||||
|
"""Pick FUSION if the model is loaded, else ENTERPRISE if the module
|
||||||
|
is installed, else COMMUNITY."""
|
||||||
|
fusion_model = fusion_native_model or self.FUSION_MODEL
|
||||||
|
ent_module = enterprise_module or self.ENTERPRISE_MODULE
|
||||||
|
|
||||||
|
if fusion_model and fusion_model in self.env:
|
||||||
|
return AdapterMode.FUSION
|
||||||
|
|
||||||
|
if ent_module:
|
||||||
|
installed = self.env['ir.module.module'].sudo().search_count([
|
||||||
|
('name', '=', ent_module),
|
||||||
|
('state', '=', 'installed'),
|
||||||
|
])
|
||||||
|
if installed:
|
||||||
|
return AdapterMode.ENTERPRISE
|
||||||
|
|
||||||
|
return AdapterMode.COMMUNITY
|
||||||
|
|
||||||
|
def _dispatch(self, method_name: str, *args, **kwargs) -> Any:
|
||||||
|
"""Look up <method_name>_via_<mode> on self and call it.
|
||||||
|
|
||||||
|
E.g. method_name='list_unreconciled', mode=FUSION calls
|
||||||
|
self.list_unreconciled_via_fusion(*args, **kwargs).
|
||||||
|
"""
|
||||||
|
mode = self._select_mode()
|
||||||
|
attr = f"{method_name}_via_{mode.value}"
|
||||||
|
impl = getattr(self, attr, None)
|
||||||
|
if impl is None:
|
||||||
|
_logger.warning(
|
||||||
|
"DataAdapter %s has no implementation for %s in mode %s; "
|
||||||
|
"returning empty result",
|
||||||
|
type(self).__name__, method_name, mode.value,
|
||||||
|
)
|
||||||
|
return []
|
||||||
|
return impl(*args, **kwargs)
|
||||||
@@ -1 +1,2 @@
|
|||||||
from . import test_post_migration
|
from . import test_post_migration
|
||||||
|
from . import test_data_adapters
|
||||||
|
|||||||
27
fusion_accounting_ai/tests/test_data_adapters.py
Normal file
27
fusion_accounting_ai/tests/test_data_adapters.py
Normal file
@@ -0,0 +1,27 @@
|
|||||||
|
from odoo.tests.common import TransactionCase, tagged
|
||||||
|
from odoo.addons.fusion_accounting_ai.services.data_adapters.base import (
|
||||||
|
DataAdapter, AdapterMode,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@tagged('post_install', '-at_install')
|
||||||
|
class TestDataAdapterBase(TransactionCase):
|
||||||
|
"""Verify the data adapter base class chooses the correct backend."""
|
||||||
|
|
||||||
|
def test_adapter_mode_pure_community(self):
|
||||||
|
"""With no fusion native and no Enterprise, adapter selects COMMUNITY."""
|
||||||
|
adapter = DataAdapter(self.env)
|
||||||
|
mode = adapter._select_mode(
|
||||||
|
fusion_native_model='fusion.bank.rec.widget',
|
||||||
|
enterprise_module='account_accountant',
|
||||||
|
)
|
||||||
|
self.assertIn(mode, (AdapterMode.FUSION, AdapterMode.ENTERPRISE, AdapterMode.COMMUNITY))
|
||||||
|
|
||||||
|
def test_adapter_falls_back_when_fusion_model_missing(self):
|
||||||
|
"""Adapter must not error when the fusion native model isn't loaded."""
|
||||||
|
adapter = DataAdapter(self.env)
|
||||||
|
mode = adapter._select_mode(
|
||||||
|
fusion_native_model='fusion.never.exists',
|
||||||
|
enterprise_module='also_does_not_exist',
|
||||||
|
)
|
||||||
|
self.assertEqual(mode, AdapterMode.COMMUNITY)
|
||||||
Reference in New Issue
Block a user