103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
from datetime import date, datetime
|
|
from typing import List
|
|
|
|
import brewman.schemas.reports as schemas
|
|
|
|
from fastapi import APIRouter, Depends, Request, Security
|
|
from sqlalchemy.orm import Session, joinedload_all
|
|
|
|
from ...core.security import get_current_active_user as get_user
|
|
from ...core.session import get_finish_date, get_start_date, set_period
|
|
from ...db.session import SessionLocal
|
|
from ...models.voucher import Journal, Voucher, VoucherType
|
|
from ...schemas.auth import UserToken
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
# Dependency
|
|
def get_db() -> Session:
|
|
try:
|
|
db = SessionLocal()
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@router.get("", response_model=schemas.Daybook)
|
|
def report_blank(
|
|
request: Request,
|
|
user: UserToken = Security(get_user, scopes=["daybook"]),
|
|
) -> schemas.Daybook:
|
|
return schemas.Daybook(
|
|
startDate=get_start_date(request.session),
|
|
finishDate=get_finish_date(request.session),
|
|
body=[],
|
|
)
|
|
|
|
|
|
@router.get("/{start}/{finish}", response_model=schemas.Daybook)
|
|
def report_data(
|
|
start: str,
|
|
finish: str,
|
|
request: Request,
|
|
db: Session = Depends(get_db),
|
|
user: UserToken = Security(get_user, scopes=["daybook"]),
|
|
) -> schemas.Daybook:
|
|
start_date = datetime.strptime(start, "%d-%b-%Y")
|
|
finish_date = datetime.strptime(finish, "%d-%b-%Y")
|
|
body = build_report(start_date, finish_date, db)
|
|
set_period(start, finish, request.session)
|
|
return schemas.Daybook(startDate=start_date, finishDate=finish_date, body=body)
|
|
|
|
|
|
def build_report(start_date: date, finish_date: date, db: Session) -> List[schemas.DaybookItem]:
|
|
body = []
|
|
|
|
query = (
|
|
db.query(Voucher)
|
|
.options(joinedload_all(Voucher.journals, Journal.account, innerjoin=True))
|
|
.filter(Voucher.date >= start_date)
|
|
.filter(Voucher.date <= finish_date)
|
|
.filter(Voucher.type != VoucherType.by_name("Issue").id)
|
|
.order_by(Voucher.date)
|
|
.order_by(Voucher.last_edit_date)
|
|
.all()
|
|
)
|
|
|
|
for voucher in query:
|
|
debit = 0
|
|
credit = 0
|
|
name_debit = ""
|
|
name_credit = ""
|
|
for journal in voucher.journals:
|
|
if journal.debit == 1:
|
|
debit += journal.amount
|
|
name_debit += "{0} / ".format(journal.account.name)
|
|
else:
|
|
credit += journal.amount
|
|
name_credit += "{0} / ".format(journal.account.name)
|
|
name_debit = name_debit[:-3]
|
|
name_credit = name_credit[:-3]
|
|
|
|
body.append(
|
|
schemas.DaybookItem(
|
|
id=voucher.id,
|
|
date=voucher.date.strftime("%d-%b-%Y"),
|
|
url=[
|
|
"/",
|
|
VoucherType.by_id(voucher.type).name.replace(" ", "-").lower(),
|
|
str(voucher.id),
|
|
],
|
|
type=VoucherType.by_id(voucher.type).name,
|
|
narration=voucher.narration,
|
|
posted=voucher.posted,
|
|
debitText=name_debit,
|
|
debitAmount=debit,
|
|
creditText=name_credit,
|
|
creditAmount=credit,
|
|
)
|
|
)
|
|
return body
|