Files
Fulfillment-Backend/utils/init_marketplaces.py
2024-07-21 10:56:59 +03:00

38 lines
987 B
Python

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())