scripts/setmainpage.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from django.core.management import setup_environ
   7 from wolnelektury import settings
   8 import sys
   9
  10 setup_environ(settings)
  11
  12 from catalogue.models import Tag
  13
  14
  15 MAIN_PAGE_THEMES = [
  16     u'Obywatel',
  17     u'Car',
  18     u'Błoto',
  19     u'Krew',
  20     u'Danse macabre',
  21     u'Obcy',
  22     u'Matka',
  23     u'Gotycyzm',
  24 ]
  25
  26
  27 for tag in Tag.objects.all():
  28     if tag.category in ('epoch', 'genre', 'author', 'kind'):
  29         tag.main_page = True
  30     elif tag.category == 'theme' and tag.name in MAIN_PAGE_THEMES:
  31         tag.main_page = True
  32     else:
  33         tag.main_page = False
  34
  35     tag.save()
  36     sys.stderr.write('.')
  37
  38