From c423a13f85285852f827641ceb2891cda5b6aaad Mon Sep 17 00:00:00 2001 From: Daniel Quinn Date: Thu, 11 Feb 2016 12:24:18 +0000 Subject: [PATCH] Added a simple re-tagger --- .../management/commands/document_retagger.py | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 src/documents/management/commands/document_retagger.py diff --git a/src/documents/management/commands/document_retagger.py b/src/documents/management/commands/document_retagger.py new file mode 100644 index 000000000..79f6d2c66 --- /dev/null +++ b/src/documents/management/commands/document_retagger.py @@ -0,0 +1,34 @@ +from django.core.management.base import BaseCommand + +from documents.models import Document, Tag + + +class Command(BaseCommand): + + help = """ + Using the current set of tagging rules, apply said rules to all + documents in the database, effectively allowing you to back-tag all + previously indexed documents with tags created (or modified) after their + initial import. + """.replace(" ", "") + + def __init__(self, *args, **kwargs): + self.verbosity = 0 + BaseCommand.__init__(self, *args, **kwargs) + + def handle(self, *args, **options): + + self.verbosity = options["verbosity"] + + for document in Document.objects.all(): + tags = Tag.objects.exclude( + pk__in=document.tags.values_list("pk", flat=True)) + for tag in tags: + if tag.matches(document.content): + self._render( + 'Tagging {} with "{}"'.format(document, tag), 1) + document.tags.add(tag) + + def _render(self, text, verbosity): + if self.verbosity >= verbosity: + print(text)