wassname/sloth
1
0
Fork 0
mirror of https://github.com/wassname/sloth.git synced 2026-07-04 23:21:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

make appendfiles command handle videos

Browse Source
This commit is contained in:
Martin Baeuml
2012-02-20 12:55:39 +01:00
parent cefe16181b
commit 1327f2c976
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sloth/core/commands.py
+15 -1
View File
@@ -85,8 +85,14 @@ class AppendFilesCommand(BaseCommand):
option_list = BaseCommand.option_list + (
make_option('-u', '--unlabeled', action='store_true', default=False,
help='Mark appended files as unlabeled.'),
make_option( '--image', action='store_true', default=False,
help='Force appended files to be recognized as images.'),
make_option( '--video', action='store_true', default=False,
help='Force appended files to be recognized as videos.'),
)
video_extensions = ['.vob', '.idx', '.mpg', '.mpeg']
def handle(self, *args, **options):
if len(args) < 2:
raise CommandError("Expect at least 2 arguments.")
@@ -98,7 +104,15 @@ class AppendFilesCommand(BaseCommand):
rel_filename = os.path.relpath(filename, os.path.dirname(args[0]))
except:
pass
item = self.labeltool.addImageFile(rel_filename)
_, ext = os.path.splitext(rel_filename)
if (not options['image'] and ext.lower() in self.video_extensions) or options['video']:
logger.debug("Adding video file: %s" % rel_filename)
item = self.labeltool.addVideoFile(rel_filename)
else:
logger.debug("Adding image file: %s" % rel_filename)
item = self.labeltool.addImageFile(rel_filename)
if options['unlabeled']:
item.setUnlabeled(True)
self.labeltool.saveAnnotations(args[0])