def set_avatar(request, pk):
photo = get_or_none(Photo, user=request.user, pk=pk)
old_avatar = get_or_none(Photo, user=request.user, avatar=True)
if old_avatar:
old_avatar.avatar = False
old_avatar.save()
photo.avatar = True
photo.save()
return HttpResponse('Avatar set')
def get(self, request, pk):
client = get_or_none(Client, pk=pk)
if not client and int(pk) != request.user.pk:
raise Http404('Client with id = %s not found' % pk)
elif client and int(pk) != request.user.pk:
return redirect('client:detail', pk=pk)
return super(ClientUpdateProfile, self).get(request, pk)
def remove_from_friends(request, pk):
user = get_or_none(User, pk=pk)
if not user:
return HttpResponse('User not found')
user.client.friends.remove(request.user.client)
request.user.client.friends.remove(user.client)
return HttpResponse('Deleted')
def add_to_friends(request, pk):
user = get_or_none(User, pk=pk)
if not user:
return HttpResponse('User not found')
user.client.friends.add(request.user.client)
request.user.client.friends.add(user.client)
return HttpResponse('Added')
def photo_save(request):
if request.user.client_photo.count() >= 7:
return HttpResponse('Max is 7')
avatar = get_or_none(Photo, user=request.user, avatar=True)
photo = Photo(user=request.user, photo=request.FILES['file'])
if not avatar:
photo.avatar = True
photo.save()
return HttpResponse('Uploaded')
def get_context_data(self, object): context = super(ClientDetailProfile, self).get_context_data(object=object) photos = Photo.objects.filter(user=object.user) if photos: context['avatar'] = get_or_none(Photo, avatar=True, user=object.user) context['photos'] = photos.filter(avatar=False) # if current user not in object friends and we don't locate or our profile page if self.request.user.client not in object.user.client.friends.all() and not self.request.user == object.user: context['add_to_friends'] = True return context
def post(self, request, pk):
if 'cancel' in request.POST:
return redirect('client:detail', pk=pk)
self.object = super(ClientUpdateProfile, self).get_object()
user = self.object.user
user_form = ClientUserChangeForm(request.POST, instance=user)
if user_form.is_valid():
we = user_form.save()
else:
context = self.get_context_data(request=request,pk=pk)
context['user_form'] = user_form
return render(request, self.template_name, context)
if request.POST.get('city', None):
self.object.city = get_or_none(City, pk=request.POST.get('city'))
self.object.save()
return super(ClientUpdateProfile, self).post(request, pk)
def photo_delete(request, pk):
photo = get_or_none(Photo, user=request.user, pk=pk)
if photo:
photo.delete()
return HttpResponse('Deleted')
return HttpResponse('Photo does not exist')
def goal_add_me(request, pk):
goal = get_or_none(Goal, pk=pk)
goal.users.add(request.user)
request.user.client.goals.add(goal)
return redirect("goal:detail", pk=pk)
