import datetime

from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from ..core.security import get_current_active_user as get_user
from ..db.session import SessionLocal
from ..models.voucher import Batch
from ..schemas.auth import UserToken


router = APIRouter()


# Dependency
def get_db() -> Session:
    try:
        db = SessionLocal()
        yield db
    finally:
        db.close()


@router.get("")
def batch_term(
    q: str,
    c: int = None,
    d: str = None,
    db: Session = Depends(get_db),
    current_user: UserToken = Depends(get_user),
):
    filter_ = q if q is not None and q.strip() != "" else None
    date = None if not d else datetime.datetime.strptime(d, "%d-%b-%Y")
    list_ = []
    for index, item in enumerate(
        Batch.list(filter_, include_nil=False, date=date, db=db)
    ):
        text = f"{item.product.name} ({item.product.units}) {item.quantity_remaining:.2f}@{item.rate:.2f} from {item.name.strftime('%d-%b-%Y')}"
        list_.append(
            {
                "id": item.id,
                "name": text,
                "quantityRemaining": item.quantity_remaining,
                "rate": item.rate,
                "tax": item.tax,
                "discount": item.discount,
                "product": {
                    "id": item.product.id,
                    "name": item.product.name,
                    "units": item.product.units,
                },
            }
        )
        if c is not None and index == c - 1:
            break
    return list_