Skip to content

Commit

Permalink
Updated: help text - default value + pep-8 is happy now.
Browse files Browse the repository at this point in the history
  • Loading branch information
im-n1 committed Sep 13, 2019
1 parent d2be699 commit 3a591f5
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions django_seed/management/commands/seed.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,9 @@

from django.core.management.base import AppCommand
from django.db.models import ForeignKey
from django_seed import Seed
from django_seed.exceptions import SeederCommandError
from django_seed.toposort import toposort_flatten
from optparse import make_option
import django


class Command(AppCommand):
Expand All @@ -15,15 +13,14 @@ class Command(AppCommand):

option_list = [
make_option('--number', dest='number', default=10,
help='number of each model to seed'),
help='number of each model to seed (default 10)'),
]

def add_arguments(self, parser):
super(Command, self).add_arguments(parser)

parser.add_argument('--number', nargs='?', type=int, default=10, const=10,
help='number of each model to seed')

help_text = 'number of each model to seed (default 10)'
parser.add_argument('--number', nargs='?', type=int, default=10, const=10, help=help_text)

def handle_app_config(self, app_config, **options):
if app_config.models_module is None:
Expand Down Expand Up @@ -60,4 +57,3 @@ def sorted_models(self, app_config):
return toposort_flatten(dependencies)
except ValueError as ex:
raise SeederCommandError(str(ex))

0 comments on commit 3a591f5

Please sign in to comment.