def add_shirt(request, shirt_id=None):
if shirt_id:
shirt = get_object_or_404(Shirt, pk=shirt_id)
if not (request.user.is_superuser or
shirt.uploaded_by == request.user):
messages.error(request, 'Yeah whatevah dude, nice try.')
redirect('view_shirt', shirt.id)
else:
shirt = Shirt(uploaded_by=request.user)
shot = ShirtShot(shirt=shirt, uploaded_by=request.user)
shot_form = ShotForm(request.POST or None, request.FILES or None,
instance=shot)
shirt_form = ShirtForm(request.POST or None, instance=shirt)
if ((shirt_form.is_valid() and not shirt_id and shot_form.is_valid()) or
(shirt_form.is_valid() and shirt_id)):
shirt_form.save()
message = 'Yeah, shirt edited!'
if not shirt_id:
shot_form.instance.shirt = shirt_form.instance
shot_form.save()
message = 'Yeah, shirt added!'
messages.success(request, message)
else:
import pdb
pdb.set_trace()
messages.error(request, 'Arrrrgh something went wrong, try again')
return redirect('view_shirt', shirt.id)
def add_shirt(request, shirt_id=None):
if shirt_id:
shirt = get_object_or_404(Shirt, pk=shirt_id)
if not (request.user.is_superuser
or shirt.uploaded_by == request.user):
messages.error(request, 'Yeah whatevah dude, nice try.')
redirect('view_shirt', shirt.id)
else:
shirt = Shirt(uploaded_by=request.user)
shot = ShirtShot(shirt=shirt, uploaded_by=request.user)
shot_form = ShotForm(request.POST or None,
request.FILES or None,
instance=shot)
shirt_form = ShirtForm(request.POST or None, instance=shirt)
if ((shirt_form.is_valid() and not shirt_id and shot_form.is_valid())
or (shirt_form.is_valid() and shirt_id)):
shirt_form.save()
message = 'Yeah, shirt edited!'
if not shirt_id:
shot_form.instance.shirt = shirt_form.instance
shot_form.save()
message = 'Yeah, shirt added!'
messages.success(request, message)
else:
import pdb
pdb.set_trace()
messages.error(request, 'Arrrrgh something went wrong, try again')
return redirect('view_shirt', shirt.id)
def add_shot(request, shirt_id):
shirt = get_object_or_404(Shirt, pk=shirt_id)
shot = ShirtShot(shirt=shirt, uploaded_by=request.user)
shot_form = ShotForm(request.POST or None, request.FILES or None,
instance=shot)
if shot_form.is_valid():
messages.success(request, 'Yeah, shot added!')
shot_form.save()
else:
messages.error(request, 'Arrrrgh something went wrong, try again.')
return redirect('view_shirt', shirt_id)
def add_shot(request, shirt_id):
shirt = get_object_or_404(Shirt, pk=shirt_id)
shot = ShirtShot(shirt=shirt, uploaded_by=request.user)
shot_form = ShotForm(request.POST or None,
request.FILES or None,
instance=shot)
if shot_form.is_valid():
messages.success(request, 'Yeah, shot added!')
shot_form.save()
else:
messages.error(request, 'Arrrrgh something went wrong, try again.')
return redirect('view_shirt', shirt_id)
