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"]))
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())