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

32 lines
743 B
Python

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