def setUp(self): trap = io.StringIO() with redirect_stdout(trap): check_default_models() admin_user = User.objects.create_superuser(**ADMIN_CREDENTIALS) user_data = UserData() user_data.user = admin_user user_data.save() self.client.post(reverse('login'), ADMIN_CREDENTIALS, follow=True)
def setUp(self): user = User.objects.create_user(**USER_CREDENTIALS) user_data = UserData() user_data.user = user user_data.save() admin_user = User.objects.create_superuser(**ADMIN_CREDENTIALS) user_data = UserData() user_data.user = admin_user user_data.save()
def setUp(self): trap = io.StringIO() with redirect_stdout(trap): init() user = User.objects.create_user(**USER_CREDENTIALS) user_data = UserData() user_data.user = user user_data.save() cruise = Cruise(leader=user) cruise.save() self.cruise = cruise
def set_as_admin(request, pk): user = get_object_or_404(User, pk=pk) if request.user.is_superuser: user.is_staff = True user.is_admin = True user.is_superuser = True try: user_data = user.userdata except UserData.DoesNotExist: user_data = UserData() user_data.user = user user_data.save() old_role = user_data.role user_data.role = "admin" user.is_staff = True user.is_superuser = True user_data.save() user.save() action = Action(user=request.user, target=str(user)) action.action = "set user as admin" action.timestamp = timezone.now() action.save() Cruise.objects.filter(leader=user).update( missing_information_cache_outdated=True) messages.add_message(request, messages.WARNING, mark_safe('User ' + str(user) + ' set as admin.')) if old_role == "": send_user_approval_email(request, user) else: raise PermissionDenied return redirect(request.META['HTTP_REFERER'])
def check_for_and_fix_users_without_userdata(): from django.contrib.auth.models import User from django.core.exceptions import PermissionDenied, ObjectDoesNotExist from reserver.models import UserData, Organization for user in User.objects.all(): # check for users without user data, and add them to unapproved users if they're not admins # these may be legacy accounts or accounts created using manage.py's adduser try: user.userdata except ObjectDoesNotExist: user_data = UserData() if user.is_superuser: user_data.role = "admin" user_data.organization = Organization.objects.get(name="R/V Gunnerus") else: user_data.role = "" user_data.user = user user_data.save()