From 60ece0c155a31fc9035e512ccc2f5c252ceefd98 Mon Sep 17 00:00:00 2001 From: Alexander Burdeiny Date: Thu, 5 May 2016 10:52:03 +0300 Subject: [PATCH] 1322 theme/management/commands/themeblog_to_blog.py added --- theme/management/commands/themeblog_to_blog.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 theme/management/commands/themeblog_to_blog.py diff --git a/theme/management/commands/themeblog_to_blog.py b/theme/management/commands/themeblog_to_blog.py new file mode 100644 index 00000000..4a720ec9 --- /dev/null +++ b/theme/management/commands/themeblog_to_blog.py @@ -0,0 +1,14 @@ +from django.core.management.base import BaseCommand, CommandError +from theme.models import Theme, ThemeBlog +from article.models import Article + + +class Command(BaseCommand): + def handle(self, *args, **options): + for article in Article.objects.filter(type=Article.blog).prefetch_related('blog_theme'): + blogthemes = article.blog_theme.all().values_list('id', flat=True) + themes = Theme.objects.filter( + types=Theme.types.article, + blogtheme_linking_id__in=blogthemes) + article.theme.clear() + article.theme.add(*themes)