diff --git a/src/watson/management/commands/installwatson.py b/src/watson/management/commands/installwatson.py index f206728..96fffbf 100644 --- a/src/watson/management/commands/installwatson.py +++ b/src/watson/management/commands/installwatson.py @@ -2,12 +2,12 @@ from __future__ import unicode_literals -from django.core.management.base import NoArgsCommand +from django.core.management.base import BaseCommand from watson.search import get_backend -class Command(NoArgsCommand): +class Command(BaseCommand): help = "Creates the database indices needed by django-watson." diff --git a/src/watson/management/commands/listwatson.py b/src/watson/management/commands/listwatson.py index 6292bee..34763b2 100644 --- a/src/watson/management/commands/listwatson.py +++ b/src/watson/management/commands/listwatson.py @@ -1,9 +1,9 @@ """Exposed the watson.get_registered_models() function as management command for debugging purpose. """ -from django.core.management.base import NoArgsCommand +from django.core.management.base import BaseCommand from watson import search as watson -class Command(NoArgsCommand): +class Command(BaseCommand): help = "List all registed models by django-watson." diff --git a/src/watson/management/commands/uninstallwatson.py b/src/watson/management/commands/uninstallwatson.py index a537e24..411dbed 100644 --- a/src/watson/management/commands/uninstallwatson.py +++ b/src/watson/management/commands/uninstallwatson.py @@ -2,12 +2,12 @@ from __future__ import unicode_literals -from django.core.management.base import NoArgsCommand +from django.core.management.base import BaseCommand from watson.search import get_backend -class Command(NoArgsCommand): +class Command(BaseCommand): help = "Destroys the database indices needed by django-watson."