diff --git a/prediction_service/__pycache__/forms.cpython-310.pyc b/prediction_service/__pycache__/forms.cpython-310.pyc index 652e31d73c46f97533467bb7939be82466183871..87e6316e37c8ebcfd756aff190c90623093425ce 100644 Binary files a/prediction_service/__pycache__/forms.cpython-310.pyc and b/prediction_service/__pycache__/forms.cpython-310.pyc differ diff --git a/prediction_service/__pycache__/views.cpython-310.pyc b/prediction_service/__pycache__/views.cpython-310.pyc index e6336417a5b6bdf19b05b40494f4f6ed9448aeec..5c08f782f32a42bb8aa06bec6bfd83129caa0bfa 100644 Binary files a/prediction_service/__pycache__/views.cpython-310.pyc and b/prediction_service/__pycache__/views.cpython-310.pyc differ diff --git a/prediction_service/forms.py b/prediction_service/forms.py index c37aa3d43e0033bb044d015fed62dfe736cca96b..71d460d7ff24ecff3e572ccfcbce6052436b9420 100644 --- a/prediction_service/forms.py +++ b/prediction_service/forms.py @@ -4,7 +4,7 @@ from django.contrib.auth.models import User class UserRegisterForm(UserCreationForm): email = forms.EmailField() - role = forms.ChoiceField(choices=[('user', 'User'), ('ai_engineer', 'AI Engineer')], widget=forms.RadioSelect) + role = forms.ChoiceField(choices=[('user', 'User'), ('AI Engineer', 'AI Engineer')], widget=forms.RadioSelect) class Meta: model = User diff --git a/prediction_service/views.py b/prediction_service/views.py index 852e7a004aee49556b8bbd2901dcfa970f47bdf5..4024f7deb51bd100f8d70b54b4a0d2e053160a8a 100644 --- a/prediction_service/views.py +++ b/prediction_service/views.py @@ -67,10 +67,11 @@ def register(request): # Create EndUser if role is 'user' if role == 'user': EndUser.objects.create(user=user_profile) - elif role == 'ai_engineer': + elif role == 'AI Engineer': AIEngineer.objects.create(user=user_profile, is_authorized=False) messages.success(request, f'Account created for {username}!') + messages.success(request, f'Role value {role}!') return redirect('MLAAS-login') else: form = UserRegisterForm() @@ -92,7 +93,7 @@ def MLAAS_login(request): if user is not None: try: - if user.userprofile.role == 'ai_engineer': + if user.userprofile.role == 'AI Engineer': ai_engineer = AIEngineer.objects.get(user=user.userprofile) if not ai_engineer.is_authorized: messages.error(request, 'You are not authorized to access the MLaaS system. Please wait for admin authorization.') @@ -122,7 +123,7 @@ def create_post(request): post.author = request.user # Check if the user is an AI engineer or an admin - if request.user.userprofile.role == 'ai_engineer' or request.user.is_superuser: + if request.user.userprofile.role == 'AI Engineer' or request.user.is_superuser: post.save() messages.success(request, 'Your post has been created successfully.') return redirect('MLAAS-home', pk=post.pk) # Redirect to the post detail page