import asyncio

from sqlalchemy import insert
from sqlalchemy.ext.asyncio import AsyncSession

import models
from backend.session import session_maker
from enums.base_marketplace import BaseMarketplace


async def main():
    session: AsyncSession = session_maker()
    marketplaces = [
        {
            'key': BaseMarketplace.WILDBERRIES,
            'name': 'Wildberries',
            'icon_url': '/api/static/icons/wb.svg'
        },
        {
            'key': BaseMarketplace.OZON,
            'name': 'OZON',
            'icon_url': '/api/static/icons/ozon.svg'
        },
        {
            'key': BaseMarketplace.YANDEX_MARKET,
            'name': 'Яндекс Маркет',
            'icon_url': '/api/static/icons/ym.svg'
        }
    ]
    await session.execute(insert(models.BaseMarketplace), marketplaces)
    await session.commit()
    await session.close()


if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())