From 9edaaffec230130566289deb16cc79ee37597a9e Mon Sep 17 00:00:00 2001
From: rr-
Date: Tue, 3 Jan 2017 21:37:38 +0100
Subject: server/posts: fix post relations

Trying to relate post to itself resulted in 500 ISE.
---
 server/szurubooru/func/posts.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'server/szurubooru/func/posts.py')

diff --git a/server/szurubooru/func/posts.py b/server/szurubooru/func/posts.py
index 6ff3a87..d190b8d 100644
--- a/server/szurubooru/func/posts.py
+++ b/server/szurubooru/func/posts.py
@@ -377,6 +377,8 @@ def update_post_relations(post, new_post_ids):
         .all()
     if len(new_posts) != len(new_post_ids):
         raise InvalidPostRelationError('One of relations does not exist.')
+    if post.post_id in new_post_ids:
+        raise InvalidPostRelationError('Post cannot relate to itself.')
 
     relations_to_del = [p for p in old_posts if p.post_id not in new_post_ids]
     relations_to_add = [p for p in new_posts if p.post_id not in old_post_ids]
-- 
cgit v1.3