class GroupOfImages(QGroupBox):
def __init__(self):
super(GroupOfImages, self).__init__()
self.GroupOfImagesLayout = QBoxLayout(QBoxLayout.TopToBottom)
GroupOfImages.setFixedHeight(self, 300)
GroupOfImages.setFixedWidth(self, 300)
self.height = 300
def addImage(self, image):
self.GroupOfImagesLayout.addWidget(image)
self.height += 300
GroupOfImages.setFixedHeight(self, self.height + 300)
self.setLayout(self.GroupOfImagesLayout)
def removeImage(self, image):
self.GroupOfImagesLayout.removeWidget(image)
self.setLayout(self.GroupOfImagesLayout)
def refreshView(self):
self.setLayout(self.GroupOfImagesLayout)
class HV(QWidget):
def __init__(self, labels):
QWidget.__init__(self)
self.labels = labels
self.layout = QBoxLayout(QBoxLayout.LeftToRight)
for label in labels.labels:
self.layout.addWidget(label)
self.setLayout(self.layout)
def setLabels(self, labels):
for label in self.labels.labels:
self.layout.removeWidget(label)
label.setParent(None)
self.labels = labels
for label in labels.labels:
self.layout.addWidget(label)
def setLeftToRight(self):
self.layout.setDirection(QtWidgets.QBoxLayout.LeftToRight)
def setTopToBottom(self):
self.layout.setDirection(QtWidgets.QBoxLayout.TopToBottom)
def remove_widget_from_scroll(widget: QWidget, parent_layout: QBoxLayout,
data_list: List):
"""Remove a widget from a scroll area, and remove the associated data.
Parameters
----------
widget : QWidget
Widget that should be removed
parent_layout : QBoxLayout
The layout of the scroll area the widget should be removed from
data_list : List
The list of data related to the widgets in this scroll area.
Information
-----------
Only removes the widget and data if there will still be widgets
in the scroll area after this widget is removed.
"""
if parent_layout.count() > 2: # ? 2 instead of 1 because of spacer.
index = parent_layout.indexOf(widget)
del data_list[index]
remove_all_widgets(widget.layout())
parent_layout.removeWidget(widget)
widget.deleteLater()
class CToolBarArea(QWidget):
signalDragStart = pyqtSignal()
signalDragEnd = pyqtSignal()
signalItemDropped = pyqtSignal(CToolBarAreaItem, QWidget, int)
def __init__(self, parent, orientation):
super().__init__(parent)
self.editor = parent
self.orientation = orientation
self.dropTargetSpacer = None
self.actionContextPosition = QPoint()
self.isLocked = False
if self.orientation == Qt.Horizontal:
self.layout_ = QBoxLayout(QBoxLayout.LeftToRight)
else:
self.layout_ = QBoxLayout(QBoxLayout.TopToBottom)
self.layout_.setSpacing(0)
self.layout_.setContentsMargins(0, 0, 0, 0)
self.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.setLayout(self.layout_)
self.setAcceptDrops(True)
self.setContextMenuPolicy(Qt.CustomContextMenu)
# print ( "CToolBarArea.__init__", parent )
def getOrientation(self):
return self.orientation
def getToolBars(self):
items = getAllItems(self)
toolBarItems = []
for item in items:
if item.getType() != CToolBarAreaType.ToolBar:
continue
toolBarItems.append(item)
return toolBarItems
def getLargestItemMinimumSize(self):
minSize = self.minimumSize()
items = getAllItems(self)
for item in items:
minSize = minSize.expandedTo(item.getMinimumSize())
return minSize
def setOrientation(self, orientation):
self.orientation = orientation
self.layout_.setDirection(QBoxLayout.LeftToRight if self.orientation ==
Qt.Horizontal else QBoxLayout.TopToBottom)
self.updateLayoutAlignment()
items = getAllItems(self)
for item in items:
item.setOrientation(self.orientation)
def setActionContextPosition(self, actionContextPosition):
self.actionContextPosition = actionContextPosition
def fillContextMenu(self, menu):
pass
def addItem(self, item, targetIndex):
item.signalDragStart.connect(self.onDragStart)
item.signalDragEnd.connect(self.onDragEnd)
item.setOrientation(self.orientation)
item.setLocked(self.isLocked)
self.layout_.insertWidget(targetIndex, item)
item.setArea(self)
def addToolBar(self, toolBar, targetIndex):
self.addItem(CToolBarItem(self, toolBar, self.orientation),
targetIndex)
def addSpacer(self, spacerType, targetIndex):
self.addItem(CSpacerItem(self, spacerType, self.orientation),
targetIndex)
if spacerType == CSpacerType.Expanding:
self.layout_.setAlignment(0)
def removeItem(self, item):
self.layout_.removeWidget(item)
item.signalDragStart.disconnect(self)
item.signalDragEnd.disconnect(self)
self.updateLayoutAlignment(item)
def deleteToolBar(self, toolBarItem):
if self.isAncestorOf(toolBarItem):
qWarning("Trying to remove non-owned toolbar from area")
return
toolBarItem.deleteLater()
def hideAll(self):
items = getAllItems(self)
for item in items:
item.setVisible(False)
def findToolBarByName(self, szToolBarName):
toolBarItems = self.findChildren(CToolBarItem, "",
Qt.FindDirectChildrenOnly)
for toolBarItem in toolBarItems:
if toolBarItem.getName() == szToolBarName:
return toolBarItem
print("Toolbar not found: %s" % szToolBarName)
return None
def setLocked(self, isLocked):
self.isLocked = isLocked
items = getAllItems(self)
for item in items:
item.setLocked(isLocked)
def getState(self):
pass
def setState(self, state):
pass
def dragEnterEvent(self, event):
print("CToolBarArea.dragEnterEvent", event)
dragDropData = CDragDropData.fromMimeData(event.mimeData())
if dragDropData.hasCustomData(CToolBarAreaItem.getMimeType()):
byteArray = dragDropData.getCustomData(
CToolBarAreaItem.getMimeType())
stream = QDataStream(byteArray)
value = None
stream >> value
draggedItem = value if isinstance(value,
CToolBarAreaItem) else None
if not self.parentWidget().isAncestorOf(draggedItem):
return
event.acceptProposedAction()
self.setProperty("dragHover", True)
self.style().unpolish(self)
self.style().polish(self)
def dragMoveEvent(self, event):
dragDropData = CDragDropData.fromMimeData(event.mimeData())
if dragDropData.hasCustomData(CToolBarAreaItem.getMimeType()):
targetIndex = self.getPlacementIndexFromPosition(
self.mapToGlobal(event.pos()))
event.acceptProposedAction()
if self.dropTargetSpacer and targetIndex == self.layout_.indexOf(
self.dropTargetSpacer):
return
byteArray = dragDropData.getCustomData(
CToolBarAreaItem.getMimeType())
stream = QDataStream(byteArray)
value = None
stream >> value
draggedItem = value if isinstance(value,
CToolBarAreaItem) else None
if not self.parentWidget().isAncestorOf(draggedItem):
return
spacerWidth = draggedItem.width()
spacerHeight = draggedItem.height()
if draggedItem.getOrientation() != self.orientation:
tempWidth = spacerWidth
spacerWidth = spacerHeight
spacerHeight = tempWidth
if self.dropTargetSpacer == None:
self.dropTargetSpacer = QSpacerItem(spacerWidth, spacerHeight,
QSizePolicy.Fixed,
QSizePolicy.Fixed)
else:
spacerIndex = self.layout_.indexOf(self.dropTargetSpacer)
if spacerIndex == targetIndex - 1 or (
targetIndex == -1
and spacerIndex == self.layout_.count() - 1):
return
self.removeDropTargetSpacer()
self.dropTargetSpacer = QSpacerItem(spacerWidth, spacerHeight,
QSizePolicy.Fixed,
QSizePolicy.Fixed)
self.layout_.insertSpacerItem(targetIndex, self.dropTargetSpacer)
def dragLeaveEvent(self, event):
self.removeDropTargetSpacer()
self.setProperty("dragHover", False)
self.style().unpolish(self)
self.style().polish(self)
def dropEvent(self, event):
dragDropData = CDragDropData.fromMimeData(event.mimeData())
if dragDropData.hasCustomData(CToolBarAreaItem.getMimeType()):
event.acceptProposedAction()
byteArray = dragDropData.getCustomData(
CToolBarAreaItem.getMimeType())
stream = QDataStream(byteArray)
value = None
stream >> value
item = value if isinstance(value, CToolBarAreaItem) else None
targetIndex = -1
if self.dropTargetSpacer:
targetIndex = self.layout_.indexOf(self.dropTargetSpacer)
containerIndex = self.layout_.indexOf(item)
if containerIndex >= 0 and containerIndex < targetIndex:
targetIndex -= 1
self.removeDropTargetSpacer()
if targetIndex >= self.layout_.count():
targetIndex = -1
self.signalItemDropped.emit(item, self, targetIndex)
def customEvent(self, event):
pass
def paintEvent(self, event):
styleOption = QStyleOption()
styleOption.initFrom(self)
painter = QPainter(self)
self.style().drawPrimitive(QStyle.PE_Widget, styleOption, painter,
self)
def onDragStart(self, item):
self.updateLayoutAlignment(item)
item.setVisiable(False)
self.signalDragStart.emit()
def onDragEnd(self, item):
item.setVisiable(True)
self.signalDragEnd.emit()
def removeDropTargetSpacer(self):
if self.dropTargetSpacer:
self.layout_.removeItem(self.dropTargetSpacer)
self.dropTargetSpacer = None
def indexForItem(self, item):
return self.layout_.indexOf(item)
def moveItem(self, item, destinationIndex):
sourceIndex = self.indexForItem(item)
if sourceIndex == destinationIndex:
return
self.layout_.insertWidget(destinationIndex, item)
def updateLayoutAlignment(self, itemToBeRemoved=None):
spacers = self.findChildren(CSpacerItem, "", Qt.FindDirectChildrenOnly)
expandingSpacers = []
for spacer in spacers:
if spacer.getSpacerType() == CSpacerType.Expanding:
expandingSpacers.append(spacer)
if len(expandingSpacers) == 0 or len(expandingSpacers) != 0 or (
len(expandingSpacers) == 1
and expandingSpacers[0] == itemToBeRemoved):
if self.orientation == Qt.Horizontal:
self.layout_.setAlignment(Qt.AlignLeft | Qt.AlignVCenter)
else:
self.layout_.setAlignment(Qt.AlignTop | Qt.AlignHCenter)
def getItemAtPosition(self, globalPos):
object = QApplication.widgetAt(globalPos)
if object == None or object == self:
return None
item = object if isinstance(object, CToolBarAreaItem) else None
while item == None and object.parent():
object = object.parent()
item = object if isinstance(object, CToolBarAreaItem) else None
return item
def getPlacementIndexFromPosition(self, globalPos):
targetIndex = -1
item = self.getItemAtPosition(globalPos)
if item == None:
localPos = self.mapFromGlobal(globalPos)
if self.dropTargetSpacer:
geom = self.dropTargetSpacer.geometry()
if geom.contains(localPos):
targetIndex = self.layout_.indexOf(self.dropTargetSpacer)
return targetIndex
targetIndex = self.indexForItem(item)
if self.orientation == Qt.Horizontal and item.mapFromGlobal(
globalPos).x() > item.width() / 2:
targetIndex += 1
elif self.orientation == Qt.Vertical and item.mapFromGlobal(
globalPos).y() > item.height() / 2:
targetIndex += 1
if targetIndex >= self.layout_.count():
targetIndex = -1
return targetIndex
class QCustomWindowFrame(QFrame):
contentsChanged = pyqtSignal(QWidget)
def __init__(self):
super(QCustomWindowFrame, self).__init__(None)
self.titleBar = None
self.contents = None
self.sizeGrip = None
self.resizeMargin = 4
self.setMouseTracking(True)
self.layout = QBoxLayout(QBoxLayout.TopToBottom)
self.layout.setSpacing(0)
self.layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.layout)
def __del__(self):
pass
@staticmethod
def wrapWidget(w):
windowFrame = QCustomWindowFrame()
windowFrame.internalSetContents(w)
return windowFrame
def ensureTitleBar(self):
if self.titleBar is None:
self.titleBar = QCustomTitleBar(self)
def internalSetContents(self, widget, useContentsGeometry=True):
if self.contents != None:
if self.contents.parentWidget() == self:
self.contents.setParent(None)
self.layout.removeWidget(self.sizeGrip)
self.layout.removeWidget(self.contents)
self.layout.removeWidget(self.titleBar)
self.contents.removeEventFilter(self)
self.contents.windowTitleChanged.disconnect(self.setWindowTitle)
self.contents.windowIconChanged.disconnect(self.onIconChange)
self.contents = widget
if self.contents != None:
self.contents.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
if self.contents.testAttribute(QtCore.Qt.WA_QuitOnClose):
self.contents.setAttribute(QtCore.Qt.WA_QuitOnClose, False)
self.setAttribute(QtCore.Qt.WA_QuitOnClose)
self.setAttribute(
QtCore.Qt.WA_DeleteOnClose,
self.contents.testAttribute(QtCore.Qt.WA_DeleteOnClose))
self.setWindowTitle(self.contents.windowTitle())
self.setWindowIcon(self.contents.windowIcon())
self.contents.installEventFilter(self)
self.contents.windowTitleChanged.connect(self.setWindowTitle)
self.contents.windowIconChanged.connect(self.onIconChange)
if useContentsGeometry:
self.contents.show()
self.setGeometry(self.contents.geometry())
self.contents.setParent(self)
else:
self.contents.setParent(self)
self.contents.show()
self.updateWindowFlags()
self.ensureTitleBar()
self.layout.addWidget(self.titleBar)
self.layout.addWidget(self.contents)
if self.sizeGrip == None:
self.sizeGrip = QSizeGrip(self)
self.sizeGrip.setMaximumHeight(2)
import sys
if sys.platform == 'darwin':
self.layout.addWidget(self.sizeGrip, 2) # osx
else:
self.layout.addWidget(self.sizeGrip, 2, QtCore.Qt.AlignBottom
| QtCore.Qt.AlignRight) # window
self.contentsChanged.emit(self.contents)
# def nativeEvent ( self, eventType, message, result ):
# if not self.titleBar:
# return False
# return super ().nativeEvent ( eventType, message, result )
def event(self, e):
if e.type() == QEvent.Show:
if not self.contents.isVisibleTo(self):
self.contents.show()
if e.type() == QEvent.Hide:
if self.contents != None:
if self.contents.isVisibleTo(self) and self.windowState(
) != QtCore.Qt.WindowMinimized:
self.contents.hide()
return super().event(e)
def closeEvent(self, e):
if self.contents != None and self.contents.isVisibleTo(self):
if not self.contents.close():
e.ignore()
super().closeEvent(e)
def changeEvent(self, e):
if (e.type() == QEvent.WindowStateChange
or e.type() == QEvent.ActivationChange
) and getMainWindow() and getMainWindow().windowState(
) == QtCore.Qt.WindowMinimized:
getMainWindow().setWindowState(QtCore.Qt.WindowNoState)
if self.titleBar != None:
self.titleBar.updateWindowStateButtons()
super().changeEvent(e)
def eventFilter(self, o, e):
if o == self.contents:
if self.contents.parentWidget() == self:
if e.type() == QEvent.Close:
self.close()
elif e.type() == QEvent.HideToParent:
if self.isVisible():
self.hide()
elif e.type() == QEvent.ShowToParent:
if not self.isVisible():
self.show()
return super().eventFilter(o, e)
def mouseReleaseEvent(self, e):
if self.titleBar is None:
e.ignore()
return
if e.button() == QtCore.Qt.LeftButton:
self.titleBar.dragging = False
super().mouseReleaseEvent(e)
# def nudgeWindow ( self ):
# pass
def calcFrameWindowFlags(self):
flags = self.windowFlags()
if self.contents:
# Get all window flags from contents, excluding WindowType flags.
flags = (flags & QtCore.Qt.WindowType_Mask) | (
self.contents.windowFlags() & ~QtCore.Qt.WindowType_Mask)
import sys
if sys.platform == 'darwin':
return flags | QtCore.Qt.FramelessWindowHint # osx FIXME: CustomizeWindowHint flag will show cocoa title bar.
else:
return flags | QtCore.Qt.FramelessWindowHint | QtCore.Qt.CustomizeWindowHint # windows
def updateWindowFlags(self):
if self.contents != None:
contentsWindowType = self.contents.windowFlags(
) & QtCore.Qt.WindowType_Mask
# Contents has window type set, assign them it to us.
if contentsWindowType != QtCore.Qt.Widget:
# Clear window type flags
self.setWindowFlags(self.windowFlags()
& ~QtCore.Qt.WindowType_Mask)
# Set window type from contents' flags, then clear contents' window type flags.
self.setWindowFlags(self.windowFlags() | contentsWindowType)
self.contents.setWindowFlags(self.windowFlags()
& ~QtCore.Qt.WindowType_Mask)
# No window type has been set on the contents, and we don't have any window flags
elif (self.windowFlags()
& QtCore.Qt.WindowType_Mask) == QtCore.Qt.Widget:
self.setWindowFlags(self.windowFlags() | QtCore.Qt.Window)
self.setWindowFlags(self.calcFrameWindowFlags())
def onIconChange(self):
if self.contents != None and self.windowIcon().cacheKey(
) != self.contents.windowIcon().cacheKey():
self.setWindowIcon(self.contents.windowIcon())
def getResizeMargin(self):
return self.resizeMargin
def setResizeMargin(self, resizeMargin):
self.resizeMargin = resizeMargin
class NTFSLogFileDialog(QDialog, QObject):
complete = pyqtSignal()
def __init__(self, parent=None):
QDialog.__init__(self, parent)
QObject.__init__(self)
self.selected_partition = -1
self.ui()
def ui(self):
self.setWindowTitle("Import File System Log File")
self.setFixedSize(self.sizeHint())
self.layout = QBoxLayout(QBoxLayout.TopToBottom, self)
spacer_item1 = QSpacerItem(10, 5)
self.layout.addItem(spacer_item1)
self.setLayout(self.layout)
# First Group
self.disk_raw_chk_box = QCheckBox(
"In this case, it's possible to carve some files.", self)
self.disk_raw_chk_box.stateChanged.connect(
lambda: self.select_type(self.disk_raw_chk_box))
self.disk_raw_group_box = QGroupBox(self)
self.disk_raw_group_box.setStyleSheet("margin-top: 0;")
self.disk_raw_group_box.setDisabled(True)
disk_raw_group_box_layout = QHBoxLayout(self.disk_raw_group_box)
self.disk_raw_group_box.setLayout(disk_raw_group_box_layout)
self.disk_raw_label = QLabel("Disk Raw: ", self)
self.disk_raw_text_box = QLineEdit()
self.disk_raw_text_box.setReadOnly(True)
self.disk_raw_text_box.setFixedWidth(400)
self.browse_disk_raw_btn = QPushButton("...", self)
self.browse_disk_raw_btn.setFixedWidth(50)
self.browse_disk_raw_btn.clicked.connect(self.btn_clicekd)
self.browse_disk_raw_btn.setCursor(QCursor(Qt.PointingHandCursor))
disk_raw_group_box_layout.addWidget(self.disk_raw_label)
disk_raw_group_box_layout.addWidget(self.disk_raw_text_box)
disk_raw_group_box_layout.addWidget(self.browse_disk_raw_btn)
self.layout.addWidget(self.disk_raw_chk_box)
self.layout.addWidget(self.disk_raw_group_box)
# Second Group
self.ntfs_log_file_chk_box = QCheckBox(
"In this case, NTFS Log analysis only supported.", self)
self.ntfs_log_file_chk_box.stateChanged.connect(
lambda: self.select_type(self.ntfs_log_file_chk_box))
self.ntfs_log_group_box = QGroupBox(self)
self.ntfs_log_group_box.setStyleSheet("margin-top: 0;")
self.ntfs_log_group_box.setDisabled(True)
ntfs_log_group_box_layout = QGridLayout(self)
self.ntfs_log_group_box.setLayout(ntfs_log_group_box_layout)
self.mft_label = QLabel("$MFT: ", self)
self.mft_path_text_box = QLineEdit(self)
self.mft_path_text_box.setReadOnly(True)
self.mft_path_text_box.setFixedWidth(400)
self.browse_mft_btn = QPushButton("...", self)
self.browse_mft_btn.setFixedWidth(50)
self.browse_mft_btn.clicked.connect(self.btn_clicekd)
self.browse_mft_btn.setCursor(QCursor(Qt.PointingHandCursor))
self.usnjrnl_label = QLabel("$UsnJrnl: ", self)
self.usnjrnl_path_text_box = QLineEdit(self)
self.usnjrnl_path_text_box.setReadOnly(True)
self.usnjrnl_path_text_box.setFixedWidth(400)
self.browse_usnjrnl_btn = QPushButton("...", self)
self.browse_usnjrnl_btn.setFixedWidth(50)
self.browse_usnjrnl_btn.clicked.connect(self.btn_clicekd)
self.browse_usnjrnl_btn.setCursor(QCursor(Qt.PointingHandCursor))
self.logfile_label = QLabel("$LogFile: ", self)
self.logfile_path_text_box = QLineEdit(self)
self.logfile_path_text_box.setReadOnly(True)
self.logfile_path_text_box.setFixedWidth(400)
self.browse_logfile_btn = QPushButton("...", self)
self.browse_logfile_btn.setFixedWidth(50)
self.browse_logfile_btn.clicked.connect(self.btn_clicekd)
self.browse_logfile_btn.setCursor(QCursor(Qt.PointingHandCursor))
ntfs_log_group_box_layout.addWidget(self.mft_label, 0, 0)
ntfs_log_group_box_layout.addWidget(self.mft_path_text_box, 0, 1)
ntfs_log_group_box_layout.addWidget(self.browse_mft_btn, 0, 2)
ntfs_log_group_box_layout.addWidget(self.usnjrnl_label, 1, 0)
ntfs_log_group_box_layout.addWidget(self.usnjrnl_path_text_box, 1, 1)
ntfs_log_group_box_layout.addWidget(self.browse_usnjrnl_btn, 1, 2)
ntfs_log_group_box_layout.addWidget(self.logfile_label, 2, 0)
ntfs_log_group_box_layout.addWidget(self.logfile_path_text_box, 2, 1)
ntfs_log_group_box_layout.addWidget(self.browse_logfile_btn, 2, 2)
self.submit_btn = QPushButton("Submit", self)
self.submit_btn.setFixedSize(100, 40)
self.submit_btn.setCursor(QCursor(Qt.PointingHandCursor))
self.logging_label = QLabel("Loading...", self)
self.logging_label.setFixedHeight(20)
self.logging_label.setAlignment(Qt.AlignCenter)
self.logging_label.hide()
self.loading_bar = QProgressBar(self)
self.loading_bar.setFixedHeight(10)
self.loading_bar.setTextVisible(False)
self.loading_bar.hide()
self.bar_thread = LoadingBarThread(self)
self.bar_thread.change_value.connect(self.loading_bar.setValue)
self.spacer_item2 = QSpacerItem(10, 15)
self.spacer_item3 = QSpacerItem(10, 20)
self.layout.addItem(self.spacer_item2)
self.layout.addWidget(self.ntfs_log_file_chk_box)
self.layout.addWidget(self.ntfs_log_group_box)
self.layout.addItem(self.spacer_item3)
self.layout.addWidget(self.submit_btn, alignment=Qt.AlignHCenter)
# self.setWindowModality(Qt.WindowModal)
self.setWindowFlag(Qt.WindowCloseButtonHint | Qt.WindowModal)
self.show()
def keyPressEvent(self, e):
if e.key() == Qt.Key_Escape:
self.close()
def select_type(self, b):
if b is self.disk_raw_chk_box:
if b.isChecked():
self.ntfs_log_file_chk_box.setChecked(False)
self.ntfs_log_group_box.setDisabled(True)
self.disk_raw_group_box.setDisabled(False)
else:
self.disk_raw_group_box.setDisabled(True)
else:
if b.isChecked():
self.disk_raw_chk_box.setChecked(False)
self.disk_raw_group_box.setDisabled(True)
self.ntfs_log_group_box.setDisabled(False)
else:
self.ntfs_log_group_box.setDisabled(True)
def btn_clicekd(self):
sender = self.sender()
fileName = QFileDialog.getOpenFileName(self)
if sender is self.browse_disk_raw_btn:
self.disk_raw_text_box.setText(fileName[0])
elif sender is self.browse_mft_btn:
self.mft_path_text_box.setText(fileName[0])
elif sender is self.browse_usnjrnl_btn:
self.usnjrnl_path_text_box.setText(fileName[0])
elif sender is self.browse_logfile_btn:
self.logfile_path_text_box.setText(fileName[0])
def ready(self):
self.submit_btn.hide()
self.layout.removeWidget(self.submit_btn)
self.layout.addWidget(self.logging_label,
alignment=Qt.AlignBottom | Qt.AlignHCenter)
self.layout.addWidget(self.loading_bar)
self.logging_label.show()
self.loading_bar.show()
self.bar_thread.start()
def resume(self):
if self.bar_thread.cnt < 50:
self.bar_thread.cnt = 100
return
self.bar_thread.toggle_status()
def change_interface(self, contents):
from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem
self.layout.removeWidget(self.disk_raw_chk_box)
self.disk_raw_chk_box.hide()
self.layout.removeWidget(self.disk_raw_group_box)
self.disk_raw_group_box.hide()
self.layout.removeItem(self.spacer_item2)
self.layout.removeWidget(self.ntfs_log_file_chk_box)
self.ntfs_log_file_chk_box.hide()
self.layout.removeWidget(self.ntfs_log_group_box)
self.ntfs_log_group_box.hide()
self.layout.removeItem(self.spacer_item3)
self.layout.removeWidget(self.submit_btn)
self.disk_name_label = QLabel("Image Name:\t" + contents[0][0], self)
self.disk_size_label = QLabel(
"Image Size:\t{} Bytes".format(contents[0][1]), self)
self.disk_size_label.setFixedHeight(20)
self.disk_size_label.setAlignment(Qt.AlignVCenter)
self.disk_part_label = QLabel("Partition:", self)
self.disk_part_label.setFixedHeight(20)
self.disk_part_label.setAlignment(Qt.AlignBottom)
self.partition_tree = QTreeWidget(self)
self.partition_tree.setHeaderLabels([
"Order", "File System", "Active", "Starting Offset",
"Total Sector", "Size"
])
self.partition_tree.item_changed.connect(self.item_changed)
self.partition_tree.resizeColumnToContents(2)
self.partition_tree.resizeColumnToContents(3)
self.partition_tree.resizeColumnToContents(4)
self.partition_tree.headerItem().setTextAlignment(0, Qt.AlignCenter)
self.partition_tree.headerItem().setTextAlignment(1, Qt.AlignCenter)
self.partition_items = []
for row in range(1, 5):
self.partition_tree.headerItem().setTextAlignment(
row + 1, Qt.AlignCenter)
item = QTreeWidgetItem(self.partition_tree)
item.setText(0, str(row))
item.setTextAlignment(0, Qt.AlignLeft)
if not contents[row]:
item.setText(1, "None")
item.setCheckState(0, Qt.Unchecked)
item.setDisabled(True)
continue
for col in range(5):
item.setText(col + 1, contents[row][col])
item.setTextAlignment(col + 1, Qt.AlignCenter)
item.setTextAlignment(1, Qt.AlignLeft)
item.setCheckState(0, Qt.Unchecked)
self.partition_items.append(item)
self.layout.addWidget(self.disk_name_label)
self.layout.addWidget(self.disk_size_label)
self.layout.addWidget(self.disk_part_label)
self.layout.addWidget(self.partition_tree)
self.layout.addItem(QSpacerItem(10, 10))
self.layout.addWidget(self.submit_btn, alignment=Qt.AlignCenter)
def item_changed(self, changed_item, p_int):
if changed_item.checkState(0) == Qt.Checked:
self.selected_partition = int(changed_item.text(0))
for item in self.partition_items:
if item is not changed_item:
item.setCheckState(0, Qt.Unchecked)
