def update(self, instance, validated_data):
positions_data = self.context['request'].data['positions']
user_id = self.context['request'].data['id']
user = get_object_or_404(User, pk=user_id)
instance = super(UserDetailSerializer, self).update(instance, validated_data)
new_positions = []
for row in positions_data:
position = get_object_or_404(Position, pk=row)
new_positions.append(position)
current_positions = UserPosition.objects.filter(user=user_id)
current_positions.delete() # TODO not delete already existing positions
for new_pos in new_positions:
UserPosition.objects.create(user=user, position=new_pos)
categories_data = self.context['request'].data['categories']
new_categories = []
for row in categories_data:
category = get_object_or_404(Category, pk=row)
new_categories.append(category)
current_categories = UserCategory.objects.filter(user=user_id)
current_categories.delete() # TODO not delete already existing categories
for new_cat in new_categories:
UserCategory.objects.create(user=user, category=new_cat)
units_data = self.context['request'].data['units']
new_units = []
for row in units_data:
unit = get_object_or_404(Unit, pk=row)
new_units.append(unit)
current_units = UserUnit.objects.filter(user=user_id)
current_units.delete() # TODO not delete already existing units
for new_un in new_units:
UserUnit.objects.create(user=user, unit=new_un)
faculties_data = self.context['request'].data['faculties']
new_faculties = []
for row in faculties_data:
faculty = get_object_or_404(Faculty, pk=row)
new_faculties.append(faculty)
current_faculties = UserFaculty.objects.filter(user=user_id)
current_faculties.delete() # TODO not delete already existing faculties
for new_fac in new_faculties:
UserFaculty.objects.create(user=user, faculty=new_fac)
role_data = self.context['request'].data['role']
clear_roles(user)
for row in role_data:
assign_role(user, retrieve_role(row))
return instance
def test_retrieve_unknowun_role(self):
role = retrieve_role('unknowun_role')
self.assertIsNone(role)
def test_retrive_role3(self):
self.assertEquals(retrieve_role('sho_new_name'), ShoRole3)
def test_retrive_role2(self):
self.assertEquals(retrieve_role('sho_role2'), ShoRole2)
def test_retrieve_unknowun_role(self):
role = retrieve_role('unknowun_role')
self.assertIsNone(role)
def test_retrive_role3(self):
self.assertEquals(retrieve_role('sho_new_name'), ShoRole3)
def test_retrive_role2(self):
self.assertEquals(retrieve_role('sho_role2'), ShoRole2)