from __future__ import annotations

from typing import Optional

from datetime import date

from domain import model

from domain.model import OrderLine

from adapters.repository import AbstractRepository

class InvalidSku(Exception):

pass

def is_valid_sku(sku, batches):

return sku in {b.sku for b in batches}

def add_batch(

ref: str, sku: str, qty: int, eta: Optional[date],

repo: AbstractRepository, session,

) -> None:

repo.add(model.Batch(ref, sku, qty, eta))

session.commit()

def allocate(

orderid: str, sku: str, qty: int,

repo: AbstractRepository, session

) -> Optional[str]:

line = OrderLine(orderid, sku, qty)

batches = repo.list()

if not is_valid_sku(line.sku, batches):

raise InvalidSku(f"Invalid sku {line.sku}")

updated_batch = model.allocate(line, batches)

session.commit()

return updated_batch.ref if updated_batch else None