style & test

This commit is contained in:
Ovv
2018-02-25 19:20:51 +01:00
committed by Daniel Quinn
parent d89dbbe537
commit b10c2c770c
3 changed files with 21 additions and 14 deletions

View File

@@ -30,17 +30,25 @@ class Command(BaseCommand):
BaseCommand.__init__(self, *args, **kwargs)
def add_arguments(self, parser):
parser.add_argument("directory", default=settings.CONSUMPTION_DIR, nargs='?')
parser.add_argument("--loop-time", default=settings.CONSUMER_LOOP_TIME, type=int)
parser.add_argument(
"directory",
default=settings.CONSUMPTION_DIR,
nargs="?"
)
parser.add_argument(
"--loop-time",
default=settings.CONSUMER_LOOP_TIME,
type=int
)
parser.add_argument("--mail-delta", default=10, type=int)
parser.add_argument("--oneshot", action='store_true')
parser.add_argument("--oneshot", action="store_true")
def handle(self, *args, **options):
self.verbosity = options["verbosity"]
directory = options['directory']
loop_time = options['loop_time']
mail_delta = datetime.timedelta(minutes=options['mail_delta'])
directory = options["directory"]
loop_time = options["loop_time"]
mail_delta = datetime.timedelta(minutes=options["mail_delta"])
try:
self.file_consumer = Consumer(consume=directory)
@@ -58,7 +66,7 @@ class Command(BaseCommand):
"Starting document consumer at {}".format(directory)
)
if options['oneshot']:
if options["oneshot"]:
self.loop(mail_delta=mail_delta)
else:
try: