diff --git a/file_cmds.py b/file_cmds.py index 7e516d9..1e93d2c 100644 --- a/file_cmds.py +++ b/file_cmds.py @@ -5,6 +5,11 @@ class FileCmds: def __init__(self, filePath): super().__init__() self.filePath = filePath + self.trackCount = self.track_count() + self.basicInfo = self.generate_basic() + self.advancedInfo = self.generate_advanced() + self.tabsList = self.generate_tabs_list() + self.tabsContent = self.generate_tabs_content() def execute_bash(self, cmd): @@ -12,7 +17,7 @@ class FileCmds: Execute a bash command :type cmd: str - :return: output + :return: str ''' process = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE) output, error = process.communicate() @@ -20,34 +25,17 @@ class FileCmds: def file_exist(self): + ''' + Check if the file exists + + :rtype: bool + ''' if os.path.exists(self.filePath): return True else: print("The file doesn't exist") return False - def get_mkv_infos(self): - ''' - Process MKV infos into a dictionary - - :rtype: list or None - ''' - - process = self.execute_bash(f"mediainfo --Output=JSON {quote(self.filePath)}") - mkvInfo = json.loads(process)["media"]["track"] - return mkvInfo - - # def get_video_track(self): - # ''' - # Get the video track infos - - # :rtype: dict or None - # ''' - # for track in self.mkvInfo: - # if track["@type"] == "Video": - # return track - # print("The file does not contain a video track") - # return(None) def media_inform(self, inform): ''' @@ -61,11 +49,11 @@ class FileCmds: return output - def tracksCount(self): + def track_count(self): ''' Return the number of track of each type - :rtype: dict + :rtype: dict[str,int] ''' trackTypes = [ "Video", @@ -81,10 +69,11 @@ class FileCmds: count = self.media_inform(f"General;%{trackType}Count%").replace('\n', '') if not count: count = 0 - output[trackType] = count + output[trackType] = int(count) return output + def generate_basic(self): ''' Return the basic tab @@ -97,18 +86,14 @@ class FileCmds: informText = r"Text;T: %Language_String%, %Format%\n" informs = [informGeneral] - - count = self.tracksCount() - if not count["Video"] == 0: + if not self.trackCount["Video"] == 0: informs.append(informVideo) - count = self.tracksCount() - if not count["Audio"] == 0: + if not self.trackCount["Audio"] == 0: informs.append(informAudio) - count = self.tracksCount() - if not count["Text"] == 0: + if not self.trackCount["Text"] == 0: informs.append(informText) output = "" @@ -116,7 +101,7 @@ class FileCmds: for inform in informs: output += self.media_inform(inform) - if not count["Menu"] == 0: + if not self.trackCount["Menu"] == 0: output += "M: Menu\n\n" output += "\n" @@ -131,4 +116,67 @@ class FileCmds: :rtype: string ''' output = self.execute_bash(f"mediainfo --Full {quote(self.filePath)}").decode('utf-8') - return output \ No newline at end of file + return output + + + def generate_tabs_list(self): + ''' + Return the list of item for the QListWidget + + :rtype: list + ''' + tabsStr = "" + tabs = ["Basic", "Advanced"] + informGeneral = r"General;General (%Format%)\n" + + informs = [informGeneral] + + if self.trackCount["Video"] == 1: + informVideo = r'Video;Video (%Format%)\n' + informs.append(informVideo) + elif self.trackCount["Video"] > 1: + informVideo = r'Video;Video #%StreamKindPos% (%Format%)\n' + informs.append(informVideo) + + if self.trackCount["Audio"] == 1: + informAudio = r'Audio;Audio (%Format%)\n' + informs.append(informAudio) + elif self.trackCount["Audio"] > 1: + informAudio = r'Audio;Audio #%StreamKindPos% (%Format%)\n' + informs.append(informAudio) + + if self.trackCount["Text"] == 1: + informText = r'Text;Text (%Format%)\n' + informs.append(informText) + elif self.trackCount["Text"] > 1: + informText = r'Text;Text #%StreamKindPos% (%Format%)\n' + informs.append(informText) + + for inform in informs: + tabsStr += self.media_inform(inform) + + if not self.trackCount["Menu"] == 0: + tabsStr += "Menu" + + tabs.extend(list(filter(None, tabsStr.split("\n")))) + return tabs + + + def generate_tabs_content(self): + ''' + Return the content of the tabs + + :rtype: dict[str:str] + ''' + tracksList = self.advancedInfo.split("\n\n") + del tracksList[-1] + + tabs = {} + tabs["Basic"] = self.basicInfo + tabs["Advanced"] = self.advancedInfo + + + for i in range(len(tracksList)): + tabs.update({self.tabsList[i+2]: tracksList[i]}) + + return tabs \ No newline at end of file diff --git a/main_window.py b/main_window.py index 7570ff7..893c41b 100644 --- a/main_window.py +++ b/main_window.py @@ -1,19 +1,27 @@ from PySide6.QtWidgets import QWidget from ui_main_window import Ui_MainWindow from file_cmds import FileCmds +from PySide6.QtGui import QFont class MainWindow(QWidget, Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) - self.setWindowTitle("mediainfoSama") + self.windowTitle = "mediainfoSama" + self.setWindowTitle(self.windowTitle) self.setAcceptDrops(True) self.setupConnections() + #TEMP to remove + #Set font to Mononoki system wise + self.media_detail_plain_text_edit.setFont(QFont("Mononoki Nerd Font", 10)) + + def setupConnections(self): pass + def dragEnterEvent(self, event): event.accept() @@ -25,29 +33,16 @@ class MainWindow(QWidget, Ui_MainWindow): def dropEvent(self, event): self.filePath = event.mimeData().urls()[0].toLocalFile() self.fileCmds = FileCmds(self.filePath) - self.fileCmds.file_exist() if self.fileCmds.file_exist() == False: return - self.populate() + self.setWindowTitle(f"{self.filePath} - {self.windowTitle}") + self.populate_tabs() + - - print("YEAH") - # videoTrack = self.mkv_infos.get_video_track() - # if videoTrack == None: - # return - - - - def populate(self, path = ''): + def populate_tabs(self, path = ''): self.media_tracks_list_widget.clear() - informGeneral = r"General;G: %Format%, %FileSize_String%, %Duration_String%, %BitRate_String%\n" - informVideo = r"Video;V: %Format%, %Format_Profile%, %StreamSize_String%, %Width%x%Height%, %FrameRate% FPS, %BitRate_String%\n" - informAudio = r"Audio;A: %Language_String%, %Format%, %StreamSize_String%, %BitRate_String%, %Channel(s)_String%, %SamplingRate_String%\n" - informText = r"Text;T: %Language_String%, %Format%\n" - - text = self.fileCmds.generate_basic() - self.media_detail_plain_text_edit.setPlainText(text) - - print(self.fileCmds.tracksCount()) \ No newline at end of file + self.media_tracks_list_widget.addItems(self.fileCmds.tabsList) + text = self.fileCmds.tabsContent["Basic"] + self.media_detail_plain_text_edit.setPlainText(text) \ No newline at end of file