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)