178 lines
7.7 KiB
Python
178 lines
7.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
#########################################################
|
||
|
|
# python
|
||
|
|
import os
|
||
|
|
from datetime import datetime
|
||
|
|
import traceback
|
||
|
|
import logging
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
import re
|
||
|
|
import threading
|
||
|
|
import json
|
||
|
|
import enum
|
||
|
|
import platform
|
||
|
|
|
||
|
|
# third-party
|
||
|
|
from sqlalchemy import desc
|
||
|
|
from sqlalchemy import or_, and_, func, not_
|
||
|
|
|
||
|
|
# sjva 공용
|
||
|
|
from framework.logger import get_logger
|
||
|
|
from framework import app, db, scheduler, path_app_root, socketio, path_data
|
||
|
|
from framework.job import Job
|
||
|
|
from framework.util import Util
|
||
|
|
from system.logic import SystemLogic
|
||
|
|
|
||
|
|
# 패키지
|
||
|
|
from .model import ModelSetting
|
||
|
|
|
||
|
|
|
||
|
|
# 로그
|
||
|
|
package_name = __name__.split('.')[0]
|
||
|
|
logger = get_logger(package_name)
|
||
|
|
#########################################################
|
||
|
|
|
||
|
|
class Status(enum.Enum):
|
||
|
|
READY = 0
|
||
|
|
WRONG_URL = 1
|
||
|
|
WRONG_DIRECTORY = 2
|
||
|
|
EXCEPTION = 3
|
||
|
|
ERROR = 4
|
||
|
|
HTTP_FORBIDDEN = 11
|
||
|
|
|
||
|
|
DOWNLOADING = 5
|
||
|
|
|
||
|
|
USER_STOP = 6
|
||
|
|
COMPLETED = 7
|
||
|
|
TIME_OVER = 8
|
||
|
|
PF_STOP = 9
|
||
|
|
FORCE_STOP = 10 #강제중단
|
||
|
|
ALREADY_DOWNLOADING = 12 #이미 목록에 있고 다운로드중
|
||
|
|
|
||
|
|
def __int__(self):
|
||
|
|
return self.value
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
kor = [r'준비', r'URL에러', r'폴더에러', r'실패(Exception)', r'실패(에러)', r'다운로드중', r'사용자중지', r'완료', r'시간초과', r'PF중지', r'강제중지',
|
||
|
|
r'403에러', r'임시파일이 이미 있음']
|
||
|
|
return kor[int(self)]
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_instance(value):
|
||
|
|
tmp = [ Status.READY, Status.WRONG_URL, Status.WRONG_DIRECTORY, Status.EXCEPTION, Status.ERROR,
|
||
|
|
Status.DOWNLOADING, Status.USER_STOP, Status.COMPLETED, Status.TIME_OVER, Status.PF_STOP,
|
||
|
|
Status.FORCE_STOP, Status.HTTP_FORBIDDEN, Status.ALREADY_DOWNLOADING ]
|
||
|
|
return tmp[value]
|
||
|
|
|
||
|
|
class Logic(object):
|
||
|
|
db_default = {
|
||
|
|
'temp_path' : os.path.join(path_data, 'tmp'),
|
||
|
|
'save_path' : os.path.join(path_data, 'download'),
|
||
|
|
'max_pf_count' : '0',
|
||
|
|
'if_fail_remove_tmp_file' : 'True',
|
||
|
|
'timeout_minute' : '60',
|
||
|
|
'ffmpeg_path' : 'ffmpeg' if platform.system() != 'Windows' else os.path.join(path_data, 'bin', 'ffmpeg.exe'),
|
||
|
|
}
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def db_init():
|
||
|
|
try:
|
||
|
|
for key, value in Logic.db_default.items():
|
||
|
|
if db.session.query(ModelSetting).filter_by(key=key).count() == 0:
|
||
|
|
db.session.add(ModelSetting(key, value))
|
||
|
|
db.session.commit()
|
||
|
|
except Exception as exception:
|
||
|
|
logger.error('Exception:%s', exception)
|
||
|
|
logger.error(traceback.format_exc())
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def plugin_load():
|
||
|
|
try:
|
||
|
|
Logic.db_init()
|
||
|
|
except Exception as exception:
|
||
|
|
logger.error('Exception:%s', exception)
|
||
|
|
logger.error(traceback.format_exc())
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def plugin_unload():
|
||
|
|
try:
|
||
|
|
from ffmpeg.interface_program_ffmpeg import Ffmpeg
|
||
|
|
Ffmpeg.plugin_unload()
|
||
|
|
except Exception as exception:
|
||
|
|
logger.error('Exception:%s', exception)
|
||
|
|
logger.error(traceback.format_exc())
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def ffmpeg_listener(**args):
|
||
|
|
#logger.debug('ffmpeg_listener : %s', args)
|
||
|
|
#logger.debug('ffmpeg_listener : type:%s status:%s', args['type'], args['status'])
|
||
|
|
# status_change ready, downloading, completed
|
||
|
|
refresh_type = None
|
||
|
|
if args['type'] == 'status_change':
|
||
|
|
if args['status'] == Status.DOWNLOADING:
|
||
|
|
refresh_type = 'status_change'
|
||
|
|
elif args['status'] == Status.COMPLETED:
|
||
|
|
refresh_type = 'status_change'
|
||
|
|
elif args['status'] == Status.READY:
|
||
|
|
data = {'type':'info', 'msg' : u'다운로드중 Duration(%s)' % args['data']['duration_str'] + '<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
# 1번에서 리스트화면, 2번에서 추가시 1번도 추가되도록
|
||
|
|
refresh_type = 'add'
|
||
|
|
elif args['type'] == 'last':
|
||
|
|
|
||
|
|
if args['status'] == Status.WRONG_URL:
|
||
|
|
data = {'type':'warning', 'msg' : u'잘못된 URL입니다'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'add'
|
||
|
|
elif args['status'] == Status.WRONG_DIRECTORY:
|
||
|
|
data = {'type':'warning', 'msg' : u'잘못된 디렉토리입니다.<br>' + args['data']['save_fullpath']}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'add'
|
||
|
|
elif args['status'] == Status.ERROR or args['status'] == Status.EXCEPTION:
|
||
|
|
data = {'type':'warning', 'msg' : u'다운로드 시작 실패.<br>' + args['data']['save_fullpath']}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'add'
|
||
|
|
elif args['status'] == Status.USER_STOP:
|
||
|
|
data = {'type':'warning', 'msg' : u'다운로드가 중지 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.COMPLETED:
|
||
|
|
data = {'type':'success', 'msg' : u'다운로드가 완료 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.TIME_OVER:
|
||
|
|
data = {'type':'warning', 'msg' : u'시간초과로 중단 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.PF_STOP:
|
||
|
|
data = {'type':'warning', 'msg' : u'PF초과로 중단 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.FORCE_STOP:
|
||
|
|
data = {'type':'warning', 'msg' : u'강제 중단 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.HTTP_FORBIDDEN:
|
||
|
|
data = {'type':'warning', 'msg' : u'403에러로 중단 되었습니다.<br>' + args['data']['save_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
elif args['status'] == Status.ALREADY_DOWNLOADING:
|
||
|
|
data = {'type':'warning', 'msg' : u'임시파일폴더에 파일이 있습니다.<br>' + args['data']['temp_fullpath'], 'url':'/ffmpeg/list'}
|
||
|
|
socketio.emit("notify", data, namespace='/framework', broadcast=True)
|
||
|
|
refresh_type = 'last'
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#elif args['type'] == 'log':
|
||
|
|
# socketio.emit("log", args['data'], namespace='/%s' % package_name, broadcast=True)
|
||
|
|
|
||
|
|
elif args['type'] == 'normal':
|
||
|
|
if args['status'] == Status.DOWNLOADING:
|
||
|
|
refresh_type = 'status'
|
||
|
|
|
||
|
|
if refresh_type is not None:
|
||
|
|
socketio.emit(refresh_type, args['data'], namespace='/%s' % package_name, broadcast=True)
|