Annotate commands/export_usermessage_batch.py.

This commit is contained in:
Steve Howell 2016-09-11 15:52:04 -07:00 committed by Tim Abbott
parent 85e30a6d35
commit bb6fcaa4e7
1 changed files with 5 additions and 0 deletions

View File

@ -1,6 +1,9 @@
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import print_function from __future__ import print_function
from typing import Any
from argparse import ArgumentParser
from django.core.management.base import BaseCommand, CommandError from django.core.management.base import BaseCommand, CommandError
from django.core.exceptions import ValidationError from django.core.exceptions import ValidationError
@ -17,6 +20,7 @@ class Command(BaseCommand):
help = """UserMessage fetching helper for export.py""" help = """UserMessage fetching helper for export.py"""
def add_arguments(self, parser): def add_arguments(self, parser):
# type: (ArgumentParser) -> None
parser.add_argument('--path', parser.add_argument('--path',
dest='path', dest='path',
action="store", action="store",
@ -29,6 +33,7 @@ class Command(BaseCommand):
help='Thread ID') help='Thread ID')
def handle(self, *args, **options): def handle(self, *args, **options):
# type: (*Any, **Any) -> None
logging.info("Starting UserMessage batch thread %s" % (options['thread'],)) logging.info("Starting UserMessage batch thread %s" % (options['thread'],))
files = set(glob.glob(os.path.join(options['path'], 'messages-*.json.partial'))) files = set(glob.glob(os.path.join(options['path'], 'messages-*.json.partial')))
for partial_path in files: for partial_path in files: