POC working GUI

ED2K.py

@@ -71,10 +71,20 @@
 """
 
 
+import sys
 import os
 import binascii
+from collections import namedtuple
 
 
+MetEntry = namedtuple('met_entry',
+                      ['Filename', 'ED2KHash', 'LastChanged', 'Filesize', 'Transfered', 'Requested', 'Accepted',
+                       'Transfered_HI', 'PartFilename', 'AICHashCount', 'AICHash'])
+
+METENTRY_FIELDS = ('FT_FILENAME', 'ED2KHASH', 'NUMPARTS', 'LASTCHANGED', 'PARTHASHES', 'FT_FILESIZE',
+                   'FT_ATTRANSFERRED', 'FT_ATREQUESTED', 'FT_ATACCEPTED', 'FT_ATTRANSFERREDHI', 'FT_PARTFILENAME',
+                   'FT_AICHHASHSET', 'FT_AICH_HASH', 'FT_ULPRIORITY', 'FT_KADLASTPUBLISHSRC')
+
 HEADER_KNOWN_MET_FILE = b'\x0e'
 HEADER_PART_MET_FILE = b'\xe0'
 
@@ -132,34 +142,31 @@ def record_read_hash16(fd):
 
 
 def record_tags(fd):
-    tagcount = met_readinteger(fd)
-    print('Tags:                {}'.format(tagcount))
-    for index in range(tagcount):
+    tags = {}
+    tags['TAGCOUNT'] = met_readinteger(fd)
+    print('Tags:                {}'.format(tags['TAGCOUNT']))
+    for index in range(tags['TAGCOUNT']):
         filetag = fd.read(4)
         if filetag[:3] not in [b'\x02\x01\x00', b'\x03\x01\x00']:
             print('<<< Premature end of record at offset 0x{:08X}'.format(fd.tell()))
-            return True
+            return tags
         tagid = filetag[-1]
         if tagid == FT_FILENAME:
-            filename = met_readstring(fd)
-            print('Filename:            {}'.format(filename))
+            tags['FT_FILENAME'] = met_readstring(fd)
         elif tagid == FT_FILESIZE:
-            filesize = met_readinteger(fd)
-            print('Filesize:            {}'.format(filesize))
+            tags['FT_FILESIZE'] = met_readinteger(fd)
         elif tagid == FT_ATTRANSFERRED:
-            print('Transfered:          {}'.format(met_readinteger(fd)))
+            tags['FT_ATTRANSFERRED'] = met_readinteger(fd)
         elif tagid == FT_ATREQUESTED:
-            print('Requested:           {}'.format(met_readinteger(fd)))
+            tags['FT_ATREQUESTED'] = met_readinteger(fd)
         elif tagid == FT_ATACCEPTED:
-            print('Accepted:            {}'.format(met_readinteger(fd)))
+            tags['FT_ATACCEPTED'] = met_readinteger(fd)
         elif tagid == FT_ATTRANSFERREDHI:
-            print('Transfered (HI):     {}'.format(met_readinteger(fd)))
+            tags['FT_ATTRANSFERREDHI'] = met_readinteger(fd)
         elif tagid == FT_ULPRIORITY:
-            print('FT_ULPRIORITY')
-            met_readinteger(fd)
+            tags['FT_ULPRIORITY'] = met_readinteger(fd)
         elif tagid == FT_KADLASTPUBLISHSRC:
-            print('FT_KADLASTPUBLISHSRC')
-            met_readinteger(fd)
+            tags['FT_KADLASTPUBLISHSRC'] = met_readinteger(fd)
         elif tagid == FT_KADLASTPUBLISHNOTES:
             print('FT_KADLASTPUBLISHNOTES')
             met_readinteger(fd)
@@ -173,15 +180,13 @@ def record_tags(fd):
             print('FT_KADLASTPUBLISHKEY')
             met_readinteger(fd)
         elif tagid == FT_AICH_HASH:
-            print('AIC Hash:            {}'.format(met_readstring(fd)))
+            tags['FT_AICH_HASH'] = met_readstring(fd)
         elif tagid == FT_LASTSHARED:
-            print('FT_LASTSHARED')
-            met_readinteger(fd)
+            tags['FT_LASTSHARED'] =  met_readinteger(fd)
         elif tagid == FT_AICHHASHSET:
-            print('FT_AICHHASHSET')
-            aichashcount = met_readinteger(fd, 1)
+            tags['FT_AICHHASHSET'] = met_readinteger(fd, 1)
         elif tagid == FT_PARTFILENAME:
-            print('Part filename:       {}'.format(met_readstring(fd)))
+            tags['FT_PARTFILENAME'] = met_readstring(fd)
         elif tagid == FT_MEDIA_LENGTH:
             print('Media Length:        {}'.format(met_readinteger(fd)))
         elif tagid == FT_MEDIA_CODEC:
@@ -200,7 +205,7 @@ def record_tags(fd):
         elif tagid in FT_INT:
             print('... Reading data with tag: {}'.format(tagid))
             met_readstring(fd)
-    return True
+    return tags
 
 
 def record_num_parts(fd):
@@ -208,17 +213,27 @@ def record_num_parts(fd):
 
 
 def load_record(fd):
-    last_changed = record_last_changed(fd)
-    aichash = record_read_hash16(fd)
-    partshashes = []
-    for i in range(record_num_parts(fd)):
-        partshashes.append(record_read_hash16(fd))
-    record_tags(fd)
+    met_tags = {}
 
-    print('Last changed:        {}'.format(last_changed))
-    print('ED2K Hash:           {}'.format(aichash))
-    print('Num Parts:           {}'.format(len(partshashes)))
-    print('Parts Hashes:        {}'.format(partshashes))
+    met_tags['LASTCHANGED'] = record_last_changed(fd)
+    met_tags['ED2KHASH'] = record_read_hash16(fd)
+    met_tags['NUMPARTS'] = record_num_parts(fd)
+    partshashes = []
+    for i in range(met_tags['NUMPARTS']):
+        partshashes.append(record_read_hash16(fd))
+    met_tags['PARTHASHES'] = partshashes
+    met_tags.update(record_tags(fd))
+
+    for field in METENTRY_FIELDS:
+        if field not in met_tags:
+            met_tags[field] = 0
+    #print(met_tags)
+    metentry = MetEntry(met_tags['FT_FILENAME'], met_tags['ED2KHASH'], met_tags['LASTCHANGED'], met_tags['FT_FILESIZE'],
+                        met_tags['FT_ATTRANSFERRED'], met_tags['FT_ATREQUESTED'], met_tags['FT_ATACCEPTED'],
+                        met_tags['FT_ATTRANSFERREDHI'], met_tags['FT_PARTFILENAME'], met_tags['FT_AICHHASHSET'],
+                        met_tags['PARTHASHES'])
+    print(metentry)
+    return metentry
 
 
 def load_knownfiles(filename):
@@ -230,14 +245,13 @@ def load_knownfiles(filename):
         if magic == HEADER_KNOWN_MET_FILE:
             for index in range(count):
                 print('>>> [{}] NEW RECORD ***'.format(index+1))
-                print('Record offset:       {}'.format(fd.tell()))
-                load_record(fd)
-                print('')
+                yield load_record(fd)
         elif magic == HEADER_PART_MET_FILE:
             print('>>> [{}] NEW RECORD ***'.format(0))
-            load_record(fd)
+            yield load_record(fd)
 
 
 if __name__ == '__main__':
-    # load_knownfiles('.\\data\\known.met')
-    load_knownfiles('.\\data\\008.part.met')
+    load_knownfiles('.\\data\\known.met')
+    # load_knownfiles('.\\data\\008.part.met')
+    #load_knownfiles(sys.argv[-1])

MuleAnalyzerUI.py

@@ -0,0 +1,99 @@
+# -*- coding: utf-8 -*-
+
+# Form implementation generated from reading ui file 'main.ui'
+#
+# Created by: PyQt5 UI code generator 5.9
+#
+# WARNING! All changes made in this file will be lost!
+
+from PyQt5 import QtCore, QtGui, QtWidgets
+
+class Ui_MainWindow(object):
+    def setupUi(self, MainWindow):
+        MainWindow.setObjectName("MainWindow")
+        MainWindow.resize(797, 662)
+        self.centralwidget = QtWidgets.QWidget(MainWindow)
+        self.centralwidget.setObjectName("centralwidget")
+        self.verticalLayout = QtWidgets.QVBoxLayout(self.centralwidget)
+        self.verticalLayout.setObjectName("verticalLayout")
+        self.tabWidget = QtWidgets.QTabWidget(self.centralwidget)
+        self.tabWidget.setObjectName("tabWidget")
+        self.tab = QtWidgets.QWidget()
+        self.tab.setObjectName("tab")
+        self.verticalLayout_2 = QtWidgets.QVBoxLayout(self.tab)
+        self.verticalLayout_2.setObjectName("verticalLayout_2")
+        self.horizontalLayout = QtWidgets.QHBoxLayout()
+        self.horizontalLayout.setObjectName("horizontalLayout")
+        self.tableView = QtWidgets.QTableView(self.tab)
+        self.tableView.setEnabled(True)
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Maximum)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(self.tableView.sizePolicy().hasHeightForWidth())
+        self.tableView.setSizePolicy(sizePolicy)
+        self.tableView.setMinimumSize(QtCore.QSize(243, 100))
+        self.tableView.setMaximumSize(QtCore.QSize(16777215, 100))
+        self.tableView.setFrameShape(QtWidgets.QFrame.Box)
+        self.tableView.setFrameShadow(QtWidgets.QFrame.Plain)
+        self.tableView.setObjectName("tableView")
+        self.horizontalLayout.addWidget(self.tableView)
+        self.verticalLayout_3 = QtWidgets.QVBoxLayout()
+        self.verticalLayout_3.setObjectName("verticalLayout_3")
+        self.lblSearch = QtWidgets.QLabel(self.tab)
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Minimum)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(self.lblSearch.sizePolicy().hasHeightForWidth())
+        self.lblSearch.setSizePolicy(sizePolicy)
+        self.lblSearch.setObjectName("lblSearch")
+        self.verticalLayout_3.addWidget(self.lblSearch)
+        self.horizontalLayout_2 = QtWidgets.QHBoxLayout()
+        self.horizontalLayout_2.setObjectName("horizontalLayout_2")
+        self.inputSearch = QtWidgets.QLineEdit(self.tab)
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Fixed)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(self.inputSearch.sizePolicy().hasHeightForWidth())
+        self.inputSearch.setSizePolicy(sizePolicy)
+        self.inputSearch.setObjectName("inputSearch")
+        self.horizontalLayout_2.addWidget(self.inputSearch)
+        self.btnSearch = QtWidgets.QPushButton(self.tab)
+        self.btnSearch.setObjectName("btnSearch")
+        self.horizontalLayout_2.addWidget(self.btnSearch)
+        self.verticalLayout_3.addLayout(self.horizontalLayout_2)
+        self.horizontalLayout.addLayout(self.verticalLayout_3)
+        self.verticalLayout_2.addLayout(self.horizontalLayout)
+        self.treeView = QtWidgets.QTreeView(self.tab)
+        self.treeView.setFrameShape(QtWidgets.QFrame.Box)
+        self.treeView.setFrameShadow(QtWidgets.QFrame.Plain)
+        self.treeView.setObjectName("treeView")
+        self.verticalLayout_2.addWidget(self.treeView)
+        self.tabWidget.addTab(self.tab, "")
+        self.tab_2 = QtWidgets.QWidget()
+        self.tab_2.setObjectName("tab_2")
+        self.tabWidget.addTab(self.tab_2, "")
+        self.verticalLayout.addWidget(self.tabWidget)
+        MainWindow.setCentralWidget(self.centralwidget)
+        self.menubar = QtWidgets.QMenuBar(MainWindow)
+        self.menubar.setGeometry(QtCore.QRect(0, 0, 797, 18))
+        self.menubar.setObjectName("menubar")
+        self.menuFile = QtWidgets.QMenu(self.menubar)
+        self.menuFile.setObjectName("menuFile")
+        MainWindow.setMenuBar(self.menubar)
+        self.statusBar = QtWidgets.QStatusBar(MainWindow)
+        self.statusBar.setObjectName("statusBar")
+        MainWindow.setStatusBar(self.statusBar)
+        self.menubar.addAction(self.menuFile.menuAction())
+
+        self.retranslateUi(MainWindow)
+        QtCore.QMetaObject.connectSlotsByName(MainWindow)
+
+    def retranslateUi(self, MainWindow):
+        _translate = QtCore.QCoreApplication.translate
+        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
+        self.lblSearch.setText(_translate("MainWindow", "Filter:"))
+        self.btnSearch.setText(_translate("MainWindow", "Search"))
+        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab), _translate("MainWindow", "Tab 1"))
+        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2), _translate("MainWindow", "Tab 2"))
+        self.menuFile.setTitle(_translate("MainWindow", "File"))
+

main.ui

@@ -0,0 +1,139 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>MainWindow</class>
+ <widget class="QMainWindow" name="MainWindow">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>797</width>
+    <height>662</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>MainWindow</string>
+  </property>
+  <widget class="QWidget" name="centralwidget">
+   <layout class="QVBoxLayout" name="verticalLayout">
+    <item>
+     <widget class="QTabWidget" name="tabWidget">
+      <widget class="QWidget" name="tab">
+       <attribute name="title">
+        <string>Tab 1</string>
+       </attribute>
+       <layout class="QVBoxLayout" name="verticalLayout_2">
+        <item>
+         <layout class="QHBoxLayout" name="horizontalLayout">
+          <item>
+           <widget class="QTableView" name="tableView">
+            <property name="enabled">
+             <bool>true</bool>
+            </property>
+            <property name="sizePolicy">
+             <sizepolicy hsizetype="Expanding" vsizetype="Maximum">
+              <horstretch>0</horstretch>
+              <verstretch>0</verstretch>
+             </sizepolicy>
+            </property>
+            <property name="minimumSize">
+             <size>
+              <width>243</width>
+              <height>100</height>
+             </size>
+            </property>
+            <property name="maximumSize">
+             <size>
+              <width>16777215</width>
+              <height>100</height>
+             </size>
+            </property>
+            <property name="frameShape">
+             <enum>QFrame::Box</enum>
+            </property>
+            <property name="frameShadow">
+             <enum>QFrame::Plain</enum>
+            </property>
+           </widget>
+          </item>
+          <item>
+           <layout class="QVBoxLayout" name="verticalLayout_3">
+            <item>
+             <widget class="QLabel" name="lblSearch">
+              <property name="sizePolicy">
+               <sizepolicy hsizetype="Minimum" vsizetype="Minimum">
+                <horstretch>0</horstretch>
+                <verstretch>0</verstretch>
+               </sizepolicy>
+              </property>
+              <property name="text">
+               <string>Filter:</string>
+              </property>
+             </widget>
+            </item>
+            <item>
+             <layout class="QHBoxLayout" name="horizontalLayout_2">
+              <item>
+               <widget class="QLineEdit" name="inputSearch">
+                <property name="sizePolicy">
+                 <sizepolicy hsizetype="Preferred" vsizetype="Fixed">
+                  <horstretch>0</horstretch>
+                  <verstretch>0</verstretch>
+                 </sizepolicy>
+                </property>
+               </widget>
+              </item>
+              <item>
+               <widget class="QPushButton" name="btnSearch">
+                <property name="text">
+                 <string>Search</string>
+                </property>
+               </widget>
+              </item>
+             </layout>
+            </item>
+           </layout>
+          </item>
+         </layout>
+        </item>
+        <item>
+         <widget class="QTreeView" name="treeView">
+          <property name="frameShape">
+           <enum>QFrame::Box</enum>
+          </property>
+          <property name="frameShadow">
+           <enum>QFrame::Plain</enum>
+          </property>
+         </widget>
+        </item>
+       </layout>
+      </widget>
+      <widget class="QWidget" name="tab_2">
+       <attribute name="title">
+        <string>Tab 2</string>
+       </attribute>
+      </widget>
+     </widget>
+    </item>
+   </layout>
+  </widget>
+  <widget class="QMenuBar" name="menubar">
+   <property name="geometry">
+    <rect>
+     <x>0</x>
+     <y>0</y>
+     <width>797</width>
+     <height>18</height>
+    </rect>
+   </property>
+   <widget class="QMenu" name="menuFile">
+    <property name="title">
+     <string>File</string>
+    </property>
+   </widget>
+   <addaction name="menuFile"/>
+  </widget>
+  <widget class="QStatusBar" name="statusBar"/>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>

pyMuleAnalyzerQT.py

@@ -0,0 +1,62 @@
+#! /usr/bin/env python3
+
+import sys
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import QApplication, QMainWindow
+from PyQt5.QtGui import QStandardItemModel
+from MuleAnalyzerUI import Ui_MainWindow
+
+from ED2K import load_knownfiles
+
+class MainWindow(QMainWindow, Ui_MainWindow):
+
+    TreeHeaders = ('Filename', 'ED2K Hash', 'Last Changed', 'Filesize', 'Transfered', 'Requested', 'Accepted',
+                   'Transfered (HI)', 'Part Filename', 'Number of Parts', 'Parts Hashes')
+
+    def __init__(self):
+        super(MainWindow, self).__init__()
+        self.setupUi(self)
+
+        self.treeModel = self.createED2KModel(self)
+        self.treeView.setModel(self.treeModel)
+        self.treeView.setRootIsDecorated(False)
+        self.treeView.setAlternatingRowColors(True)
+
+        # TODO
+        self.loadMetFile(None)
+
+    def createED2KModel(self, parent):
+        model = QStandardItemModel(0, len(self.TreeHeaders), parent)
+        for i, title in enumerate(self.TreeHeaders):
+            model.setHeaderData(i, Qt.Horizontal, title)
+        return model
+
+    def addED2K(self, model, metinfo):
+        model.insertRow(0)
+        model.setData(model.index(0, 0), metinfo.Filename)
+        model.setData(model.index(0, 1), metinfo.ED2KHash)
+        model.setData(model.index(0, 2), metinfo.LastChanged)
+        model.setData(model.index(0, 3), metinfo.Filesize)
+        model.setData(model.index(0, 4), metinfo.Transfered)
+        model.setData(model.index(0, 5), metinfo.Requested)
+        model.setData(model.index(0, 6), metinfo.Accepted)
+        model.setData(model.index(0, 7), metinfo.Transfered_HI)
+        model.setData(model.index(0, 8), metinfo.PartFilename)
+        model.setData(model.index(0, 9), metinfo.AICHashCount)
+        model.setData(model.index(0, 10), metinfo.AICHash)
+
+
+    def loadMetFile(self, filename):
+        for met_entry in load_knownfiles('.\\data\\known.met'):
+            self.addED2K(self.treeModel, met_entry)
+
+
+def main(argv):
+    app = QApplication(argv)
+    window = MainWindow()
+    window.show()
+    sys.exit(app.exec_())
+
+
+if __name__ == '__main__':
+    main(sys.argv)