From 96ffe46ed194b842c5df711fd0135d51655b880d Mon Sep 17 00:00:00 2001 From: =?utf8?q?Marek=20St=C4=99pniowski?= Date: Mon, 29 Sep 2008 15:22:49 +0200 Subject: [PATCH] Save shelves for each book during reimport. --- apps/catalogue/models.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/apps/catalogue/models.py b/apps/catalogue/models.py index 8a7722749..18e7e804d 100644 --- a/apps/catalogue/models.py +++ b/apps/catalogue/models.py @@ -157,8 +157,14 @@ class Book(models.Model): book_info = dcparser.parse(xml_file) book_base, book_slug = book_info.url.rsplit('/', 1) book, created = Book.objects.get_or_create(slug=book_slug) - if not created and not overwrite: - raise Book.AlreadyExists('Book %s already exists' % book_slug) + + if created: + book_shelves = [] + else: + if not overwrite: + raise Book.AlreadyExists('Book %s already exists' % book_slug) + # Save shelves for this book + book_shelves = book.tags.filter(category='set') book.title = book_info.title book._short_html = '' @@ -224,7 +230,7 @@ class Book(models.Model): book_themes += themes book_themes = set(book_themes) - book.tags = list(book.tags) + list(book_themes) + book.tags = list(book.tags) + list(book_themes) + list(book_shelves) book.save() return book -- 2.20.1