ulysia/szurubooru
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

server/posts: fix relation updating

Browse source 
Fixes #103
This commit is contained in:
rr- 2016-10-02 17:21:15 +02:00
parent b0c5031001
commit b853caf6f5
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
server/szurubooru/func/posts.py
View file

@ -355,8 +355,13 @@ def update_post_tags(post, tag_names):
def update_post_relations(post, new_post_ids): def update_post_relations(post, new_post_ids):
assert post assert post
try:
new_post_ids = [int(id) for id in new_post_ids]
except ValueError:
raise InvalidPostRelationError(
'A relation must be numeric post ID.')
old_posts = post.relations old_posts = post.relations
old_post_ids = [p.post_id for p in old_posts] old_post_ids = [int(p.post_id) for p in old_posts]
new_posts = db.session \ new_posts = db.session \
.query(db.Post) \ .query(db.Post) \
.filter(db.Post.post_id.in_(new_post_ids)) \ .filter(db.Post.post_id.in_(new_post_ids)) \

1
server/szurubooru/tests/func/test_posts.py
View file

@ -521,6 +521,7 @@ def test_update_post_relations_bidirectionality(post_factory):
db.session.flush() db.session.flush()
post = post_factory() post = post_factory()
posts.update_post_relations(post, [relation1.post_id, relation2.post_id]) posts.update_post_relations(post, [relation1.post_id, relation2.post_id])
db.session.flush()
posts.update_post_relations(relation1, []) posts.update_post_relations(relation1, [])
assert len(post.relations) == 1 assert len(post.relations) == 1
assert post.relations[0].post_id == relation2.post_id assert post.relations[0].post_id == relation2.post_id