implemented feature: Removes the tasks belonging to a deactivated par…

…ents

docs/commands.rst

@@ -37,3 +37,10 @@ Update the number of words to translate in every task. It counts the number of w
 
 :code:`python manage.py update_number_of_words`
 
+
+Delete disabled parent tasks
+----------------------------
+This commands deletes the tasks that no longer have a parent enabled object. It is a command that
+can be executed from time to time or it can be added to the *cron*.
+
+:code:`python manage.py remove_tasks_for_disbled_parents`

transmanager/__init__.py

@@ -2,6 +2,6 @@
 
 __author__ = 'Andreu Vallbona'
 __email__ = '[email protected]'
-__version__ = '0.3.3'
+__version__ = '0.3.4'
 
 default_app_config = 'transmanager.apps.TransManagerConfig'

transmanager/management/commands/remove_tasks_for_disabled_parents.py

@@ -0,0 +1,38 @@
+# -*- encoding: utf-8 -*-
+
+from django.core.management.base import BaseCommand
+from transmanager.models import TransModelLanguage
+from transmanager.settings import TM_DEFAULT_ENABLED_ATTRIBUTE_NAME
+from transmanager.tasks.tasks import delete_translations_for_item_and_its_children
+from transmanager.utils import has_field
+
+
+class Command(BaseCommand):
+    help = "Removes the tasks belonging to a deactivated parents."
+
+    def handle(self, *args, **options):
+
+        filter_by = {
+            TM_DEFAULT_ENABLED_ATTRIBUTE_NAME: False
+        }
+
+        whole_ids = []
+
+        self.stdout.write('Start')
+        # generate the translations tasks for every record on the translatable models
+        for model in TransModelLanguage.objects.all():
+            mc = model.get_model_class()
+            if not has_field(mc, TM_DEFAULT_ENABLED_ATTRIBUTE_NAME):
+                continue
+
+            # if the model has the enabled attr, the we search the disabled items ids
+            disabled_ids = list(mc.objects.filter(**filter_by).values_list('id', flat=True))
+            whole_ids += disabled_ids
+            self.stdout.write('Model: {} has {} ids'.format(mc.__name__, len(disabled_ids)))
+            if disabled_ids:
+                for item_id in disabled_ids:
+                    delete_translations_for_item_and_its_children.delay(mc, item_id)
+
+        self.stdout.write('{} ids will be processed'.format(len(whole_ids)))
+
+        self.stdout.write('End')

transmanager/tasks/tasks.py

@@ -5,7 +5,6 @@
 from django.contrib.auth.models import User
 
 from django.utils.translation import ugettext_lazy as _
-from django.core.files import File
 from django.core.files.base import ContentFile
 from rq.decorators import job
 

transmanager/utils.py

@@ -2,6 +2,7 @@
 
 import re
 from django.contrib.contenttypes.models import ContentType
+from django.core.exceptions import FieldDoesNotExist
 from hvad.models import TranslatableModel
 
 
@@ -55,3 +56,18 @@ def get_num_words(text):
         return len(num_words)
     except TypeError:
         return 0
+
+
+def has_field(mc, field_name):
+    """
+    detect if a model has a given field has
+
+    :param field_name:
+    :param mc:
+    :return:
+    """
+    try:
+        mc._meta.get_field(field_name)
+    except FieldDoesNotExist:
+        return False
+    return True