모델이 티어에 따라 2~17GB 를 차지하는데 %APPDATA% 에 박혀 있어서 C 드라이브가 좁은 사람은 손쓸 방법이 없었다. 처음 켤 때 한 번 묻고, 그 선택을 기억하고, 설정에서 바꿀 수 있게 한다. 닭과 달걀 문제가 하나 있다 — "어디에 둘지" 는 사용자가 정하는데 그 선택 자체는 고정된 곳에 적어둬야 다음 실행에 찾을 수 있다. 그래서 OS 표준 폴더에는 위치를 가리키는 쪽지(location.json) 한 장만 두고, 내용물은 전부 고른 폴더에 둔다. - storage.py 신설: 쪽지 읽기/쓰기, 경로 검증, 여유 공간, 데이터 이동 - config.py 의 CONFIG_PATH 를 config_path() 로 바꾼다. import 시점 상수라 위치를 바꿔도 옛 경로를 계속 가리켰다 - 위치가 정해지기 전에는 파일 로그를 열지 않는다. 열면 고르지도 않은 기본 폴더가 만들어지고 거기에 로그가 남는다 - 이전 버전 사용자에게는 묻지 않는다. 기본 위치에 쓰던 흔적이 있으면 그대로 쓴다 — 같은 자리를 다시 고르는 헛수고를 시킬 이유가 없다 - 고른 디스크가 빠져도 죽지 않고 기본 위치로 물러난다 - 위치를 바꿀 때 받아둔 모델을 옮길지 물어본다. 수 GB 라 창이 얼어붙지 않도록 별도 스레드에서 옮긴다 - windows_smoke 에 이 화면 렌더링 점검을 추가하고, 스크린샷을 호스트로 회수한다. 경로 표기와 디스크 조회는 OS 마다 다르게 도는 부분이다
This commit is contained in:
EJClaw
211
src/livesub/ui/data_location.py
Normal file
211
src/livesub/ui/data_location.py
Normal file
@@ -0,0 +1,211 @@
|
||||
"""저장 위치를 고르는 화면.
|
||||
|
||||
첫 실행 때 한 번 묻고, 그 뒤로는 설정에서 바꿀 수 있다. 두 경우가 묻는 내용이
|
||||
같으므로 같은 대화상자를 쓰고 문구만 바꾼다.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
from PySide6.QtCore import Qt, QThread, Signal
|
||||
from PySide6.QtWidgets import (
|
||||
QDialog,
|
||||
QDialogButtonBox,
|
||||
QFileDialog,
|
||||
QHBoxLayout,
|
||||
QLabel,
|
||||
QLineEdit,
|
||||
QMessageBox,
|
||||
QProgressDialog,
|
||||
QPushButton,
|
||||
QVBoxLayout,
|
||||
QWidget,
|
||||
)
|
||||
|
||||
from ..constants import APP_NAME
|
||||
from ..storage import (
|
||||
DATA_ENTRIES,
|
||||
check_target,
|
||||
data_dir,
|
||||
default_data_dir,
|
||||
free_space,
|
||||
human_size,
|
||||
move_data,
|
||||
set_data_dir,
|
||||
)
|
||||
|
||||
#: 가장 무거운 티어까지 받았을 때를 기준으로 한 권장 여유 공간.
|
||||
RECOMMENDED_FREE = 20 * 1024**3
|
||||
|
||||
|
||||
class DataLocationDialog(QDialog):
|
||||
"""폴더 하나를 고르고, 쓸 수 있는 곳인지 즉시 알려준다."""
|
||||
|
||||
def __init__(self, start: Path, *, first_run: bool, parent: QWidget | None = None):
|
||||
super().__init__(parent)
|
||||
self.setWindowTitle(f"{APP_NAME} 저장 위치")
|
||||
self.setMinimumWidth(520)
|
||||
self._first_run = first_run
|
||||
self._current = None if first_run else start
|
||||
|
||||
root = QVBoxLayout(self)
|
||||
root.setSpacing(12)
|
||||
|
||||
title = QLabel("어디에 저장할까요?" if first_run else "저장 위치 바꾸기")
|
||||
title.setStyleSheet("font-size: 16px; font-weight: 700;")
|
||||
root.addWidget(title)
|
||||
|
||||
why = QLabel(
|
||||
"음성인식·번역 모델과 설정, 번역 기록이 이 폴더에 저장됩니다.\n"
|
||||
"모델은 품질 티어에 따라 2~16GB 를 차지하니 여유 있는 디스크를 고르세요.\n"
|
||||
"나중에 설정에서 언제든 바꿀 수 있습니다."
|
||||
)
|
||||
why.setObjectName("Hint")
|
||||
why.setWordWrap(True)
|
||||
root.addWidget(why)
|
||||
|
||||
row = QWidget()
|
||||
row.setObjectName("Row")
|
||||
row_layout = QHBoxLayout(row)
|
||||
row_layout.setContentsMargins(0, 0, 0, 0)
|
||||
self.path_edit = QLineEdit(str(start))
|
||||
self.path_edit.textChanged.connect(self._revalidate)
|
||||
browse = QPushButton("찾아보기…")
|
||||
browse.clicked.connect(self._browse)
|
||||
row_layout.addWidget(self.path_edit, 1)
|
||||
row_layout.addWidget(browse)
|
||||
root.addWidget(row)
|
||||
|
||||
self.status = QLabel()
|
||||
self.status.setWordWrap(True)
|
||||
root.addWidget(self.status)
|
||||
|
||||
self.buttons = QDialogButtonBox()
|
||||
ok_text = "이 위치로 시작" if first_run else "이 위치로 바꾸기"
|
||||
self.ok_button = self.buttons.addButton(ok_text, QDialogButtonBox.ButtonRole.AcceptRole)
|
||||
cancel_text = "종료" if first_run else "취소"
|
||||
self.buttons.addButton(cancel_text, QDialogButtonBox.ButtonRole.RejectRole)
|
||||
self.buttons.accepted.connect(self.accept)
|
||||
self.buttons.rejected.connect(self.reject)
|
||||
root.addWidget(self.buttons)
|
||||
|
||||
self._revalidate()
|
||||
|
||||
def selected(self) -> Path:
|
||||
return Path(self.path_edit.text().strip()).expanduser()
|
||||
|
||||
def _browse(self) -> None:
|
||||
start = self.path_edit.text().strip() or str(default_data_dir())
|
||||
picked = QFileDialog.getExistingDirectory(self, "저장할 폴더 선택", start)
|
||||
if picked:
|
||||
self.path_edit.setText(picked)
|
||||
|
||||
def _revalidate(self) -> None:
|
||||
text = self.path_edit.text().strip()
|
||||
problem = check_target(text, self._current)
|
||||
self.ok_button.setEnabled(problem is None)
|
||||
if problem is not None:
|
||||
self.status.setText(f"⚠ {problem}")
|
||||
self.status.setStyleSheet("color: #E5484D;")
|
||||
return
|
||||
|
||||
free = free_space(text)
|
||||
line = f"이 디스크의 남은 공간 {human_size(free)}"
|
||||
if free and free < RECOMMENDED_FREE:
|
||||
self.status.setText(
|
||||
f"{line} — 높은 티어 모델까지 받기에는 빠듯합니다. 그래도 진행할 수 있습니다."
|
||||
)
|
||||
self.status.setStyleSheet("color: #F5A623;")
|
||||
else:
|
||||
self.status.setText(line)
|
||||
self.status.setStyleSheet("")
|
||||
|
||||
|
||||
def ask_initial_location(parent: QWidget | None = None) -> bool:
|
||||
"""첫 실행 — 위치를 고르게 한다. 사용자가 종료를 누르면 False."""
|
||||
dialog = DataLocationDialog(default_data_dir(), first_run=True, parent=parent)
|
||||
if dialog.exec() != QDialog.DialogCode.Accepted:
|
||||
return False
|
||||
set_data_dir(dialog.selected())
|
||||
return True
|
||||
|
||||
|
||||
def ask_change_location(parent: QWidget | None = None) -> Path | None:
|
||||
"""설정 — 위치를 바꾼다. 바꿨으면 새 경로, 취소했으면 None."""
|
||||
current = data_dir()
|
||||
dialog = DataLocationDialog(current, first_run=False, parent=parent)
|
||||
if dialog.exec() != QDialog.DialogCode.Accepted:
|
||||
return None
|
||||
|
||||
target = dialog.selected()
|
||||
if target.expanduser().resolve() == current.resolve():
|
||||
return None
|
||||
|
||||
if _has_data(current) and not _move_with_progress(current, target, parent):
|
||||
return None
|
||||
|
||||
return set_data_dir(target)
|
||||
|
||||
|
||||
def _has_data(path: Path) -> bool:
|
||||
return any((path / name).exists() for name in DATA_ENTRIES)
|
||||
|
||||
|
||||
def _move_with_progress(old: Path, new: Path, parent: QWidget | None) -> bool:
|
||||
"""기존 데이터를 옮길지 묻고, 옮기기로 했으면 옮긴다. 취소면 False."""
|
||||
answer = QMessageBox.question(
|
||||
parent,
|
||||
"기존 데이터",
|
||||
f"지금 위치에 받아둔 모델과 설정이 있습니다.\n\n"
|
||||
f"{old}\n\n"
|
||||
"새 위치로 옮길까요? 옮기지 않으면 모델을 다시 받아야 합니다.",
|
||||
QMessageBox.StandardButton.Yes
|
||||
| QMessageBox.StandardButton.No
|
||||
| QMessageBox.StandardButton.Cancel,
|
||||
QMessageBox.StandardButton.Yes,
|
||||
)
|
||||
if answer == QMessageBox.StandardButton.Cancel:
|
||||
return False
|
||||
if answer == QMessageBox.StandardButton.No:
|
||||
return True
|
||||
|
||||
# 수 GB 를 옮기는 동안 창이 얼어붙으면 Windows 가 "응답 없음"을 띄운다.
|
||||
# 별도 스레드로 옮기고 여기서는 진행 표시만 돌린다.
|
||||
worker = _MoveWorker(old, new)
|
||||
progress = QProgressDialog("기존 데이터를 옮기는 중입니다…", "", 0, 0, parent)
|
||||
progress.setWindowTitle("저장 위치 변경")
|
||||
progress.setCancelButton(None) # 옮기는 도중에 끊으면 데이터가 갈라진다
|
||||
progress.setWindowModality(Qt.WindowModality.WindowModal)
|
||||
worker.finished.connect(progress.close)
|
||||
worker.start()
|
||||
progress.exec()
|
||||
worker.wait()
|
||||
|
||||
if worker.error:
|
||||
QMessageBox.warning(
|
||||
parent,
|
||||
"옮기지 못했습니다",
|
||||
f"기존 데이터를 옮기는 중 문제가 생겼습니다.\n\n{worker.error}\n\n"
|
||||
"저장 위치는 바꾸지 않았습니다.",
|
||||
)
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
class _MoveWorker(QThread):
|
||||
done = Signal()
|
||||
|
||||
def __init__(self, old: Path, new: Path):
|
||||
super().__init__()
|
||||
self._old = old
|
||||
self._new = new
|
||||
self.moved: list[str] = []
|
||||
self.error = ""
|
||||
|
||||
def run(self) -> None:
|
||||
try:
|
||||
self.moved = move_data(self._old, self._new)
|
||||
except OSError as exc:
|
||||
self.error = str(exc)
|
||||
self.done.emit()
|
||||
@@ -9,6 +9,7 @@ from PySide6.QtWidgets import (
|
||||
QHBoxLayout,
|
||||
QLabel,
|
||||
QLineEdit,
|
||||
QMessageBox,
|
||||
QPushButton,
|
||||
QSlider,
|
||||
QSpinBox,
|
||||
@@ -19,6 +20,7 @@ from PySide6.QtWidgets import (
|
||||
from ...config import AppConfig
|
||||
from ...constants import APP_NAME, APP_SLOGAN, APP_VERSION, user_data_dir
|
||||
from ...models import detect_gpu
|
||||
from ...storage import free_space, human_size
|
||||
from ..theme import SPACING, palette
|
||||
from ..widgets import Card, PageHeader
|
||||
|
||||
@@ -45,6 +47,7 @@ class SettingsPage(QWidget):
|
||||
root.addWidget(self._build_hotkey_card())
|
||||
root.addWidget(self._build_audio_card())
|
||||
root.addWidget(self._build_behavior_card())
|
||||
root.addWidget(self._build_storage_card())
|
||||
root.addWidget(self._build_about_card())
|
||||
root.addStretch(1)
|
||||
|
||||
@@ -258,20 +261,61 @@ class SettingsPage(QWidget):
|
||||
self.log_check.setChecked(self.config.log_transcripts)
|
||||
self.log_check.toggled.connect(self._on_behavior_changed)
|
||||
form.addRow("", self.log_check)
|
||||
|
||||
open_folder = QPushButton("설정 폴더 열기")
|
||||
open_folder.clicked.connect(self._open_folder)
|
||||
form.addRow("", open_folder)
|
||||
return card
|
||||
|
||||
# --- 저장 위치 -------------------------------------------------------
|
||||
def _build_storage_card(self) -> Card:
|
||||
card = Card(
|
||||
"저장 위치",
|
||||
"모델과 설정, 번역 기록이 여기에 저장됩니다. 모델은 품질 티어에 따라 "
|
||||
"2~16GB 를 차지하므로 여유 있는 디스크가 좋습니다.",
|
||||
)
|
||||
self.storage_label = QLabel()
|
||||
self.storage_label.setObjectName("Hint")
|
||||
self.storage_label.setWordWrap(True)
|
||||
self.storage_label.setTextInteractionFlags(
|
||||
Qt.TextInteractionFlag.TextSelectableByMouse
|
||||
)
|
||||
card.add(self.storage_label)
|
||||
|
||||
row = QWidget()
|
||||
row_layout = QHBoxLayout(row)
|
||||
row_layout.setContentsMargins(0, 0, 0, 0)
|
||||
change = QPushButton("위치 변경…")
|
||||
change.clicked.connect(self._change_storage)
|
||||
open_folder = QPushButton("폴더 열기")
|
||||
open_folder.clicked.connect(self._open_folder)
|
||||
row_layout.addWidget(change)
|
||||
row_layout.addWidget(open_folder)
|
||||
row_layout.addStretch(1)
|
||||
card.add(row)
|
||||
|
||||
self._refresh_storage_label()
|
||||
return card
|
||||
|
||||
def _refresh_storage_label(self) -> None:
|
||||
path = user_data_dir()
|
||||
self.storage_label.setText(f"{path}\n이 디스크의 남은 공간 {human_size(free_space(path))}")
|
||||
|
||||
def _change_storage(self) -> None:
|
||||
from ..data_location import ask_change_location
|
||||
|
||||
target = ask_change_location(self)
|
||||
if target is None:
|
||||
return
|
||||
# 옮기지 않기를 골랐다면 새 위치에 설정 파일이 없다. 지금 상태를 써둔다.
|
||||
self.config.save()
|
||||
self._refresh_storage_label()
|
||||
QMessageBox.information(
|
||||
self,
|
||||
"저장 위치를 바꿨습니다",
|
||||
f"이제 여기를 씁니다.\n\n{target}\n\n"
|
||||
"이미 메모리에 올라가 있는 모델은 그대로 쓰고, 다음에 받는 것부터 "
|
||||
"새 위치에 저장됩니다. 로그 파일은 다음 실행부터 옮겨집니다.",
|
||||
)
|
||||
|
||||
def _build_about_card(self) -> Card:
|
||||
card = Card(f"{APP_NAME} v{APP_VERSION}", APP_SLOGAN)
|
||||
path = QLabel(f"설정 위치: {user_data_dir()}")
|
||||
path.setObjectName("Hint")
|
||||
path.setTextInteractionFlags(Qt.TextInteractionFlag.TextSelectableByMouse)
|
||||
path.setWordWrap(True)
|
||||
card.add(path)
|
||||
return card
|
||||
return Card(f"{APP_NAME} v{APP_VERSION}", APP_SLOGAN)
|
||||
|
||||
# --- 동작 -----------------------------------------------------------
|
||||
def _on_audio_changed(self, *_) -> None:
|
||||
|
||||
Reference in New Issue
Block a user