def run_batch(self, batch):
"""runs the single batch
prints batch title
then loops through the query set
and prints progress in %
afterwards there will be a short summary
"""
sys.stdout.write(batch['title'])
changed_count = 0
checked_count = 0
total_count = batch['query_set'].count()
if total_count == 0:
transaction.commit()
return
for item in batch['query_set'].all():
item_changed = batch['function'](item)
transaction.commit()
if item_changed:
changed_count += 1
checked_count += 1
progress = 100*float(checked_count)/float(total_count)
console.print_progress(FORMAT_STRING, progress)
print FORMAT_STRING % 100
if changed_count:
print batch['changed_count_message'] % changed_count
else:
print batch['nothing_changed_message']
def run_command(self):
"""method that runs the actual command"""
#go through tags and find character case duplicates and eliminate them
tagnames = models.Tag.objects.values_list('name', flat = True)
for name in tagnames:
dupes = models.Tag.objects.filter(name__iexact = name)
first_tag = dupes[0]
if dupes.count() > 1:
line = 'Found duplicate tags for %s: ' % first_tag.name
print line,
for idx in xrange(1, dupes.count()):
print dupes[idx].name + ' ',
dupes[idx].delete()
print ''
if openode_settings.FORCE_LOWERCASE_TAGS:
lowercased_name = first_tag.name.lower()
if first_tag.name != lowercased_name:
print 'Converting tag %s to lower case' % first_tag.name
first_tag.name = lowercased_name
first_tag.save()
transaction.commit()
#go through questions and fix tag records on each
threads = models.Thread.objects.all()
checked_count = 0
found_count = 0
total_count = threads.count()
print "Searching for questions with inconsistent tag records:",
for thread in threads:
tags = thread.tags.all()
denorm_tag_set = set(thread.get_tag_names())
norm_tag_set = set(thread.tags.values_list('name', flat=True))
if norm_tag_set != denorm_tag_set:
if thread.last_edited_by:
user = thread.last_edited_by
timestamp = thread.last_edited_at
else:
user = thread.author
timestamp = thread.added_at
tagnames = forms.TagNamesField().clean(thread.tagnames)
thread.update_tags(
tagnames = tagnames,
user = user,
timestamp = timestamp
)
thread.tagnames = tagnames
thread.save()
found_count += 1
transaction.commit()
checked_count += 1
progress = 100*float(checked_count)/float(total_count)
console.print_progress(FORMAT_STRING, progress)
print FORMAT_STRING % 100
if found_count:
print '%d problem questions found, tag records restored' % found_count
else:
print 'Did not find any problems'
