from typing import Optional
from sqlalchemy import desc
from db import db
from models import StreamRate, Territory


class ReportingRepository:
    def get_revenue_stream_rates_by_country(self, country_id: int) -> Optional[StreamRate]:
        return (
            db.session.query(StreamRate)
            .join(Territory, Territory.id == StreamRate.territory_id)
            .filter(Territory.id == country_id)
            .order_by(desc(StreamRate.imported_date))
            .one_or_none()
        )