mirror of
https://github.com/3minbe/DBC_Converter.git
synced 2026-05-17 01:23:58 +09:00
379 lines
17 KiB
Python
379 lines
17 KiB
Python
import os
|
|
import sys
|
|
from PyQt5 import QtWidgets, QtGui, QtCore
|
|
from datetime import datetime
|
|
import json
|
|
import subprocess
|
|
|
|
class MainView(QtWidgets.QMainWindow):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setWindowIcon(QtGui.QIcon("icon/icon.ico")) # 프로그램 아이콘 설정
|
|
self.version = "1.0.0" # 프로그램 버전 설정
|
|
self.default_save_path = os.path.join(os.path.expanduser("~"), "Desktop") # 바탕화면 경로 설정
|
|
self.file_paths = [] # 파일 경로 저장 리스트
|
|
self.loadSettings() # 설정 로드
|
|
self.setupUI() # UI 설정
|
|
self.centerWindow() # 창을 화면 중앙에 위치
|
|
self.sortTreeView(0, True) # 기본 파일명 오름차순 정렬
|
|
self.channel_options = ["1", "2", "3", "4", "5", "6"] # 채널 옵션 설정
|
|
|
|
def loadSettings(self):
|
|
self.settings_file = "settings.json"
|
|
if os.path.exists(self.settings_file):
|
|
with open(self.settings_file, "r", encoding="utf-8") as file:
|
|
self.settings = json.load(file) # 설정 파일 로드
|
|
self.default_save_path = self.settings.get("default_save_path", self.default_save_path)
|
|
self.file_paths = self.settings.get("file_paths", [])
|
|
self.last_opened_dir = self.settings.get("last_opened_dir", os.path.expanduser("~"))
|
|
else:
|
|
self.settings = {"theme": "light"} # 기본 설정
|
|
self.last_opened_dir = os.path.expanduser("~")
|
|
|
|
def saveSettings(self):
|
|
self.settings["default_save_path"] = self.default_save_path
|
|
self.settings["file_paths"] = self.file_paths
|
|
self.settings["last_opened_dir"] = self.last_opened_dir
|
|
with open(self.settings_file, "w", encoding="utf-8") as file:
|
|
json.dump(self.settings, file, ensure_ascii=False, indent=4) # 설정 파일 저장
|
|
|
|
def setupUI(self):
|
|
self.setWindowTitle("DBC to C Converter")
|
|
self.setGeometry(100, 100, 1000, 600) # 창 크기 조정
|
|
|
|
# 메뉴바 추가
|
|
self.menu_bar = self.menuBar()
|
|
|
|
file_menu = self.menu_bar.addMenu("파일")
|
|
edit_menu = self.menu_bar.addMenu("편집")
|
|
view_menu = self.menu_bar.addMenu("보기")
|
|
tools_menu = self.menu_bar.addMenu("도구")
|
|
|
|
# 파일 메뉴 항목 추가
|
|
add_file_action = QtWidgets.QAction("파일 추가", self)
|
|
add_file_action.setShortcut('Ctrl+O')
|
|
add_file_action.triggered.connect(self.FilesOpen)
|
|
file_menu.addAction(add_file_action)
|
|
|
|
delete_selected_action = QtWidgets.QAction("선택 파일 삭제", self)
|
|
delete_selected_action.setShortcut('Delete')
|
|
delete_selected_action.triggered.connect(self.deleteSelectedFiles)
|
|
file_menu.addAction(delete_selected_action)
|
|
|
|
delete_all_action = QtWidgets.QAction("모든 파일 삭제", self)
|
|
delete_all_action.setShortcut('Ctrl+Shift+Delete')
|
|
delete_all_action.triggered.connect(self.deleteAllFiles)
|
|
file_menu.addAction(delete_all_action)
|
|
|
|
# 편집 메뉴 항목 추가
|
|
select_all_action = QtWidgets.QAction("모두 선택", self)
|
|
select_all_action.setShortcut('Ctrl+A')
|
|
select_all_action.triggered.connect(self.selectAllFiles)
|
|
edit_menu.addAction(select_all_action)
|
|
|
|
invert_selection_action = QtWidgets.QAction("선택 반전", self)
|
|
invert_selection_action.setShortcut('Ctrl+I')
|
|
invert_selection_action.triggered.connect(self.invertSelection)
|
|
edit_menu.addAction(invert_selection_action)
|
|
|
|
# 보기 메뉴 항목 추가
|
|
size_menu = view_menu.addMenu("창크기")
|
|
size_menu.addAction("작게", lambda: self.setWindowSize("small")).setShortcut('Ctrl+1')
|
|
size_menu.addAction("보통", lambda: self.setWindowSize("medium")).setShortcut('Ctrl+2')
|
|
size_menu.addAction("크게", lambda: self.setWindowSize("large")).setShortcut('Ctrl+3')
|
|
size_menu.addAction("자동", lambda: self.setWindowSize("auto")).setShortcut('Ctrl+4')
|
|
|
|
sort_menu = view_menu.addMenu("정렬")
|
|
sort_menu.addAction("파일명 오름차순", lambda: self.sortTreeView(0, False)).setShortcut('Ctrl+Shift+N')
|
|
sort_menu.addAction("파일명 내림차순", lambda: self.sortTreeView(0, True)).setShortcut('Ctrl+Shift+M')
|
|
sort_menu.addAction("파일경로 오름차순", lambda: self.sortTreeView(1, False)).setShortcut('Ctrl+Shift+P')
|
|
sort_menu.addAction("파일경로 내림차순", lambda: self.sortTreeView(1, True)).setShortcut('Ctrl+Shift+Q')
|
|
sort_menu.addAction("파일크기 오름차순", lambda: self.sortTreeView(3, False)).setShortcut('Ctrl+Shift+S')
|
|
sort_menu.addAction("파일크기 내림차순", lambda: self.sortTreeView(3, True)).setShortcut('Ctrl+Shift+T')
|
|
|
|
# 도구 메뉴 항목 추가
|
|
settings_action = QtWidgets.QAction("설정", self)
|
|
settings_action.setShortcut('Ctrl+P')
|
|
settings_action.triggered.connect(self.openSettings)
|
|
tools_menu.addAction(settings_action)
|
|
|
|
about_action = QtWidgets.QAction("프로그램 정보", self)
|
|
about_action.setShortcut('Ctrl+I')
|
|
about_action.triggered.connect(self.openAbout)
|
|
tools_menu.addAction(about_action)
|
|
|
|
tools_menu.addSeparator() # 구분선 추가
|
|
|
|
calculator_action = QtWidgets.QAction("계산기", self)
|
|
calculator_action.setShortcut('Ctrl+Shift+C')
|
|
calculator_action.triggered.connect(self.openCalculator)
|
|
tools_menu.addAction(calculator_action)
|
|
|
|
# 툴바 추가
|
|
self.toolbar = self.addToolBar("Main Toolbar")
|
|
add_file_action = QtWidgets.QAction(QtGui.QIcon("img/add_file.png"), "파일 추가", self)
|
|
add_file_action.setShortcut('Ctrl+O')
|
|
add_file_action.triggered.connect(self.FilesOpen)
|
|
self.toolbar.addAction(add_file_action)
|
|
|
|
delete_file_action = QtWidgets.QAction(QtGui.QIcon("img/delete_file.png"), "선택 파일 삭제", self)
|
|
delete_file_action.setShortcut('Delete')
|
|
delete_file_action.triggered.connect(self.deleteSelectedFiles)
|
|
self.toolbar.addAction(delete_file_action)
|
|
|
|
delete_all_action = QtWidgets.QAction(QtGui.QIcon("img/delete_all.png"), "모든 파일 삭제", self)
|
|
delete_all_action.setShortcut('Ctrl+Shift+Delete')
|
|
delete_all_action.triggered.connect(self.deleteAllFiles)
|
|
self.toolbar.addAction(delete_all_action)
|
|
|
|
delete_description_action = QtWidgets.QAction(QtGui.QIcon("img/delete_description.png"), "알림창 내용 삭제", self)
|
|
delete_description_action.setShortcut('Ctrl+Shift+D')
|
|
delete_description_action.triggered.connect(self.clearAlerts)
|
|
self.toolbar.addAction(delete_description_action)
|
|
|
|
convert_action = QtWidgets.QAction(QtGui.QIcon("img/convert.png"), "변환", self)
|
|
convert_action.setShortcut('Ctrl+R')
|
|
convert_action.triggered.connect(self.convertFiles)
|
|
self.toolbar.addAction(convert_action)
|
|
|
|
# 상태바 추가
|
|
self.status_bar = self.statusBar()
|
|
self.status_bar.showMessage("준비 완료")
|
|
|
|
# 메인 위젯 설정
|
|
main_widget = QtWidgets.QWidget()
|
|
self.setCentralWidget(main_widget)
|
|
main_layout = QtWidgets.QVBoxLayout(main_widget)
|
|
|
|
# 파일 리스트 설정
|
|
self.tree = QtWidgets.QTreeWidget()
|
|
self.tree.setColumnCount(4)
|
|
self.tree.setHeaderLabels(['파일명', '파일경로', '채널', '파일크기'])
|
|
self.tree.setColumnWidth(0, 400) # 파일명 너비 설정
|
|
self.tree.setColumnWidth(1, 450) # 파일경로 너비 설정
|
|
self.tree.setColumnWidth(2, 50) # 채널 너비 설정
|
|
self.tree.setColumnWidth(3, 50) # 파일크기 너비 설정
|
|
self.tree.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection)
|
|
self.tree.header().sectionClicked.connect(self.onHeaderClicked)
|
|
self.tree.itemDoubleClicked.connect(self.onItemDoubleClicked)
|
|
main_layout.addWidget(self.tree)
|
|
|
|
# 알림창 설정
|
|
self.alert_text = QtWidgets.QTextEdit()
|
|
self.alert_text.setReadOnly(True)
|
|
main_layout.addWidget(self.alert_text)
|
|
|
|
def openCalculator(self):
|
|
subprocess.Popen('calc.exe')
|
|
|
|
def centerWindow(self):
|
|
frame_geometry = self.frameGeometry()
|
|
screen_center = QtWidgets.QDesktopWidget().availableGeometry().center()
|
|
frame_geometry.moveCenter(screen_center)
|
|
self.move(frame_geometry.topLeft())
|
|
|
|
def centerDialog(self, dialog):
|
|
dialog_geometry = dialog.frameGeometry()
|
|
dialog_geometry.moveCenter(self.frameGeometry().center())
|
|
dialog.move(dialog_geometry.topLeft())
|
|
|
|
def FilesOpen(self):
|
|
file_paths, _ = QtWidgets.QFileDialog.getOpenFileNames(
|
|
self, "파일 열기", self.last_opened_dir, "DBC 파일 (*.dbc);;모든 파일 (*.*)"
|
|
)
|
|
if file_paths:
|
|
self.last_opened_dir = os.path.dirname(file_paths[0])
|
|
existing_files = [self.tree.topLevelItem(i).text(0) for i in range(self.tree.topLevelItemCount())]
|
|
duplicate_files = [os.path.split(file_path)[1] for file_path in file_paths if os.path.split(file_path)[1] in existing_files]
|
|
new_files = [file_path for file_path in file_paths if os.path.split(file_path)[1] not in existing_files]
|
|
|
|
if duplicate_files:
|
|
self.showDuplicateFilesWarning(duplicate_files, new_files, [os.path.split(file_path)[1] for file_path in new_files])
|
|
self.updateAlertText(f"중복 파일 경고", duplicate_files)
|
|
if new_files:
|
|
self.populateTreeView(new_files)
|
|
self.file_paths.extend(new_files) # 파일 경로 저장
|
|
self.updateAlertText(f"파일 추가 완료", [os.path.split(file_path)[1] for file_path in new_files])
|
|
self.saveSettings()
|
|
|
|
def deleteSelectedFiles(self):
|
|
selected_items = self.tree.selectedItems()
|
|
if not selected_items:
|
|
self.showWarning("경고", "선택된 파일이 없습니다")
|
|
self.updateAlertText("파일 삭제 실패", ["선택된 파일이 없습니다"])
|
|
return
|
|
deleted_files = [item.text(0) for item in selected_items]
|
|
for item in selected_items:
|
|
index = self.tree.indexOfTopLevelItem(item)
|
|
self.tree.takeTopLevelItem(index)
|
|
self.updateAlertText(f"파일 삭제 완료", deleted_files)
|
|
self.saveSettings()
|
|
|
|
def deleteAllFiles(self):
|
|
if self.tree.topLevelItemCount() == 0:
|
|
self.showWarning("경고", "삭제할 파일이 없습니다")
|
|
self.updateAlertText("모든 파일 삭제 실패", ["삭제할 파일이 없습니다"])
|
|
return
|
|
deleted_files = [self.tree.topLevelItem(i).text(0) for i in range(self.tree.topLevelItemCount())]
|
|
self.tree.clear()
|
|
self.updateAlertText(f"모든 파일 삭제 완료", deleted_files)
|
|
self.saveSettings()
|
|
|
|
def clearAlerts(self):
|
|
self.alert_text.clear()
|
|
|
|
def showWarning(self, title, message):
|
|
warning_box = QtWidgets.QMessageBox(self)
|
|
warning_box.setIcon(QtWidgets.QMessageBox.Warning)
|
|
warning_box.setWindowTitle(title)
|
|
warning_box.setText(message)
|
|
warning_box.exec_()
|
|
|
|
def showDuplicateFilesWarning(self, duplicate_files, new_files, new_files_names):
|
|
duplicate_files_str = "\n".join(duplicate_files)
|
|
new_files_str = "\n".join(new_files_names)
|
|
if new_files:
|
|
self.showWarning("중복 파일 경고", f"**중복 파일이 존재합니다**\n\n{duplicate_files_str}\n\n\n**중복 파일을 제외한 신규 파일을 추가합니다**\n\n{new_files_str}")
|
|
else:
|
|
self.showWarning("중복 파일 경고", f"**중복 파일이 존재합니다**\n\n{duplicate_files_str}")
|
|
|
|
def sortTreeView(self, col, reverse):
|
|
items = []
|
|
for i in range(self.tree.topLevelItemCount()):
|
|
item = self.tree.takeTopLevelItem(0)
|
|
items.append(item)
|
|
items.sort(key=lambda x: x.text(col), reverse=reverse)
|
|
for item in items:
|
|
self.tree.addTopLevelItem(item)
|
|
|
|
def selectAllFiles(self):
|
|
for i in range(self.tree.topLevelItemCount()):
|
|
self.tree.topLevelItem(i).setSelected(True)
|
|
|
|
def invertSelection(self):
|
|
for i in range(self.tree.topLevelItemCount()):
|
|
item = self.tree.topLevelItem(i)
|
|
item.setSelected(not item.isSelected())
|
|
|
|
def setWindowSize(self, size):
|
|
if size == "small":
|
|
self.resize(600, 400)
|
|
elif size == "medium":
|
|
self.resize(800, 600)
|
|
elif size == "large":
|
|
self.resize(1000, 800)
|
|
elif size == "auto":
|
|
self.adjustSize()
|
|
|
|
def openSettings(self):
|
|
settings_window = QtWidgets.QDialog(self)
|
|
settings_window.setWindowTitle("설정")
|
|
settings_window.setGeometry(100, 100, 400, 300)
|
|
layout = QtWidgets.QVBoxLayout(settings_window)
|
|
label = QtWidgets.QLabel("설정 창입니다.")
|
|
label.setAlignment(QtCore.Qt.AlignCenter)
|
|
layout.addWidget(label)
|
|
self.centerDialog(settings_window)
|
|
settings_window.exec_()
|
|
|
|
def openAbout(self):
|
|
about_window = QtWidgets.QDialog(self)
|
|
about_window.setWindowTitle("프로그램 정보")
|
|
about_window.setGeometry(100, 100, 400, 300)
|
|
layout = QtWidgets.QVBoxLayout(about_window)
|
|
|
|
label1 = QtWidgets.QLabel("이 프로그램은 DBC 파일을 C 파일로 변환합니다.")
|
|
label1.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
label2 = QtWidgets.QLabel("프로그램 정보")
|
|
label2.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
label3 = QtWidgets.QLabel(f"버전: {self.version}")
|
|
label3.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
label2.setFont(QtGui.QFont("Arial", 12, QtGui.QFont.Bold))
|
|
|
|
layout.addWidget(label1)
|
|
layout.addWidget(label2)
|
|
layout.addWidget(label3)
|
|
self.centerDialog(about_window)
|
|
about_window.exec_()
|
|
|
|
def populateTreeView(self, file_paths):
|
|
def format_file_size(size_in_bytes):
|
|
size_in_kb = size_in_bytes / 1024
|
|
return f"{size_in_kb:.2f} KB"
|
|
|
|
for file_path in file_paths:
|
|
directory, file = os.path.split(file_path)
|
|
fsize = format_file_size(os.path.getsize(file_path))
|
|
item = QtWidgets.QTreeWidgetItem([file, directory, self.channel_options[0], fsize])
|
|
self.tree.addTopLevelItem(item)
|
|
|
|
def updateAlertText(self, message, file_list):
|
|
current_time = QtCore.QDateTime.currentDateTime().toString("yyyy-MM-dd HH:mm:ss")
|
|
self.alert_text.append(f"[{current_time}] {message}")
|
|
for file in file_list:
|
|
self.alert_text.append(f" - {file}")
|
|
self.alert_text.setAlignment(QtCore.Qt.AlignLeft)
|
|
self.alert_text.setStyleSheet("QTextEdit { line-height: 2.5; } border-radius: 3px;")
|
|
|
|
def convertFiles(self):
|
|
for file_path in self.file_paths:
|
|
try:
|
|
subprocess.run(["python", "DBC_to_C_RX.py", file_path], check=True)
|
|
subprocess.run(["python", "DBC_to_C_TX.py", file_path], check=True)
|
|
self.updateAlertText(f"변환 성공", [file_path])
|
|
except subprocess.CalledProcessError as e:
|
|
self.updateAlertText(f"변환 실패: {e}", [file_path])
|
|
|
|
def onHeaderClicked(self, logicalIndex):
|
|
header = self.tree.headerItem()
|
|
if header.text(logicalIndex).endswith("▲"):
|
|
self.sortTreeView(logicalIndex, True)
|
|
header.setText(logicalIndex, header.text(logicalIndex).replace("▲", "▼"))
|
|
else:
|
|
self.sortTreeView(logicalIndex, False)
|
|
header.setText(logicalIndex, header.text(logicalIndex).replace("▼", "▲"))
|
|
|
|
def onItemDoubleClicked(self, item, column):
|
|
if column == 0: # 파일명을 더블클릭한 경우
|
|
file_path = os.path.join(item.text(1), item.text(0))
|
|
os.startfile(file_path)
|
|
elif column == 1: # 경로를 더블클릭한 경우
|
|
os.startfile(item.text(1))
|
|
elif column == 2: # 채널을 더블클릭한 경우
|
|
self.openChannelDialog(item)
|
|
|
|
def openChannelDialog(self, item):
|
|
channel_dialog = QtWidgets.QDialog(self)
|
|
channel_dialog.setWindowTitle("채널 선택")
|
|
channel_dialog.setGeometry(100, 100, 200, 100)
|
|
layout = QtWidgets.QVBoxLayout(channel_dialog)
|
|
|
|
channel_combo = QtWidgets.QComboBox()
|
|
channel_combo.addItems(self.channel_options)
|
|
layout.addWidget(channel_combo)
|
|
|
|
button_layout = QtWidgets.QHBoxLayout()
|
|
apply_button = QtWidgets.QPushButton("확인")
|
|
apply_button.clicked.connect(lambda: self.applyChannelSelection(item, channel_combo, channel_dialog))
|
|
button_layout.addWidget(apply_button)
|
|
|
|
cancel_button = QtWidgets.QPushButton("취소")
|
|
cancel_button.clicked.connect(channel_dialog.close)
|
|
button_layout.addWidget(cancel_button)
|
|
|
|
layout.addLayout(button_layout)
|
|
self.centerDialog(channel_dialog)
|
|
channel_dialog.exec_()
|
|
|
|
def applyChannelSelection(self, item, combo, dialog):
|
|
item.setText(2, combo.currentText())
|
|
dialog.close()
|
|
|
|
if __name__ == '__main__':
|
|
app = QtWidgets.QApplication(sys.argv)
|
|
main_view = MainView()
|
|
main_view.show()
|
|
sys.exit(app.exec_()) |