forked from 0x2620/pandora
add manage.py import_srt command
This commit is contained in:
parent
adbc8666c0
commit
d9d1a600a9
3 changed files with 54 additions and 0 deletions
0
pandora/annotation/management/__init__.py
Normal file
0
pandora/annotation/management/__init__.py
Normal file
0
pandora/annotation/management/commands/__init__.py
Normal file
0
pandora/annotation/management/commands/__init__.py
Normal file
54
pandora/annotation/management/commands/import_srt.py
Normal file
54
pandora/annotation/management/commands/import_srt.py
Normal file
|
@ -0,0 +1,54 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# vi:si:et:sw=4:sts=4:ts=4
|
||||||
|
|
||||||
|
from django.core.management.base import BaseCommand
|
||||||
|
from django.db import transaction
|
||||||
|
from django.conf import settings
|
||||||
|
|
||||||
|
import ox
|
||||||
|
|
||||||
|
import monkey_patch.models
|
||||||
|
|
||||||
|
from item.models import Item
|
||||||
|
from user.models import User
|
||||||
|
|
||||||
|
from ... import models
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
"""
|
||||||
|
import annotations
|
||||||
|
"""
|
||||||
|
help = 'import annotations from srt'
|
||||||
|
args = 'username itemId layername filename.srt'
|
||||||
|
option_list = BaseCommand.option_list + (
|
||||||
|
)
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
if len(args) != 4:
|
||||||
|
print self.usage('import_srt')
|
||||||
|
return
|
||||||
|
username, itemId, layer_id, filename = args
|
||||||
|
user = User.objects.get(username=username)
|
||||||
|
item = Item.objects.get(itemId=itemId)
|
||||||
|
layer = filter(lambda l: l['id'] == layer_id, settings.CONFIG['layers'])[0]
|
||||||
|
|
||||||
|
annotations = ox.srt.load(filename)
|
||||||
|
print 'importing %d annotations into %s/%s' % (len(annotations), itemId, layer_id)
|
||||||
|
with transaction.commit_on_success():
|
||||||
|
for a in annotations:
|
||||||
|
if a['value']:
|
||||||
|
annotation = models.Annotation(
|
||||||
|
item=item,
|
||||||
|
layer=layer_id,
|
||||||
|
user=user,
|
||||||
|
start=float(a['in']), end=float(a['out']),
|
||||||
|
value=a['value'])
|
||||||
|
annotation.save()
|
||||||
|
#update facets if needed
|
||||||
|
if filter(lambda f: f['id'] == layer_id, settings.CONFIG['filters']):
|
||||||
|
item.update_layer_facet(layer_id)
|
||||||
|
Item.objects.filter(id=item.id).update(modified=annotation.modified)
|
||||||
|
annotation.item.modified = annotation.modified
|
||||||
|
annotation.item.update_find()
|
||||||
|
annotation.item.update_sort()
|
||||||
|
annotation.item.update_facets()
|
Loading…
Reference in a new issue