2086ecb5ea
ConnectWiseProvider implements PSAProvider with test_connection() that calls GET /system/info to verify credentials and connectivity. All other methods raise NotImplementedError with slice references for future work. Provider registry (get_provider_for_account) looks up the account's PsaConnection, decrypts stored credentials, and instantiates the correct provider. Currently supports connectwise only. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
"""Factory for instantiating PSA providers from stored connection data."""
|
|
from __future__ import annotations
|
|
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.psa_connection import PsaConnection
|
|
from app.services.psa.base import PSAProvider
|
|
from app.services.psa.encryption import decrypt_credentials
|
|
from app.services.psa.exceptions import PSAConnectionError
|
|
|
|
|
|
async def get_provider_for_account(
|
|
account_id: UUID, db: AsyncSession
|
|
) -> PSAProvider:
|
|
"""Look up account's PSA connection, decrypt credentials, instantiate provider."""
|
|
result = await db.execute(
|
|
select(PsaConnection).where(
|
|
PsaConnection.account_id == account_id,
|
|
PsaConnection.is_active.is_(True),
|
|
)
|
|
)
|
|
connection = result.scalar_one_or_none()
|
|
|
|
if not connection:
|
|
raise PSAConnectionError(
|
|
"No active PSA connection configured for this account.",
|
|
provider="unknown",
|
|
)
|
|
|
|
if connection.provider == "connectwise":
|
|
from app.services.psa.connectwise.client import ConnectWiseClient
|
|
from app.services.psa.connectwise.provider import ConnectWiseProvider
|
|
|
|
creds = decrypt_credentials(connection.credentials_encrypted)
|
|
client = ConnectWiseClient(
|
|
site_url=connection.site_url,
|
|
company_id=connection.company_id,
|
|
public_key=creds["public_key"],
|
|
private_key=creds["private_key"],
|
|
client_id=creds["client_id"],
|
|
)
|
|
return ConnectWiseProvider(client)
|
|
|
|
raise PSAConnectionError(
|
|
f"Unsupported PSA provider: {connection.provider}",
|
|
provider=connection.provider,
|
|
)
|