# -*- coding: utf-8 -*- ######################################################### # python import traceback from datetime import datetime # third-party from flask import jsonify # 패키지 from .plugin import logger from .my_youtube_dl import MyYoutubeDL, Status ######################################################### class LogicNormal(object): youtube_dl_list = [] @staticmethod def get_youtube_dl_version(): return MyYoutubeDL.get_version() @staticmethod def get_preset_list(): return [ ['bestvideo+bestaudio/best', '최고 화질'], ['bestvideo[height<=1080]+bestaudio/best[height<=1080]', '1080p'], ['worstvideo+worstaudio/worst', '최저 화질'], ['bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]', '최고 화질(mp4)'], ['bestvideo[ext=mp4][height<=1080]+bestaudio[ext=m4a]/best[ext=mp4][height<=1080]', '1080p(mp4)'], ['bestvideo[filesize<50M]+bestaudio/best[filesize<50M]', '50MB 미만'], ['bestaudio/best', '오디오만'], ['_custom', '사용자 정의'] ] @staticmethod def get_postprocessor_list(): return [ ['', '후처리 안함', None], ['mp4', 'MP4', '비디오 변환'], ['flv', 'FLV', '비디오 변환'], ['webm', 'WebM', '비디오 변환'], ['ogg', 'Ogg', '비디오 변환'], ['mkv', 'MKV', '비디오 변환'], ['ts', 'TS', '비디오 변환'], ['avi', 'AVI', '비디오 변환'], ['wmv', 'WMV', '비디오 변환'], ['mov', 'MOV', '비디오 변환'], ['gif', 'GIF', '비디오 변환'], ['mp3', 'MP3', '오디오 추출'], ['aac', 'AAC', '오디오 추출'], ['flac', 'FLAC', '오디오 추출'], ['m4a', 'M4A', '오디오 추출'], ['opus', 'Opus', '오디오 추출'], ['vorbis', 'Vorbis', '오디오 추출'], ['wav', 'WAV', '오디오 추출'] ] @staticmethod def get_postprocessor(): video_convertor = [] extract_audio = [] for i in LogicNormal.get_postprocessor_list(): if i[2] == '비디오 변환': video_convertor.append(i[0]) elif i[2] == '오디오 추출': extract_audio.append(i[0]) return video_convertor, extract_audio @staticmethod def download(**kwagrs): logger.debug(kwagrs) plugin = kwagrs['plugin'] url = kwagrs['url'] filename = kwagrs['filename'] temp_path = kwagrs['temp_path'] save_path = kwagrs['save_path'] opts = {} if 'format' in kwagrs and kwagrs['format']: opts['format'] = kwagrs['format'] postprocessor = [] if 'preferedformat' in kwagrs and kwagrs['preferedformat']: postprocessor.append({ 'key': 'FFmpegVideoConvertor', 'preferedformat': kwagrs['preferedformat'] }) if 'preferredcodec' in kwagrs and kwagrs['preferredcodec']: postprocessor.append({ 'key': 'FFmpegExtractAudio', 'preferredcodec': kwagrs['preferredcodec'], 'preferredquality': str(kwagrs['preferredquality']) }) if postprocessor: opts['postprocessors'] = postprocessor if 'archive' in kwagrs and kwagrs['archive']: opts['download_archive'] = kwagrs['archive'] if 'proxy' in kwagrs and kwagrs['proxy']: opts['proxy'] = kwagrs['proxy'] if 'ffmpeg_path' in kwagrs and kwagrs['ffmpeg_path']: opts['ffmpeg_location'] = kwagrs['ffmpeg_path'] youtube_dl = MyYoutubeDL(plugin, url, filename, temp_path, save_path, opts) youtube_dl.key = kwagrs.get('key') LogicNormal.youtube_dl_list.append(youtube_dl) # 리스트 추가 return youtube_dl @staticmethod def get_data(youtube_dl): try: data = {} data['plugin'] = youtube_dl.plugin data['url'] = youtube_dl.url data['filename'] = youtube_dl.filename data['temp_path'] = youtube_dl.temp_path data['save_path'] = youtube_dl.save_path data['index'] = youtube_dl.index data['status_str'] = youtube_dl.status.name data['status_ko'] = str(youtube_dl.status) data['end_time'] = '' data['extractor'] = youtube_dl.info_dict['extractor'] if youtube_dl.info_dict['extractor'] is not None else '' data['title'] = youtube_dl.info_dict['title'] if youtube_dl.info_dict['title'] is not None else youtube_dl.url data['uploader'] = youtube_dl.info_dict['uploader'] if youtube_dl.info_dict['uploader'] is not None else '' data['uploader_url'] = youtube_dl.info_dict['uploader_url'] if youtube_dl.info_dict['uploader_url'] is not None else '' data['downloaded_bytes_str'] = '' data['total_bytes_str'] = '' data['percent'] = '0' data['eta'] = youtube_dl.progress_hooks['eta'] if youtube_dl.progress_hooks['eta'] is not None else '' data['speed_str'] = LogicNormal.human_readable_size(youtube_dl.progress_hooks['speed'], '/s') if youtube_dl.progress_hooks['speed'] is not None else '' if youtube_dl.status == Status.READY: # 다운로드 전 data['start_time'] = '' data['download_time'] = '' else: if youtube_dl.end_time is None: # 완료 전 download_time = datetime.now() - youtube_dl.start_time else: download_time = youtube_dl.end_time - youtube_dl.start_time data['end_time'] = youtube_dl.end_time.strftime('%m-%d %H:%M:%S') if None not in (youtube_dl.progress_hooks['downloaded_bytes'], youtube_dl.progress_hooks['total_bytes']): # 둘 다 값이 있으면 data['downloaded_bytes_str'] = LogicNormal.human_readable_size(youtube_dl.progress_hooks['downloaded_bytes']) data['total_bytes_str'] = LogicNormal.human_readable_size(youtube_dl.progress_hooks['total_bytes']) data['percent'] = '%.2f' % (float(youtube_dl.progress_hooks['downloaded_bytes']) / float(youtube_dl.progress_hooks['total_bytes']) * 100) data['start_time'] = youtube_dl.start_time.strftime('%m-%d %H:%M:%S') data['download_time'] = '%02d:%02d' % (download_time.seconds / 60, download_time.seconds % 60) return data except Exception as e: logger.error('Exception:%s', e) logger.error(traceback.format_exc()) return None @staticmethod def get_info_dict(url, proxy): MyYoutubeDL.get_info_dict(url, proxy) @staticmethod def human_readable_size(size, suffix=''): for unit in ('Bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB'): if size < 1024.0: return '%3.1f %s%s' % (size, unit, suffix) size /= 1024.0 return '%.1f %s%s' % (size, 'YB', suffix) @staticmethod def abort(base, code): base['errorCode'] = code return jsonify(base)