import asyncio

from sqlalchemy.ext.asyncio import AsyncSession

from backend.session import session_maker
from models import Product, ProductBarcode


async def main(session: AsyncSession):
    client_ids = [8, 18]
    for client_id in client_ids:
        for i in range(1, 500 + 1):
            product = Product(
                name=f"Товар №{i}",
                article=f"Ариткул товара №{i}",
                client_id=client_id
            )
            session.add(product)
            await session.flush()
            for j in range(1, 5 + 1):
                barcode = ProductBarcode(
                    barcode=f"Штрихкод №{j} для товара №{i}",
                    product_id=product.id
                )
                session.add(barcode)
            await session.flush()
    await session.commit()


async def preload():
    async with session_maker() as session:
        await main(session)


if __name__ == '__main__':
    asyncio.run(preload())