Ejemplo n.º 1
0
def convert(url, client_ip=None):
    """
    Convert the YouTube video to MP3 audio.

    Steps:
        1. Get the video's information to make sure the provided url is valid.
        2. If info is returned and the duration is no more than
           MAX_DURATION_SECONDS, log the request and start the conversion.
        3. Return the download link if conversion completes successfully.
    """
    result = None
    duration = None

    info = get_video_info(url)
    if info:
        duration = info.get('duration')

    if info and duration and duration <= settings.MAX_DURATION_SECONDS:
        youtube_id = info['id']
        title = info['title']

        audio_filename = create_filename(info['title'])

        video, created = Video.objects.get_or_create(youtube_id=youtube_id)
        video.url = url
        video.title = title
        video.duration = duration
        video.save()

        ActivityLog.objects.create(
            video=video,
            client_ip=client_ip,
            action=ActivityLog.CONVERT,
        )

        result = {
            'youtube_id': youtube_id,
            'title': title
        }

        # Simply return the filename and update the video object if the file
        # already exists, otherwise, start the conversion.
        output_filepath = os.path.join(settings.MEDIA_ROOT, audio_filename)
        if os.path.exists(output_filepath):
            result['filename'] = audio_filename

            # Update the video object.
            video.audio_filename = audio_filename
            video.audio_filesize = os.path.getsize(output_filepath)
            video.save()
        else:
            conversion_result = start_conversion(url, audio_filename, video)

            # If extraction result is 0, extraction is successful.
            if conversion_result == 0:
                result['filename'] = audio_filename

    return result
Ejemplo n.º 2
0
def convert(url, client_ip=None):
    """
    Convert the YouTube video to MP3 audio.

    Steps:
        1. Get the video's information to make sure the provided url is valid.
        2. If info is returned and the duration is no more than
           MAX_DURATION_SECONDS, log the request and start the conversion.
        3. Return the download link if conversion completes successfully.
    """
    result = None
    duration = None

    info = get_video_info(url)
    if info:
        duration = info.get('duration')

    if info and duration and duration <= settings.MAX_DURATION_SECONDS:
        youtube_id = info['id']
        title = info['title']

        audio_filename = create_filename(info['title'])

        video, created = Video.objects.get_or_create(youtube_id=youtube_id)
        video.url = url
        video.title = title
        video.duration = duration
        video.save()

        ActivityLog.objects.create(
            video=video,
            client_ip=client_ip,
            action=ActivityLog.CONVERT,
        )

        result = {'youtube_id': youtube_id, 'title': title}

        # Simply return the filename and update the video object if the file
        # already exists, otherwise, start the conversion.
        output_filepath = os.path.join(settings.MEDIA_ROOT, audio_filename)
        if os.path.exists(output_filepath):
            result['filename'] = audio_filename

            # Update the video object.
            video.audio_filename = audio_filename
            video.audio_filesize = os.path.getsize(output_filepath)
            video.save()
        else:
            conversion_result = start_conversion(url, audio_filename, video)

            # If extraction result is 0, extraction is successful.
            if conversion_result == 0:
                result['filename'] = audio_filename

    return result