fix task
This commit is contained in:
parent
ade9a049c2
commit
41e256be5d
2 changed files with 8 additions and 18 deletions
20
tasks.py
20
tasks.py
|
@ -6,10 +6,11 @@ from app.celery import app
|
|||
from django.conf import settings
|
||||
|
||||
from item import utils
|
||||
from user.models import User
|
||||
from item.models import Item
|
||||
from taskqueue.models import Task
|
||||
from user.models import User
|
||||
|
||||
from . import transcribe
|
||||
from .transcribe import extract_subtitles
|
||||
|
||||
logger = logging.getLogger('pandora.' + __name__)
|
||||
|
||||
|
@ -17,24 +18,13 @@ logger = logging.getLogger('pandora.' + __name__)
|
|||
def transcribe(**kwargs):
|
||||
user = User.objects.get(username=kwargs["user"])
|
||||
item = Item.objects.get(public_id=kwargs["item"])
|
||||
subtitles = utils.get_by_key(layers, 'isSubtitles', True)
|
||||
subtitles = utils.get_by_key(settings.CONFIG['layers'], 'isSubtitles', True)['id']
|
||||
|
||||
gpu = getattr(settings, "TRANSCRIBE_GPU", False)
|
||||
join_sentences = getattr(settings, "TRANSCRIBE_JOIN", False)
|
||||
model = getattr(settings, "TRANSCRIBE_MODEL", "small")
|
||||
translate = getattr(settings, "TRANSCRIBE_TRANSLATE", "")
|
||||
logger.error(
|
||||
"transcribe %s %s %s %s %s %s %s",
|
||||
item,
|
||||
user,
|
||||
subtitles,
|
||||
translate,
|
||||
gpu,
|
||||
join_sentences,
|
||||
model
|
||||
)
|
||||
|
||||
transcribe.extract_subtitles(
|
||||
extract_subtitles(
|
||||
item,
|
||||
user,
|
||||
subtitles,
|
||||
|
|
6
views.py
6
views.py
|
@ -21,11 +21,11 @@ def transcribeAudio(request, data):
|
|||
item: id // item id
|
||||
}
|
||||
'''
|
||||
response = {}
|
||||
response = json_response()
|
||||
item = get_object_or_404_json(Item, public_id=data["item"])
|
||||
if has_capability(request.user, 'canTranscribeAudio') and item.editalbe(request.user):
|
||||
if has_capability(request.user, 'canTranscribeAudio') and item.editable(request.user):
|
||||
t = tasks.transcribe.delay(item=data["item"], user=request.user.username)
|
||||
response["taskId"] = t.task_id
|
||||
response["data"]["taskId"] = t.task_id
|
||||
else:
|
||||
response = json_response(status=403, text='permission denied')
|
||||
return render_to_json_response(response)
|
||||
|
|
Loading…
Add table
Reference in a new issue