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
from enums.payroll import PaySchemeType


async def main():
    session: AsyncSession = session_maker()
    schemas = [
        {
            'key': PaySchemeType.hourly,
            'name': 'Почасовая',
        },
        {
            'key': PaySchemeType.daily,
            'name': 'Подневная',
        },
        {
            'key': PaySchemeType.monthly,
            'name': 'Помесячная',

        }
    ]
    await session.execute(insert(models.PayrollScheme), schemas)
    await session.commit()
    await session.close()


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