コード例 #1
0
    def test__clean_class_objects(self):
        """Runs the models._clean_class_objects function."""
        # This function returns the number of ojects saved/cleaned
        obj = JanitorTestModel(content="<p><x>Foo</x></p>")
        obj.save()

        self.assertEqual(_clean_class_objects([JanitorTestModel]), 1)
コード例 #2
0
ファイル: clean_model.py プロジェクト: genba/django-janitor
    def handle_label(self, label, **options):
        try:
            app_label, model = label.lower().split('.')
        except ValueError:
            raise CommandError(
                "Invalid app_label.model_name: {0}".format(label)
            )

        self.stdout.write("\nCleaning {0}.{1}\n".format(app_label, model))

        qs = FieldSanitizer.objects.filter(
            content_type__app_label=app_label,
            content_type__model=model
        )
        try:
            assert qs.count() > 0
        except AssertionError:
            raise CommandError("It looks like there are no FieldSanitizers "
                "defined for {0}.{1}".format(app_label, model)
            )

        klass_list = [fs.content_type.model_class() for fs in qs]
        object_count = _clean_class_objects(klass_list)

        self.stdout.write("Cleaned {0} Objects\n".format(object_count))
コード例 #3
0
    def handle_noargs(self, **options):
        sanitizers = FieldSanitizer.objects.all()
        klass_list = [fs.content_type.model_class() for fs in sanitizers]
        object_count = _clean_class_objects(klass_list)

        self.stdout.write("\nCleaned {0} Objects\n".format(object_count))
コード例 #4
0
 def handle(self, *args, **options):
     sanitizers = FieldSanitizer.objects.all()
     klass_list = [fs.content_type.model_class() for fs in sanitizers]
     updated = _clean_class_objects(klass_list)
     self.stdout.write("\nCleaned {0} Objects\n".format(updated))