src/toolbar/management/commands/fixbuttons.py

   1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4
   5 from django.core.management.base import NoArgsCommand
   6 from toolbar.models import Button, ButtonGroup
   7 import json
   8 import re
   9
  10
  11 class Command(NoArgsCommand):
  12
  13     def handle_noargs(self, **options):
  14         buttons = Button.objects.all()
  15         print("Validating parameters... ")
  16         for b in buttons:
  17             params = b.params
  18             try:
  19                 v = json.loads(b.params)
  20             except ValueError as e:
  21                 print('Trying to fix button "%s" ...' % b.slug)
  22                 if params[0] == u'(':
  23                     params = params[1:]
  24                 if params[-1] == u')':
  25                     params = params[:-1]
  26                 try:
  27                     v = son.loads(re.sub(u'([\\w-]+)\\s*:', u'"\\1": ', params).encode('utf-8'))
  28                 except ValueError as e:
  29                     print("Unable to fix '%s' " % b.params)
  30                     print("Try to fix this button manually and rerun the script.")
  31                     return False
  32
  33             # resave
  34             b.params = json.dumps(v)
  35             b.save()
  36
  37         print("Merge duplicate buttons (if any)...")
  38         hash = {}
  39         for b in buttons:
  40             if b.slug not in hash:
  41                 hash[b.slug] = b
  42                 continue
  43
  44             # duplicate button
  45             print("Found duplicate of '%s'" % b.slug)
  46             a = hash[b.slug]
  47
  48             remove_duplicate = True
  49             if a.params != b.params:
  50                 print("Conflicting params for duplicate of '%s'." % b.slug)
  51                 print("Groups will be joined, but won't remove duplicates.")
  52                 remove_duplicate = False
  53
  54             for g in b.group.all():
  55                 a.group.add(g)
  56
  57             b.group.clear()
  58
  59             a.save()
  60             if remove_duplicate:
  61                 b.delete()
  62
  63         print("Searching for empty groups and orphaned buttons:")
  64         for g in ButtonGroup.objects.all():
  65             if len(g.button_set.all()) == 0:
  66                 print("Empty group: '%s'" % g.slug)
  67
  68         for b in Button.objects.all():
  69             if len(b.group.all()) == 0:
  70                 print("orphan: '%s'" % b.slug)