diff --git a/pandora/annotation/management/commands/import_srt.py b/pandora/annotation/management/commands/import_srt.py index cb647fdc..011591c6 100644 --- a/pandora/annotation/management/commands/import_srt.py +++ b/pandora/annotation/management/commands/import_srt.py @@ -20,19 +20,14 @@ class Command(BaseCommand): """ help = 'import annotations from srt or vtt' args = 'username item layername filename.srt' - - def add_arguments(self, parser): - parser.add_argument('username', help='username') - parser.add_argument('item', help='item') - parser.add_argument('layername', help='layer') - parser.add_argument('filename', help='filename.srt') + option_list = BaseCommand.option_list + ( + ) def handle(self, *args, **options): - username = options['username'] - public_id = options['item'] - layer_id = options['layer'] - filename = options['filename'] - + if len(args) != 4: + print(self.usage('import_srt')) + return + username, public_id, layer_id, filename = args user = User.objects.get(username=username) item = Item.objects.get(public_id=public_id) layer = list(filter(lambda l: l['id'] == layer_id, settings.CONFIG['layers']))[0] diff --git a/requirements.txt b/requirements.txt index d26eb648..6e92b824 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,3 @@ html5lib requests==2.9.1 tornado==4.1 geoip2==2.2.0 -youtube-dl