From 346cad74fe052975ac7e3f416934543be2450d74 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20P=C3=A9ters?= Date: Sat, 16 Mar 2019 10:05:04 +0100 Subject: [PATCH] mail: upate feed_mail command for django 1.11 (#31474) --- .../sources/mail/management/commands/feed_mail.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/welco/sources/mail/management/commands/feed_mail.py b/welco/sources/mail/management/commands/feed_mail.py index 26fb737..fc36f2f 100644 --- a/welco/sources/mail/management/commands/feed_mail.py +++ b/welco/sources/mail/management/commands/feed_mail.py @@ -23,17 +23,14 @@ from django.core.management.base import BaseCommand, CommandError from ...models import Mail class Command(BaseCommand): - args = ['category', '...'] + def add_arguments(self, parser): + parser.add_argument( + '--category', metavar='CATEGORY', default=None) + parser.add_argument('filenames', metavar='FILENAME', nargs='+') - option_list = BaseCommand.option_list + ( - make_option('--category', - dest='category', - default=None), - ) - - def handle(self, *args, **kwargs): + def handle(self, filenames, *args, **kwargs): count = 0 - for filepath in args: + for filepath in filenames: if not os.path.exists(filepath): continue if not open(filepath).read(5) == '%PDF-':