def gallerify(id):
""" Ads and removes image from gallery """
image = UserImages.get_image(id)
if not image:
return redirect(url_for("user_images", username=session["user"]))
UserImages.gallerify(image)
return redirect(url_for("user_images", username=session["user"]))
def gallerify(id):
""" Ads and removes image from gallery """
image = UserImages.get_image(id)
if not image:
return redirect(url_for("user_images", username=session["user"]))
UserImages.gallerify(image)
return redirect(url_for("user_images", username=session["user"]))
def test_creating_and_deleting_image(self):
with test_database(db, (Users, UserImages)):
Users.create_user(username = "******", password = "******")
user1 = Users.select().get()
UserImages.add_image(image_link = self.ADDRESS,\
description = "test",\
is_vertical = True, \
owner = user1,\
imgur_img = False,\
delete_hash = None)
image = UserImages.select().get()
self.assertEquals(self.ADDRESS ,image.image_link)
self.assertEquals("test", image.description)
self.assertEquals(user1, image.owner)
UserImages.add_image(image_link = self.ADDRESS,\
description = "test",\
is_vertical = True, \
owner = user1,\
imgur_img = False,\
delete_hash = None)
self.assertEquals(2, UserImages.get_count())
self.assertTrue(UserImages.check_exists(self.ADDRESS))
self.assertFalse(UserImages.check_exists(self.ADDRESS + "bam"))
image = UserImages.get_image(2)
UserImages.delete_image(image)
self.assertEquals(1, UserImages.get_count())
image = UserImages.get_image(1)
UserImages.delete_image(image)
self.assertEquals(0, UserImages.get_count())
def test_creating_and_deleting_image(self):
with test_database(db, (Users, UserImages)):
Users.create_user(username="******", password="******")
user1 = Users.select().get()
UserImages.add_image(image_link = self.ADDRESS,\
description = "test",\
is_vertical = True, \
owner = user1,\
imgur_img = False,\
delete_hash = None)
image = UserImages.select().get()
self.assertEquals(self.ADDRESS, image.image_link)
self.assertEquals("test", image.description)
self.assertEquals(user1, image.owner)
UserImages.add_image(image_link = self.ADDRESS,\
description = "test",\
is_vertical = True, \
owner = user1,\
imgur_img = False,\
delete_hash = None)
self.assertEquals(2, UserImages.get_count())
self.assertTrue(UserImages.check_exists(self.ADDRESS))
self.assertFalse(UserImages.check_exists(self.ADDRESS + "bam"))
image = UserImages.get_image(2)
UserImages.delete_image(image)
self.assertEquals(1, UserImages.get_count())
image = UserImages.get_image(1)
UserImages.delete_image(image)
self.assertEquals(0, UserImages.get_count())
def delete_image(id):
image = UserImages.get_image(id)
if not image:
abort(404)
if image.owner.username != session["user"]:
flash("Don't try to delete other\'s dude\'s pictures...dude")
return redirect(url_for("index"))
else:
try:
UserImages.delete_image(image)
except:
error = "Error occured when writing to database"
flash(error)
return redirect(url_for("index"))
if image.imgur_img:
resp = ImgurHandler(get_config().imgur_id).delete_image(image.delete_hash)
if not resp["success"]:
handle_errors(resp)
return redirect(url_for("user_images", username=session["user"]))
def delete_image(id):
image = UserImages.get_image(id)
if not image:
abort(404)
if image.owner.username != session["user"]:
flash("Don't try to delete other\'s dude\'s pictures...dude")
return redirect(url_for("index"))
else:
try:
UserImages.delete_image(image)
except:
error = "Error occured when writing to database"
flash(error)
return redirect(url_for("index"))
if image.imgur_img:
resp = ImgurHandler(get_config().imgur_id).delete_image(
image.delete_hash)
if not resp["success"]:
handle_errors(resp)
return redirect(url_for("user_images", username=session["user"]))
