0001-mail-upate-feed_mail-command-for-django-1.11-31474.patch
welco/sources/mail/management/commands/feed_mail.py | ||
---|---|---|
23 | 23 |
from ...models import Mail |
24 | 24 | |
25 | 25 |
class Command(BaseCommand): |
26 |
args = ['category', '...'] |
|
26 |
def add_arguments(self, parser): |
|
27 |
parser.add_argument( |
|
28 |
'--category', metavar='CATEGORY', default=None) |
|
29 |
parser.add_argument('filenames', metavar='FILENAME', nargs='+') |
|
27 | 30 | |
28 |
option_list = BaseCommand.option_list + ( |
|
29 |
make_option('--category', |
|
30 |
dest='category', |
|
31 |
default=None), |
|
32 |
) |
|
33 | ||
34 |
def handle(self, *args, **kwargs): |
|
31 |
def handle(self, filenames, *args, **kwargs): |
|
35 | 32 |
count = 0 |
36 |
for filepath in args:
|
|
33 |
for filepath in filenames:
|
|
37 | 34 |
if not os.path.exists(filepath): |
38 | 35 |
continue |
39 | 36 |
if not open(filepath).read(5) == '%PDF-': |
40 |
- |