def test_create_no_transcribe(self):
remote_media_file = settings.get('test', 'audio_test_remote_mp3')
self.model = Media.create(client=self.client, media_filename=remote_media_file, transcribe=False)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert Media.get(client=self.client, uuid=self.model.uuid).uuid == self.model.uuid
assert media_item.title == remote_media_file
assert media_item.status in (MEDIA_STATUS['UPLOAD'], MEDIA_STATUS['TRANSCODE'])
def test_create_no_transcribe(self):
remote_media_file = settings.get('test', 'audio_test_remote_mp3')
self.model = Media.create(client=self.client,
media_filename=remote_media_file,
transcribe=False)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert Media.get(client=self.client,
uuid=self.model.uuid).uuid == self.model.uuid
assert media_item.title == remote_media_file
assert media_item.status in (MEDIA_STATUS['UPLOAD'],
MEDIA_STATUS['TRANSCODE'])
def publish_unpublish(media_uuid):
"""
Publish and unpublish a media item's transcript.
Publishing a transcript will make it available to anyone that knowns its uuid
"""
media_item = Media(accept='text/xml', username=username, password=password, uid=media_uuid)
media_item.get()
#from xml.dom import minidom
#xmldoc = minidom.parseString(media_item.response.read())
# current_transcript_uuid = xmldoc.getElementsByTagName('currentTranscript')[0].getElementsByTagName('id')[0].firstChild.data
media_item.publish()
def anthrotranscribe(media_uuid):
"""
Launch transcription process for a given media
"""
register_openers()
media_item = Media.get(client=client, uuid=media_uuid)
media_item.anthrotranscribe(client=client)
def test_get_all(self):
local_media_files = ['test_mp4_short.mp4', 'test_64K_short.mp3']
for local_media_file in local_media_files:
media_filename = "%s/%s" % (settings.get(
'base', 'path.local.media'), local_media_file)
m = Media.create(
client=self.client,
media_filename=media_filename,
)
media = Media.get_all(client=self.client)
assert len(media) > 0
for m in media:
assert hasattr(m, 'title')
assert hasattr(m, '')
def anthrotranscribe(media_uuid):
"""
Launch transcription process for a given media
"""
register_openers()
media_item = Media.get(client=client, uuid=media_uuid)
media_item.anthrotranscribe(client=client)
def test_create_remote(self):
remote_media_files = [
settings.get('test', 'audio_test_remote_mp3'),
settings.get('test', 'audio_test_remote_youtube'),
settings.get('test', 'audio_test_remote_youtube_https'),
]
for remote_media_file in remote_media_files:
self.model = Media.create(
client=self.client,
media_filename=remote_media_file,
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert Media.get(client=self.client, uuid=self.model.uuid).uuid == self.model.uuid
assert media_item.title == remote_media_file
assert media_item.status == MEDIA_STATUS['ASR']
assert media_item.progress == 0
def test_init(self):
fields = {
'uuid': '69255493-583c-468c-9962-e5586f494027',
}
m = Media(fields=fields)
assert hasattr(m, 'uuid')
assert m.uuid == fields['uuid']
def test_create_local(self):
local_media_files = ['test_mp4_short.mp4', 'test_64K_short.mp3']
for local_media_file in local_media_files:
media_filename = "%s/%s" % (settings.get(
'base', 'path.local.media'), local_media_file)
self.model = Media.create(
client=self.client,
media_filename=media_filename,
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert media_item.uuid == self.model.uuid
assert media_item.title == "%s/%s" % (settings.get(
'base', 'path.local.media'), local_media_file)
assert media_item.status == MEDIA_STATUS['ASR']
assert media_item.progress == 0
def test_get_all(self):
local_media_files = [
'test_mp4_short.mp4',
'test_64K_short.mp3'
]
for local_media_file in local_media_files:
media_filename = "%s/%s" % (settings.get('base', 'path.local.media'), local_media_file)
m = Media.create(
client=self.client,
media_filename=media_filename,
)
media = Media.get_all(client=self.client)
assert len(media) > 0
for m in media:
assert hasattr(m, 'title')
assert hasattr(m, '')
def test_create_local(self):
local_media_files = [
'test_mp4_short.mp4',
'test_64K_short.mp3'
]
for local_media_file in local_media_files:
media_filename = "%s/%s" % (settings.get('base', 'path.local.media'), local_media_file)
self.model = Media.create(
client=self.client,
media_filename=media_filename,
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert media_item.uuid == self.model.uuid
assert media_item.title == "%s/%s" % (settings.get('base', 'path.local.media'), local_media_file)
assert media_item.status == MEDIA_STATUS['ASR']
assert media_item.progress == 0
def download_transcript(media_uuid):
media_item = Media.get(client=client, uuid=media_uuid)
print media_item.current_transcript
transcript = Transcript.get(client=client, uuid=media_item.current_transcript['uuid'], format='srt')
if not os.path.exists(settings.get('base','path.local.scripts.output')):
os.makedirs(settings.get('base','path.local.scripts.output'))
f = open("%s/%s.srt" % (settings.get('base','path.local.scripts.output'), media_item.title), 'w')
f.write(transcript.content)
def test_create_remote(self):
remote_media_files = [
settings.get('test', 'audio_test_remote_mp3'),
settings.get('test', 'audio_test_remote_youtube'),
settings.get('test', 'audio_test_remote_youtube_https'),
]
for remote_media_file in remote_media_files:
self.model = Media.create(
client=self.client,
media_filename=remote_media_file,
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert Media.get(client=self.client,
uuid=self.model.uuid).uuid == self.model.uuid
assert media_item.title == remote_media_file
assert media_item.status == MEDIA_STATUS['ASR']
assert media_item.progress == 0
def test_get(self):
"""
Create a media and do a get request on it
"""
fields = {
'title': 'Test media title',
'description': 'Test media description',
'local_media_file': 'test_64K_short.mp3',
}
media_filename = "%s/%s" % (settings.get('base', 'path.local.media'), fields['local_media_file'])
m = Media.create(
client=self.client,
media_filename=media_filename,
title=fields['title'],
description=fields['description'],
)
m = Media.get(client=self.client, uuid=m.uuid)
assert m.title == fields['title']
assert m.description == fields['description']
def test_align_remote(self):
media_item = Media.create(
client=self.client,
media_filename=settings.get('test', 'audio_test_remote_mp3'),
aligndata="%s/%s" % (settings.get('base', 'path.local.transcripts'), settings.get('test', 'transcript.align')),
transcribe=False
)
assert hasattr(media_item, 'process_alignment')
assert media_item.process_alignment.status == 'PENDING'
assert media_item.process_alignment.progress == 0
assert hasattr(media_item.process_alignment, 'uuid')
def test_get(self):
"""
Create a media and do a get request on it
"""
fields = {
'title': 'Test media title',
'description': 'Test media description',
'local_media_file': 'test_64K_short.mp3',
}
media_filename = "%s/%s" % (settings.get(
'base', 'path.local.media'), fields['local_media_file'])
m = Media.create(
client=self.client,
media_filename=media_filename,
title=fields['title'],
description=fields['description'],
)
m = Media.get(client=self.client, uuid=m.uuid)
assert m.title == fields['title']
assert m.description == fields['description']
def test_align_remote(self):
media_item = Media.create(
client=self.client,
media_filename=settings.get('test', 'audio_test_remote_mp3'),
aligndata="%s/%s" %
(settings.get('base', 'path.local.transcripts'),
settings.get('test', 'transcript.align')),
transcribe=False)
assert hasattr(media_item, 'process_alignment')
assert media_item.process_alignment.status == 'PENDING'
assert media_item.process_alignment.progress == 0
assert hasattr(media_item.process_alignment, 'uuid')
def test_create_local_metadata(self):
"""
Test creation of a media from a local file, by specifying some metadata (title, description, ...)
"""
local_media = {
'path': 'test_mp4_short.mp4',
'title': 'Test media title',
'description': 'Test media description',
}
media_filename = "%s/%s" % (settings.get('base', 'path.local.media'), local_media['path'])
self.model = Media.create(
client=self.client,
media_filename=media_filename,
title=local_media['title'],
description=local_media['description'],
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert media_item.title == local_media['title']
assert media_item.description == local_media['description']
def test_get_all(self):
local_media_files = [
'test_mp4_short.mp4',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3',
'test_64K_short.mp3'
]
for local_media_file in local_media_files:
media_filename = "%s/%s" % (settings.get('base', 'path.local.media'), local_media_file)
m = Media.create(
client=self.client,
media_filename=media_filename,
)
# vanilla test
media = Media.get_all(client=self.client)
assert len(media) > 0
for m in media:
assert hasattr(m, 'title')
assert len(media) == 10
# test pagination
page_size = 5
media = Media.get_all(client=self.client, start=0, count=page_size)
assert len(media) == page_size
media2 = Media.get_all(client=self.client, start=page_size-1, count=page_size)
assert media[page_size-1].uuid == media2[0].uuid
def upload_transcribe(media_filename):
"""
Upload a media file to Koemei for transcription
"""
try:
log.info("Upload and transcribe file %s ..." % media_filename)
media_item = Media.create(client=client, media_filename=media_filename)
log.info("... OK - media uuid: %s" % media_item.uuid)
except Exception, e:
log.error("... Error creating media %s ..." % media_filename)
log.error(e)
log.error(traceback.format_exc())
raise e
def upload_transcribe(media_filename):
"""
Upload a media file to Koemei for transcription
"""
try:
log.info("Upload and transcribe file %s ..." % media_filename)
media_item = Media.create(client=client, media_filename=media_filename)
log.info("... OK - media uuid: %s" % media_item.uuid)
except Exception, e:
log.error("... Error creating media %s ..." % media_filename)
log.error(e)
log.error(traceback.format_exc())
raise e
def test_create_local_metadata(self):
"""
Test creation of a media from a local file, by specifying some metadata (title, description, ...)
"""
local_media = {
'path': 'test_mp4_short.mp4',
'title': 'Test media title',
'description': 'Test media description',
}
media_filename = "%s/%s" % (settings.get(
'base', 'path.local.media'), local_media['path'])
self.model = Media.create(
client=self.client,
media_filename=media_filename,
title=local_media['title'],
description=local_media['description'],
)
media_item = Media.get(client=self.client, uuid=self.model.uuid)
assert media_item.title == local_media['title']
assert media_item.description == local_media['description']
def download_transcript(media_uuid):
media_item = Media.get(client=client, uuid=media_uuid)
print media_item.current_transcript
transcript = Transcript.get(client=client,
uuid=media_item.current_transcript['uuid'],
format='srt')
if not os.path.exists(settings.get('base', 'path.local.scripts.output')):
os.makedirs(settings.get('base', 'path.local.scripts.output'))
f = open(
"%s/%s.srt" %
(settings.get('base', 'path.local.scripts.output'), media_item.title),
'w')
f.write(transcript.content)
def publish_unpublish(media_uuid):
"""
Publish and unpublish a media item's transcript.
Publishing a transcript will make it available to anyone that knowns its uuid
"""
media_item = Media(accept='text/xml',
username=username,
password=password,
uid=media_uuid)
media_item.get()
#from xml.dom import minidom
#xmldoc = minidom.parseString(media_item.response.read())
# current_transcript_uuid = xmldoc.getElementsByTagName('currentTranscript')[0].getElementsByTagName('id')[0].firstChild.data
media_item.publish()
def upload_align(media_filename, aligndata):
"""
Upload a media file to Koemei for alignment
NOTE : you will need your account to be specially setup to use this feature
"""
try:
log.info("Upload and align media %s ..." % media_filename)
media_item = Media.create(client=client,
media_filename=media_filename,
aligndata=aligndata,
transcribe=False)
log.info("... OK - media uuid: %s" % media_item.uuid)
except Exception, e:
log.error("... Error aligning media %s ..." % media_filename)
log.error(e)
log.error(traceback.format_exc())
raise e
def upload_align(media_filename, aligndata):
"""
Upload a media file to Koemei for alignment
NOTE : you will need your account to be specially setup to use this feature
"""
register_openers()
try:
log.info("Upload and align media %s ..." % media_filename)
media_item = Media.create(
client=client,
media_filename=media_filename,
aligndata=aligndata,
transcribe=False
)
log.info("... OK - media uuid: %s" % media_item.uuid)
except Exception, e:
log.error("... Error aligning media %s ..." % media_filename)
log.error(e)
log.error(traceback.format_exc())
raise e
def delete_media():
media_item = Media.get(client=client, uuid='28b5e028-fddf-4de2-a9ec-6b222f39e215')
KObject.delete(client=client, uuid=media_item.kobject_uuid, delete_transcripts=True)
def delete_media():
media_item = Media.get(client=client,
uuid='28b5e028-fddf-4de2-a9ec-6b222f39e215')
KObject.delete(client=client,
uuid=media_item.kobject_uuid,
delete_transcripts=True)
