import asyncio

from sqlalchemy.ext.asyncio import AsyncSession

from backend.session import session_maker
from enums.user import UserRole
from models import Role

role_name_dictionary = {
    UserRole.admin: "Админ",
    UserRole.user: "Базовый пользователь",
    UserRole.manager: "Менеджер",
    UserRole.employee: "Сотрудник",
}


async def main():
    session: AsyncSession = session_maker()
    for key, name in role_name_dictionary.items():
        role = Role(
            key=key,
            name=name
        )
        session.add(role)
    await session.commit()
    await session.close()


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