mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
62e762e962
`property` is an FE-owned table the backend only ever reads — every row read carries an id — so the autoincrement-PK `Optional[int]` idiom doesn't apply here. Make it `int` and drop the now-redundant None guard in get_many. (Contrast: solar_table keeps Optional id — the backend DOES insert those, so id is genuinely None pre-flush.) Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from sqlmodel import Session, col, select
|
|
|
|
from domain.property.properties import Properties
|
|
from domain.property.property import Property, PropertyIdentity
|
|
from infrastructure.postgres.property_table import PropertyRow
|
|
from repositories.epc.epc_repository import EpcRepository
|
|
from repositories.property.property_repository import PropertyRepository
|
|
|
|
|
|
class PropertyPostgresRepository(PropertyRepository):
|
|
"""Hydrates the Property aggregate from the FE-owned ``property`` row plus the
|
|
EPC slice (via an injected `EpcRepository`). Reads only from repos — no
|
|
external IO — so a hydrated Property is a pure function of repository state
|
|
(ADR-0003).
|
|
"""
|
|
|
|
def __init__(self, session: Session, epc_repo: EpcRepository) -> None:
|
|
self._session = session
|
|
self._epc_repo = epc_repo
|
|
|
|
def get(self, property_id: int) -> Property:
|
|
row = self._session.get(PropertyRow, property_id)
|
|
if row is None:
|
|
raise ValueError(f"property {property_id} not found")
|
|
identity = PropertyIdentity(
|
|
portfolio_id=row.portfolio_id,
|
|
postcode=row.postcode,
|
|
address=row.address,
|
|
uprn=row.uprn,
|
|
landlord_property_id=row.landlord_property_id,
|
|
)
|
|
return Property(
|
|
identity=identity,
|
|
epc=self._epc_repo.get_for_property(property_id),
|
|
)
|
|
|
|
def get_many(self, property_ids: list[int]) -> Properties:
|
|
if not property_ids:
|
|
return Properties([])
|
|
rows = self._session.exec(
|
|
select(PropertyRow).where(col(PropertyRow.id).in_(property_ids))
|
|
).all()
|
|
row_by_id = {row.id: row for row in rows}
|
|
epcs = self._epc_repo.get_for_properties(property_ids)
|
|
items: list[Property] = []
|
|
for property_id in property_ids:
|
|
row = row_by_id.get(property_id)
|
|
if row is None:
|
|
raise ValueError(f"property {property_id} not found")
|
|
items.append(
|
|
Property(
|
|
identity=PropertyIdentity(
|
|
portfolio_id=row.portfolio_id,
|
|
postcode=row.postcode,
|
|
address=row.address,
|
|
uprn=row.uprn,
|
|
landlord_property_id=row.landlord_property_id,
|
|
),
|
|
epc=epcs.get(property_id),
|
|
)
|
|
)
|
|
return Properties(items)
|