DutyLog/test_report_generation.py

import asyncio
from datetime import date

from dutylog.infrastructure.database.config import create_engine, create_session_maker
from dutylog.infrastructure.database.dao.hours_transactions_dao import HoursTransactionsDAO
from dutylog.infrastructure.database.dao.residents_dao import ResidentsDAO
from dutylog.infrastructure.database.dao.rooms_dao import RoomsDAO
from dutylog.infrastructure.database.dao.floors_dao import FloorsDAO
from dutylog.infrastructure.database.dao.reporting_periods_dao import ReportingPeriodsDAO
from dutylog.infrastructure.database.dao.users_dao import UsersDAO
from dutylog.infrastructure.database.repositories.hours_transactions_repository import (
    HoursTransactionsRepository,
)
from dutylog.infrastructure.database.repositories.residents_repository import ResidentsRepository
from dutylog.infrastructure.database.repositories.rooms_repository import RoomsRepository
from dutylog.infrastructure.database.repositories.floors_repository import FloorsRepository
from dutylog.infrastructure.database.repositories.reporting_periods_repository import (
    ReportingPeriodsRepository,
)
from dutylog.infrastructure.database.repositories.users_repository import UsersRepository
from dutylog.infrastructure.utils.config import load_config
from dutylog.services.report_service import ReportService


async def test_report_generation() -> None:
    config = load_config()
    engine = create_engine(config.database.url)
    session_maker = create_session_maker(engine)
    
    async with session_maker() as session:
        hours_transactions_dao = HoursTransactionsDAO(session)
        residents_dao = ResidentsDAO(session)
        rooms_dao = RoomsDAO(session)
        floors_dao = FloorsDAO(session)
        reporting_periods_dao = ReportingPeriodsDAO(session)
        users_dao = UsersDAO(session)
        
        hours_transactions_repository = HoursTransactionsRepository(
            hours_transactions_dao, residents_dao
        )
        residents_repository = ResidentsRepository(residents_dao)
        rooms_repository = RoomsRepository(rooms_dao)
        floors_repository = FloorsRepository(floors_dao)
        reporting_periods_repository = ReportingPeriodsRepository(reporting_periods_dao)
        users_repository = UsersRepository(users_dao)
        
        report_service = ReportService(
            hours_transactions_repository,
            residents_repository,
            rooms_repository,
            floors_repository,
            users_repository,
        )
        
        active_period = await reporting_periods_repository.get_active_period()
        
        if not active_period:
            print("⚠️ Нет активного периода")
            return
        
        print(f"📅 Активный период: с {active_period.start_date}")
        
        end_date = active_period.end_date if active_period.end_date else date.today()
        
        print(f"📊 Генерирую отчёт за период {active_period.start_date} - {end_date}")
        
        report_file = await report_service.generate_period_report(
            active_period.start_date, end_date
        )
        
        filename = f"test_report_{active_period.start_date}_{end_date}.xlsx"
        with open(filename, "wb") as f:
            f.write(report_file.read())
        
        print(f"✅ Отчёт сохранён в файл: {filename}")


if __name__ == "__main__":
    asyncio.run(test_report_generation())