X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/3a100c86bdaa0dbdd25796286ef3b6d4888911d8..19e20c4e4b6a7940977a293698d61e75810ff321:/apps/toolbar/management/commands/fixbuttons.py diff --git a/apps/toolbar/management/commands/fixbuttons.py b/apps/toolbar/management/commands/fixbuttons.py old mode 100755 new mode 100644 index 9e721510..ea797fcf --- a/apps/toolbar/management/commands/fixbuttons.py +++ b/apps/toolbar/management/commands/fixbuttons.py @@ -1,35 +1,73 @@ -# -*- conding: utf-8 -__author__="lreqc" -__date__ ="$2009-09-08 14:31:26$" +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# + from django.core.management.base import NoArgsCommand -from toolbar.models import Button -import json +from toolbar.models import Button, ButtonGroup +from django.utils import simplejson as json import re + class Command(NoArgsCommand): - + def handle_noargs(self, **options): buttons = Button.objects.all() + print "Validating parameters... " for b in buttons: - params = b.params; + params = b.params try: v = json.loads(b.params) - except ValueError, e: - print 'On button %s: ' % b.label, b.params - print e - # try to fix the bad json - - # cut the parenthis + print 'Trying to fix button "%s" ...' % b.slug if params[0] == u'(': params = params[1:] if params[-1] == u')': params = params[:-1] + try: + v = son.loads(re.sub(u'([\\w-]+)\\s*:', u'"\\1": ', params).encode('utf-8')) + except ValueError, e: + print "Unable to fix '%s' " % b.params + print "Try to fix this button manually and rerun the script." + return False - v = json.loads(re.sub(u'([\\w-]+)\\s*:', u'"\\1": ', params).encode('utf-8')) + # resave b.params = json.dumps(v) b.save() + print "Merge duplicate buttons (if any)..." + hash = {} + for b in buttons: + if b.slug not in hash: + hash[b.slug] = b + continue + + # duplicate button + print "Found duplicate of '%s'" % b.slug + a = hash[b.slug] + + remove_duplicate = True + if a.params != b.params: + print "Conflicting params for duplicate of '%s'." % b.slug + print "Groups will be joined, but won't remove duplicates." + remove_duplicate = False + + for g in b.group.all(): + a.group.add(g) + + b.group.clear() + + a.save() + if remove_duplicate: + b.delete() + print "Searching for empty groups and orphaned buttons:" + for g in ButtonGroup.objects.all(): + if len(g.button_set.all()) == 0: + print "Empty group: '%s'" % g.slug - \ No newline at end of file + for b in Button.objects.all(): + if len(b.group.all()) == 0: + print "orphan: '%s'" % b.slug