CollectMangaInfo/UI.py

import sys
import os
import UtilPack as util

from io import BytesIO

from PyQt5.QtCore import Qt, QUrl, QSettings, QSize, QPoint, QByteArray
from PyQt5.QtWidgets import QApplication, QWidget, QMainWindow, QPushButton, QVBoxLayout, QLineEdit, QHBoxLayout, QListWidget, QListWidgetItem, QLabel, QFileDialog
from PyQt5.QtGui import QPixmap, QKeyEvent

QApplication.setAttribute(Qt.AA_ShareOpenGLContexts)

class MyApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.loadINI()


    def closeEvent(self, event):
        self.saveINI()
        event.accept()
        

    def initUI(self):
        layout = self.MakeUI()
            
        # 레이아웃을 윈도우에 적용
        central_widget = QWidget()
        central_widget.setLayout(layout)
        self.setCentralWidget(central_widget)
        self.setWindowTitle('Manga Database')
        self.move(10, 10)
        self.show()
    

    def loadINI(self):
        settings = QSettings('MyApp', 'settings')

        window_size = settings.value('window/size', '800, 600')
        window_position = settings.value('window/position', '100, 100')


    def saveINI(self):
        settings = QSettings('MyApp', 'settings')

        # 키-값 형식으로 데이터 저장
        settings.setValue('window/size', "1024,768")
        settings.setValue('window/position', "100,100")


    def MakeUI_Left(self):
        self.list_SrcPath = QListWidget()

        btn_Add = QPushButton("Add", self)
        btn_Add.clicked.connect(self.on_click_SrcAdd)
        btn_Del = QPushButton("Del", self)
        btn_Del.clicked.connect(self.on_click_SrcDel)
        btn_Read = QPushButton("Read!", self)
        btn_Read.clicked.connect(self.on_click_SrcRead)
        layout_Btns = QHBoxLayout()
        layout_Btns.addWidget(btn_Add)
        layout_Btns.addWidget(btn_Del)
        layout_Btns.addWidget(btn_Read)

        layout_Top = QVBoxLayout()
        layout_Top.addWidget(self.list_SrcPath)
        layout_Top.addLayout(layout_Btns)
        
        self.list_ArcList = QListWidget(self)
        self.list_ArcList.itemSelectionChanged.connect(self.on_Item_SelChanged_listArc)
        self.list_Infos = QListWidget(self)

        layout = QVBoxLayout()
        layout.addLayout(layout_Top)
        layout.addWidget(self.list_ArcList)
        layout.addWidget(self.list_Infos)

        return layout
    

    def MakeUI(self):
        layout_L = self.MakeUI_Left()
        
        # show Image in middle Layout
        self.label_Image = QLabel(self)
        
        # Load Default Image
        pixmap = QPixmap("layoutImg.jpeg")
        scaled_Pix = pixmap.scaledToHeight(600)
        self.label_Image.setPixmap(scaled_Pix)

        self.list_Items = QListWidget(self)
        self.list_Items.itemSelectionChanged.connect(self.on_Item_SelChanged_Items)

        # 레이아웃 설정
        layout = QHBoxLayout()
        layout.addLayout(layout_L)
        layout.addWidget(self.label_Image)
        layout.addWidget(self.list_Items)
        
        return layout
    

    def on_click_SrcAdd(self):
        # 폴더 선택 창을 띄움
        folder_path = QFileDialog.getExistingDirectory(self, '폴더 선택', '')
        
        # 폴더 목록에 추가
        self.list_SrcPath.addItem(folder_path)

        # 폴더 내의 파일 목록을 가져온다.
        listFiles = util.ListContainFiles(folder_path)

        # 파일 목록을 훑어서 내용을 판단
        # 1. 압축파일이 들어있나?
        # 2. 이미지 파일이 들어있나?
        isContainImg = False
        for pathFile in listFiles:
            filename = util.GetParentDirName(pathFile, 0)
            fileExt = util.GetExtStr(filename)

            # 압축파일이 들어있다면...
            if fileExt.lower() in [".zip", ".cbz", ".rar"]:
                # 이름은 폴더/압축파일
                FolderName = util.GetParentDirName(pathFile, 1)
                ItemName = os.path.join(FolderName, filename)
                item = QListWidgetItem(ItemName)
                # 압축파일의 전체 경로를 따로 저장
                item.setData(Qt.UserRole, pathFile)
                self.list_ArcList.addItem(item)

            # 이미지 파일이 들어있다면...
            if fileExt.lower() in [".jpg", ".webp", ".jpeg", ".png", ".gif"]:
                isContainImg = True
        
        # 이미지 파일이 들어있다고 알려오면...
        if True == isContainImg:
            # 이름은 폴더
            FolderName = util.GetParentDirName(folder_path, 0)
            item = QListWidgetItem(FolderName)
            # 폴더 경로를 따로 저장
            item.setData(Qt.UserRole, folder_path)
            self.list_ArcList.addItem(item)

    def on_click_SrcDel(self):
        items = self.list_SrcPath.selectedItems()

        if not items:
            return
        
        for item in items:
            row = self.list_SrcPath.row(item)
            self.list_SrcPath.takeItem(row)


    def on_click_SrcRead(self):
        pass


    def on_Item_SelChanged_listArc(self):
        items = self.list_ArcList.selectedItems()

        self.list_Infos.clear()

        self.dataCurSel = ""
        for item in items:
            self.dataCurSel = item.data(Qt.UserRole)
        
        fileExt = util.GetExtStr(self.dataCurSel)
        # 압축파일이 들어있다면...
        if fileExt.lower() in [".zip", ".cbz", ".rar"]:
        
        dataCurSel

        listZipConts = util.GetZipContentList(self.pathCurSelZip)
        self.list_Infos.addItem( QListWidgetItem(self.pathCurSelZip) )
        self.list_Infos.addItem( QListWidgetItem( f"{len(listZipConts)}" ) )

        self.list_Items.clear()
        self.list_Items.addItems( listZipConts )


    def on_Item_SelChanged_Items(self):
        items = self.list_Items.selectedItems()
        if 0 >= len(items):
            return

        data = util.GetZippedFileByte(self.pathCurSelZip, items[0].text())
        imageIO = BytesIO(data)
        pixmap = QPixmap()
        pixmap.loadFromData( QByteArray( imageIO.getvalue() ) )

        szLabel = self.label_Image.size()

        
        self.label_Image.setPixmap(pixmap)



if __name__ == '__main__':
    app = QApplication(sys.argv)
    
    app.setQuitOnLastWindowClosed(True)
    main_window = MyApp()
    #main_window.show()
    
    sys.exit(app.exec_())