from sqlalchemy import select
from sqlalchemy.orm import Session

from app.models.entities import RechargeOrder


class PaymentsRepository:
    def __init__(self, db: Session) -> None:
        self.db = db

    def get_order_by_id(self, order_id: int) -> RechargeOrder | None:
        return self.db.scalar(select(RechargeOrder).where(RechargeOrder.id == order_id))

    def list_orders(self):
        return self.db.query(RechargeOrder).order_by(RechargeOrder.id.desc())