diff options
| -rw-r--r-- | ishtar_common/management/commands/clean_ishtar.py | 35 | ||||
| -rw-r--r-- | ishtar_common/tests.py | 18 |
2 files changed, 53 insertions, 0 deletions
diff --git a/ishtar_common/management/commands/clean_ishtar.py b/ishtar_common/management/commands/clean_ishtar.py new file mode 100644 index 000000000..d3da40fd8 --- /dev/null +++ b/ishtar_common/management/commands/clean_ishtar.py @@ -0,0 +1,35 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# Copyright (C) 2017 Étienne Loks <etienne.loks_AT_peacefrogsDOTnet> + +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. + +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +# See the file COPYING for details. + +import sys + +from django.core.management.base import BaseCommand + +from archaeological_operations.models import Parcel + + +class Command(BaseCommand): + args = '' + help = 'Clean unused items' + + def handle(self, *args, **options): + for parcel in Parcel.objects.all(): + parcel.skip_history_when_saving = True + parcel.save() + self.stdout.write('Parcel cleaned.\n') diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index f22e42e38..10584e4f2 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -17,6 +17,7 @@ # See the file COPYING for details. +from StringIO import StringIO from bs4 import BeautifulSoup as Soup from django.conf import settings @@ -25,6 +26,7 @@ from django.contrib.contenttypes.models import ContentType from django.core.cache import cache from django.core.exceptions import ValidationError from django.core.files.base import File as DjangoFile +from django.core.management import call_command from django.core.urlresolvers import reverse from django.template.defaultfilters import slugify from django.test import TestCase @@ -76,6 +78,22 @@ def create_user(): return username, password, user +class CommandsTestCase(TestCase): + def test_clean_ishtar(self): + """ + Clean ishtar db + """ + from archaeological_operations.models import Parcel + p = Parcel.objects.create( + town=models.Town.objects.create(name='test', numero_insee='25000')) + parcel_nb = Parcel.objects.count() + out = StringIO() + call_command('clean_ishtar', stdout=out) + # no operation or file attached - the parcel should have disappear + self.assertEqual(parcel_nb - 1, Parcel.objects.count()) + self.assertEqual(Parcel.objects.filter(pk=p.pk).count(), 0) + + class WizardTestFormData(object): """ Test set to simulate wizard steps |