104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
import os
|
|
import sys
|
|
import platform
|
|
import traceback
|
|
import subprocess
|
|
import sqlite3
|
|
|
|
from framework import db, path_app_root, path_data
|
|
from framework.logger import get_logger
|
|
from framework.util import Util
|
|
|
|
from .logic_normal import LogicNormal
|
|
from .model import ModelSetting
|
|
|
|
package_name = __name__.split(".", maxsplit=1)[0]
|
|
logger = get_logger(package_name)
|
|
|
|
|
|
class Logic(object):
|
|
db_default = {
|
|
"db_version": "2",
|
|
"youtube_dl_package": "1",
|
|
"ffmpeg_path": ""
|
|
if platform.system() != "Windows"
|
|
else os.path.join(path_app_root, "bin", "Windows", "ffmpeg.exe"),
|
|
"temp_path": os.path.join(path_data, "download_tmp"),
|
|
"save_path": os.path.join(path_data, "download"),
|
|
"default_filename": "",
|
|
"proxy": "",
|
|
}
|
|
|
|
@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()
|
|
Logic.migration()
|
|
except Exception as error:
|
|
logger.error("Exception:%s", error)
|
|
logger.error(traceback.format_exc())
|
|
|
|
@staticmethod
|
|
def plugin_load():
|
|
try:
|
|
logger.debug("%s plugin_load", package_name)
|
|
Logic.db_init()
|
|
|
|
# youtube-dl 업데이트
|
|
youtube_dl = LogicNormal.get_youtube_dl_package(
|
|
ModelSetting.get("youtube_dl_package")
|
|
)
|
|
logger.debug(f"{youtube_dl} upgrade")
|
|
logger.debug(
|
|
subprocess.check_output(
|
|
[sys.executable, "-m", "pip", "install", "--upgrade", youtube_dl],
|
|
universal_newlines=True,
|
|
)
|
|
)
|
|
|
|
# 편의를 위해 json 파일 생성
|
|
from .plugin import plugin_info
|
|
|
|
Util.save_from_dict_to_json(
|
|
plugin_info, os.path.join(os.path.dirname(__file__), "info.json")
|
|
)
|
|
except Exception as error:
|
|
logger.error("Exception:%s", error)
|
|
logger.error(traceback.format_exc())
|
|
|
|
@staticmethod
|
|
def plugin_unload():
|
|
try:
|
|
logger.debug("%s plugin_unload", package_name)
|
|
except Exception as error:
|
|
logger.error("Exception:%s", error)
|
|
logger.error(traceback.format_exc())
|
|
|
|
@staticmethod
|
|
def migration():
|
|
try:
|
|
db_version = ModelSetting.get_int("db_version")
|
|
connect = sqlite3.connect(
|
|
os.path.join(path_data, "db", f"{package_name}.db")
|
|
)
|
|
|
|
if db_version < 2:
|
|
logger.debug("youtube-dlc uninstall")
|
|
logger.debug(
|
|
subprocess.check_output(
|
|
[sys.executable, "-m", "pip", "uninstall", "-y", "youtube-dlc"],
|
|
universal_newlines=True,
|
|
)
|
|
)
|
|
|
|
connect.commit()
|
|
connect.close()
|
|
ModelSetting.set("db_version", Logic.db_default["db_version"])
|
|
db.session.flush()
|
|
except Exception as error:
|
|
logger.error("Exception:%s", error)
|
|
logger.error(traceback.format_exc())
|