def delete_pic(picture_id):
#delete the picture from the database, and file system
if not 'MyWebsite_user_id' in session.keys():
return redirect('/')
if not User.is_logged_in(session['MyWebsite_user_id'],
session['login_session']):
return redirect('/danger')
pic = Picture.query.get(picture_id)
print("deleting: ", pic)
if os.path.exists('UserFiles/' + pic.file_path):
os.remove('UserFiles/' + pic.file_path)
if os.path.exists('UserFiles/thumbnails' + pic.file_path):
os.remove('UserFiles/thumbnails/' + pic.file_path)
Picture.delete(picture_id)
print("delete success: ", pic)
else:
print("delete fail: ", pic)
return redirect('/dashboard')
def upload_picture(request):
"""Upload picture view.
"""
if request.method == 'POST':
picture = Picture()
picture.user = request.user
form = UploadFileForm(request.POST, request.FILES, instance=picture)
response = {}
if form.is_valid():
picture.picture = request.FILES['picture']
picture.picture.name = unidecode(picture.picture.name)
picture.save()
### nasty error at 3:45 AM ;/
try:
rotate_picture(picture)
picture.update_thumb()
picture.save()
except:
import sys, traceback
traceback.print_exc(file=sys.stdout)
picture.delete()
response[
'status'] = 'UPLOAD ERROR. PUT HELMET ON AND EVACUATE!!!'
response = json.dumps(response)
return HttpResponse(response)
response['url'] = picture.picture_thumb.url
response['id'] = picture.id
response = json.dumps(response)
else:
response = "Failed to upload"
else:
if request.GET.has_key('gallery_id'):
preffered_gallery = int(request.GET['gallery_id'])
else:
preffered_gallery = None
galleries = Gallery.objects.filter(user=request.user)
return render(request, "upload.html", {
'galleries': galleries,
'preffered_gallery': preffered_gallery
})
return HttpResponse(response)
def upload_picture(request):
"""Upload picture view.
"""
if request.method == "POST":
picture = Picture()
picture.user = request.user
form = UploadFileForm(request.POST, request.FILES, instance=picture)
response = {}
if form.is_valid():
picture.picture = request.FILES["picture"]
picture.picture.name = unidecode(picture.picture.name)
picture.save()
### nasty error at 3:45 AM ;/
try:
rotate_picture(picture)
picture.update_thumb()
picture.save()
except:
import sys, traceback
traceback.print_exc(file=sys.stdout)
picture.delete()
response["status"] = "UPLOAD ERROR. PUT HELMET ON AND EVACUATE!!!"
response = json.dumps(response)
return HttpResponse(response)
response["url"] = picture.picture_thumb.url
response["id"] = picture.id
response = json.dumps(response)
else:
response = "Failed to upload"
else:
if request.GET.has_key("gallery_id"):
preffered_gallery = int(request.GET["gallery_id"])
else:
preffered_gallery = None
galleries = Gallery.objects.filter(user=request.user)
return render(request, "upload.html", {"galleries": galleries, "preffered_gallery": preffered_gallery})
return HttpResponse(response)
