def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
class QuickFormat(QtGui.QWidget):
def __init__(self, parent = None, args = None):
QtGui.QWidget.__init__(self, parent)
self.__sysargs = args
if os.getuid() != 0:
self.root_user_notification()
self.ui = Ui_QuickFormat()
self.ui.setupUi(self)
self.formatter = Formatter()
self.device_notifier = StorageNotifier()
self.first_run = True
# Initial selection is the first partition found (see next comment)
self.initial_selection = 0
# Connect Qt Signals
self.__init_signals__()
# Initialize custom widgets such as the new combobox
self.__set_custom_widgets__()
# Initialize the fancy notification widget (Derrived from PDS by Gokmen Goksel)
self.__init_notifier__()
# Uses DeviceDetector class for getting device list
self.show_volume_list()
self.generate_file_system_list()
self.volume_to_format = ""
self.__make_initial_selection__(self.initial_selection)
self.refreshing_devices = False
# Monitor USB ports for any new devices
self.device_notifier.iface_notifier.connect_to_signal('DeviceAdded', self.slot_refresh_volume_list)
self.device_notifier.iface_notifier.connect_to_signal('DeviceRemoved', self.slot_refresh_volume_list)
def root_user_notification(self):
root_msgBox = QtGui.QMessageBox(1, i18n("Quick Format"), i18n("You must be root user to use this application."))
root_msgBox.exec_()
sys.exit()
def __make_initial_selection__(self, index):
self.ui.volumeName.setCurrentIndex(index)
self.set_info()
def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
def __set_custom_widgets__(self):
self.ui.listWidget = QtGui.QListWidget(self)
self.ui.volumeName.setModel(self.ui.listWidget.model())
self.ui.volumeName.setView(self.ui.listWidget)
def __init_signals__(self):
self.connect(self.ui.volumeName, SIGNAL("activated(int)"), self.set_info)
self.connect(self.ui.btn_format, SIGNAL("clicked()"), self.format_device)
self.connect(self.ui.btn_cancel, SIGNAL("clicked()"), self.close)
# Formatter signals
self.connect(self.formatter, SIGNAL("format_started()"), self.slot_format_started)
self.connect(self.formatter, SIGNAL("format_successful()"), self.slot_format_successful)
self.connect(self.formatter, SIGNAL("format_failed()"), self.slot_format_failed)
self.connect(self.formatter, SIGNAL("partition_table_error()"), self.slot_partition_table_error)
def filter_file_system(self, volume):
if volume.get_volume_property("DeviceIsPartition") and volume.has_accepted_bus():
return True
def get_volumes(self):
volumes = []
# Get volumes
for v in self.device_notifier.get_all_storages():
# Apply filter
try:
volume = Volume(v)
if self.filter_file_system(volume):
volumes.append(volume)
except:
pass
return volumes
def show_volume_list(self):
self.ui.listWidget.clear()
self.ui.volumeName.hidePopup()
selectedIndex = 0
currentIndex = 0
volumes = self.get_volumes()
if not volumes:
self.no_device_notification()
else:
for volume in volumes:
self.add_volume_to_list(volume)
# select the appropriate volume from list
self.ui.volumeName.setCurrentIndex(selectedIndex)
# Auto-hide notifier
if not self.formatter.formatting and not self.first_run and self.notifier.okButton.isHidden():
QtCore.QTimer.singleShot(2000, self.hide_notifier)
self.set_info()
self.first_run = False
def format_confirm_message(self):
"""'Are you sure?' message for format operation"""
format_message = QtGui.QMessageBox()
format_message.setText(i18n("This operation will delete your all data. Are you sure to continue?"))
format_message.setIcon(QtGui.QMessageBox.Question)
format_message.addButton(ACCEPT, QtGui.QMessageBox.AcceptRole)
format_message.addButton(REJECT, QtGui.QMessageBox.RejectRole)
format_message.exec_()
return format_message.clickedButton().text()
def set_enabled(self, enabled_state):
self.ui.grpNonArgs.setEnabled(enabled_state)
self.ui.btn_format.setEnabled(enabled_state)
self.ui.btn_cancel.setEnabled(enabled_state)
def format_device(self):
""" Starts the formatting operation """
# Temporary variable for controlling equality of volume to format
volume_before_confirm = self.volume_to_format
# Shows a message for confirm format
if self.format_confirm_message() == ACCEPT:
if not self.get_volumes():
self.no_device_notification()
else:
# if device removes while messagebox visible, cancel formatting, press "Yes" though
if volume_before_confirm == self.volume_to_format:
selected_file_system = FILE_SYSTEMS[str(self.ui.fileSystem.currentText())]
selected_label = self.ui.volumeLabel.text()
self.formatter.set_volume_to_format(self.volume_to_format, selected_file_system, selected_label)
self.formatter.start()
def slot_format_started(self):
self.set_enabled(False)
self.notifier.notify(FORMAT_STARTED)
def slot_format_successful(self):
self.notifier.notify(FORMAT_SUCCESSFUL)
self.refresh_volume_list(notify=False)
self.set_enabled(True)
def slot_format_failed(self):
self.notifier.notify(FORMAT_FAILED)
self.set_enabled(True)
def slot_partition_table_error(self):
self.notifier.notify(PARTITION_TABLE_ERROR)
self.set_enabled(True)
def no_device_notification(self):
if self.first_run:
msgBox = QtGui.QMessageBox(1, i18n("Quick Format"), i18n("There aren't any removable devices."))
msgBox.exec_()
sys.exit()
else:
self.notifier.notify(NO_DEVICE)
self.ui.volumeLabel.setText("")
def notify_refreshing_device_list(self):
if not self.refreshing_devices and not self.formatter.formatting:
self.refreshing_devices = True
self.notifier.notify(LOADING_DEVICES)
def refresh_volume_list(self, notify=True):
if notify:
self.notify_refreshing_device_list()
self.show_volume_list()
self.refreshing_devices = False
def slot_refresh_volume_list(self, device):
#FIX doesnt work if hal is used in system
if not self.formatter.formatting:
if str(device).find("UDisks")>=0:
self.refresh_volume_list()
def hide_notifier(self):
self.notifier.animate(start=MIDCENTER, stop=MIDCENTER, direction=OUT)
def generate_file_system_list(self):
self.ui.fileSystem.clear()
# Temporary sapce for file system list
self.tempFileSystems = []
# Get file system list
for fs in FILE_SYSTEMS:
self.tempFileSystems.append(fs)
# Sort file system list
self.tempFileSystems.sort()
self.sortedFileSystems = self.tempFileSystems
# Display file system list in combobox
for fs in self.sortedFileSystems:
self.ui.fileSystem.addItem(fs)
def __find_key(self, dic, val):
"""return the key of a value of a dictionary"""
return [k for k, v in dic.iteritems() if v == val][0]
def set_info(self):
""" Displays the selected volume info on Quickformat.ui """
currentIndex = self.ui.volumeName.currentIndex()
item = self.ui.listWidget.item(currentIndex)
# Get item data (QVariant) convert to Python Object
volume = item.data(32).toPyObject()
# Display volume file system on UI if supported
try:
# find fileSystem index from list
fsIndex = self.ui.fileSystem.findText(
self.__find_key(FILE_SYSTEMS, volume.file_system))
# select fileSystem type
self.ui.fileSystem.setCurrentIndex(fsIndex)
except:
self.ui.fileSystem.setCurrentIndex(0)
print "Cannot match file system. Selecting Ext2 as default."
self.ui.volumeLabel.setText(volume.name)
self.ui.icon.setPixmap(volume.icon)
# Set selected volume as the volume to format
self.volume_to_format = volume
def _prepare_selection_text(self, volume):
if volume.name != "":
return "%s - %s" % (volume.name, volume.path)
return "Volume (%s) - %s" % (volume.file_system, volume.path)
def add_volume_to_list(self, volume):
# Create custom widget
volume_item_widget = VolumeUiItem(volume, self.ui.listWidget)
# Create an empty list widget item
# First parameter is the text shown on the combobox when a selection is made
selectionText = self._prepare_selection_text(volume)
item = QtGui.QListWidgetItem(selectionText, self.ui.listWidget)
# Add whole volume as the item data
item.setData(32, QVariant(volume))
# Set the item's widget to custom widget and append to list
self.ui.listWidget.setItemWidget(item, volume_item_widget)
item.setSizeHint(QSize(200, 70))
class QuickFormat(QtGui.QWidget):
def __init__(self, parent=None, args=None):
QtGui.QWidget.__init__(self, parent)
self.__sysargs = args
self.ui = Ui_QuickFormat()
self.ui.setupUi(self)
self.formatter = Formatter()
self.first_run = True
# Initial selection is the first partition found (see next comment)
self.initial_selection = 0
# If a partition path is given as argument, intial selection becomes the parition given
self.__process_args__()
# Connect Qt Signals
self.__init_signals__()
# Initialize custom widgets such as the new combobox
self.__set_custom_widgets__()
# Initialize the fancy notification widget (Derrived from PDS by Gokmen Goksel)
self.__init_notifier__()
self.show_volume_list()
self.generate_file_system_list()
self.volume_to_format = ""
self.__make_initial_selection__(self.initial_selection)
self.refreshing_devices = False
# Monitor USB ports for any new devices
notifier = Solid.DeviceNotifier().instance()
self.connect(notifier, SIGNAL("deviceAdded(const QString&)"),
self.slot_refresh_volume_list)
self.connect(notifier, SIGNAL("deviceRemoved(const QString&)"),
self.slot_refresh_volume_list)
def __make_initial_selection__(self, index):
self.ui.volumeName.setCurrentIndex(index)
self.set_info()
def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
def __set_custom_widgets__(self):
self.ui.listWidget = QtGui.QListWidget(self)
self.ui.volumeName.setModel(self.ui.listWidget.model())
self.ui.volumeName.setView(self.ui.listWidget)
def __process_args__(self):
self.volumePathArg = ""
if len(self.__sysargs) == 2:
if not self.__sysargs[1].startswith("-"):
self.volumePathArg = self.__sysargs[1]
def __init_signals__(self):
self.connect(self.ui.volumeName, SIGNAL("activated(int)"),
self.set_info)
self.connect(self.ui.btn_format, SIGNAL("clicked()"),
self.format_device)
self.connect(self.ui.btn_cancel, SIGNAL("clicked()"), self.close)
# Formatter signals
self.connect(self.formatter, SIGNAL("format_started()"),
self.slot_format_started)
self.connect(self.formatter, SIGNAL("format_successful()"),
self.slot_format_successful)
self.connect(self.formatter, SIGNAL("format_failed()"),
self.slot_format_failed)
self.connect(self.formatter, SIGNAL("partition_table_error()"),
self.slot_partition_table_error)
def filter_file_system(self, volume):
if volume.has_accepted_bus() and volume.has_accepted_drivetype():
return True
def get_volumes(self):
volumes = []
# Get volumes
for v in Solid.Device.listFromType(Solid.StorageDrive.StorageVolume):
# Apply filter
volume = Volume(v)
if self.filter_file_system(volume):
volumes.append(volume)
return volumes
def show_volume_list(self):
self.ui.listWidget.clear()
self.ui.volumeName.hidePopup()
selectedIndex = 0
currentIndex = 0
volumes = self.get_volumes()
if not volumes:
self.no_device_notification()
else:
for volume in volumes:
self.add_volume_to_list(volume)
# select the appropriate volume from list
self.ui.volumeName.setCurrentIndex(selectedIndex)
# Auto-hide notifier
if not self.formatter.formatting and not self.first_run and self.notifier.okButton.isHidden(
):
QtCore.QTimer.singleShot(2000, self.hide_notifier)
self.set_info()
self.first_run = False
def format_device(self):
""" Starts the formatting operation """
selected_file_system = FILE_SYSTEMS[str(
self.ui.fileSystem.currentText())]
selected_label = self.ui.volumeLabel.text()
self.formatter.set_volume_to_format(self.volume_to_format,
selected_file_system,
selected_label)
self.formatter.start()
def slot_format_started(self):
self.notifier.notify(FORMAT_STARTED)
def slot_format_successful(self):
self.notifier.notify(FORMAT_SUCCESSFUL)
self.refresh_volume_list(notify=False)
def slot_format_failed(self):
self.notifier.notify(FORMAT_FAILED)
def slot_partition_table_error(self):
self.notifier.notify(PARTITION_TABLE_ERROR)
def no_device_notification(self):
if self.first_run:
msgBox = QtGui.QMessageBox(
1, i18n("QuickFormat"),
i18n("There aren't any removable devices."))
msgBox.exec_()
sys.exit()
else:
self.notifier.notify(NO_DEVICE)
def notify_refreshing_device_list(self):
if not self.refreshing_devices and not self.formatter.formatting:
self.refreshing_devices = True
self.notifier.notify(LOADING_DEVICES)
def refresh_volume_list(self, notify=True):
if notify:
self.notify_refreshing_device_list()
self.show_volume_list()
self.refreshing_devices = False
def slot_refresh_volume_list(self, device):
#FIX doesnt work if hal is used in system
if not self.formatter.formatting:
if str(device).find("UDisks") >= 0:
self.refresh_volume_list()
def hide_notifier(self):
self.notifier.animate(start=MIDCENTER, stop=MIDCENTER, direction=OUT)
def generate_file_system_list(self):
self.ui.fileSystem.clear()
# Temporary sapce for file system list
self.tempFileSystems = []
# Get file system list
for fs in FILE_SYSTEMS:
self.tempFileSystems.append(fs)
# Sort file system list
self.tempFileSystems.sort()
self.sortedFileSystems = self.tempFileSystems
# Display file system list in combobox
for fs in self.sortedFileSystems:
self.ui.fileSystem.addItem(fs)
def __find_key(self, dic, val):
"""return the key of a value of a dictionary"""
return [k for k, v in dic.iteritems() if v == val][0]
def set_info(self):
""" Displays the selected volume info on Quickformat.ui """
currentIndex = self.ui.volumeName.currentIndex()
item = self.ui.listWidget.item(currentIndex)
# Get item data (QVariant) convert to Python Object
volume = item.data(32).toPyObject()
# Display volume file system on UI if supported
try:
# find fileSystem index from list
fsIndex = self.ui.fileSystem.findText(
self.__find_key(FILE_SYSTEMS, volume.file_system))
# select fileSystem type
self.ui.fileSystem.setCurrentIndex(fsIndex)
except:
self.ui.fileSystem.setCurrentIndex(0)
print "Cannot match file system. Selecting Ext2 as default."
self.ui.volumeLabel.setText(volume.name)
self.ui.icon.setPixmap(volume.icon)
# Set selected volume as the volume to format
self.volume_to_format = volume
def _prepare_selection_text(self, volume):
if volume.name != "":
return "%s - %s" % (volume.name, volume.path)
return "%s - %s" % (device.name, volume.path)
def add_volume_to_list(self, volume):
# Create custom widget
volume_item_widget = VolumeUiItem(volume, self.ui.listWidget)
# Create an empty list widget item
# First parameter is the text shown on the combobox when a selection is made
selectionText = self._prepare_selection_text(volume)
item = QtGui.QListWidgetItem(selectionText, self.ui.listWidget)
# Add whole volume as the item data
item.setData(32, QVariant(volume))
# Set the item's widget to custom widget and append to list
self.ui.listWidget.setItemWidget(item, volume_item_widget)
item.setSizeHint(QSize(200, 70))
if volume.path == self.volumePathArg:
self.initial_selection = self.ui.listWidget.count() - 1
class QuickFormat(QtGui.QWidget):
def __init__(self, parent=None, args=None):
QtGui.QWidget.__init__(self, parent)
self.__sysargs = args
if os.getuid() != 0:
self.root_user_notification()
self.ui = Ui_QuickFormat()
self.ui.setupUi(self)
self.formatter = Formatter()
self.device_notifier = StorageNotifier()
self.first_run = True
# Initial selection is the first partition found (see next comment)
self.initial_selection = 0
# Connect Qt Signals
self.__init_signals__()
# Initialize custom widgets such as the new combobox
self.__set_custom_widgets__()
# Initialize the fancy notification widget (Derrived from PDS by Gokmen Goksel)
self.__init_notifier__()
# Uses DeviceDetector class for getting device list
self.show_volume_list()
self.generate_file_system_list()
self.volume_to_format = ""
self.__make_initial_selection__(self.initial_selection)
self.refreshing_devices = False
# Monitor USB ports for any new devices
self.device_notifier.iface_notifier.connect_to_signal(
'DeviceAdded', self.slot_refresh_volume_list)
self.device_notifier.iface_notifier.connect_to_signal(
'DeviceRemoved', self.slot_refresh_volume_list)
def root_user_notification(self):
root_msgBox = QtGui.QMessageBox(
1, i18n("Quick Format"),
i18n("You must be root user to use this application."))
root_msgBox.exec_()
sys.exit()
def __make_initial_selection__(self, index):
self.ui.volumeName.setCurrentIndex(index)
self.set_info()
def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
def __set_custom_widgets__(self):
self.ui.listWidget = QtGui.QListWidget(self)
self.ui.volumeName.setModel(self.ui.listWidget.model())
self.ui.volumeName.setView(self.ui.listWidget)
def __init_signals__(self):
self.connect(self.ui.volumeName, SIGNAL("activated(int)"),
self.set_info)
self.connect(self.ui.btn_format, SIGNAL("clicked()"),
self.format_device)
self.connect(self.ui.btn_cancel, SIGNAL("clicked()"), self.close)
# Formatter signals
self.connect(self.formatter, SIGNAL("format_started()"),
self.slot_format_started)
self.connect(self.formatter, SIGNAL("format_successful()"),
self.slot_format_successful)
self.connect(self.formatter, SIGNAL("format_failed()"),
self.slot_format_failed)
self.connect(self.formatter, SIGNAL("partition_table_error()"),
self.slot_partition_table_error)
def filter_file_system(self, volume):
if volume.get_volume_property(
"DeviceIsPartition") and volume.has_accepted_bus():
return True
def get_volumes(self):
volumes = []
# Get volumes
for v in self.device_notifier.get_all_storages():
# Apply filter
try:
volume = Volume(v)
if self.filter_file_system(volume):
volumes.append(volume)
except:
pass
return volumes
def show_volume_list(self):
self.ui.listWidget.clear()
self.ui.volumeName.hidePopup()
selectedIndex = 0
currentIndex = 0
volumes = self.get_volumes()
if not volumes:
self.no_device_notification()
else:
for volume in volumes:
self.add_volume_to_list(volume)
# select the appropriate volume from list
self.ui.volumeName.setCurrentIndex(selectedIndex)
# Auto-hide notifier
if not self.formatter.formatting and not self.first_run and self.notifier.okButton.isHidden(
):
QtCore.QTimer.singleShot(2000, self.hide_notifier)
self.set_info()
self.first_run = False
def format_confirm_message(self):
"""'Are you sure?' message for format operation"""
format_message = QtGui.QMessageBox()
format_message.setText(
i18n(
"This operation will delete your all data. Are you sure to continue?"
))
format_message.setIcon(QtGui.QMessageBox.Question)
format_message.addButton(ACCEPT, QtGui.QMessageBox.AcceptRole)
format_message.addButton(REJECT, QtGui.QMessageBox.RejectRole)
format_message.exec_()
return format_message.clickedButton().text()
def set_enabled(self, enabled_state):
self.ui.grpNonArgs.setEnabled(enabled_state)
self.ui.btn_format.setEnabled(enabled_state)
self.ui.btn_cancel.setEnabled(enabled_state)
def format_device(self):
""" Starts the formatting operation """
# Temporary variable for controlling equality of volume to format
volume_before_confirm = self.volume_to_format
# Shows a message for confirm format
if self.format_confirm_message() == ACCEPT:
if not self.get_volumes():
self.no_device_notification()
else:
# if device removes while messagebox visible, cancel formatting, press "Yes" though
if volume_before_confirm == self.volume_to_format:
selected_file_system = FILE_SYSTEMS[str(
self.ui.fileSystem.currentText())]
selected_label = self.ui.volumeLabel.text()
self.formatter.set_volume_to_format(
self.volume_to_format, selected_file_system,
selected_label)
self.formatter.start()
def slot_format_started(self):
self.set_enabled(False)
self.notifier.notify(FORMAT_STARTED)
def slot_format_successful(self):
self.notifier.notify(FORMAT_SUCCESSFUL)
self.refresh_volume_list(notify=False)
self.set_enabled(True)
def slot_format_failed(self):
self.notifier.notify(FORMAT_FAILED)
self.set_enabled(True)
def slot_partition_table_error(self):
self.notifier.notify(PARTITION_TABLE_ERROR)
self.set_enabled(True)
def no_device_notification(self):
if self.first_run:
msgBox = QtGui.QMessageBox(
1, i18n("Quick Format"),
i18n("There aren't any removable devices."))
msgBox.exec_()
sys.exit()
else:
self.notifier.notify(NO_DEVICE)
self.ui.volumeLabel.setText("")
def notify_refreshing_device_list(self):
if not self.refreshing_devices and not self.formatter.formatting:
self.refreshing_devices = True
self.notifier.notify(LOADING_DEVICES)
def refresh_volume_list(self, notify=True):
if notify:
self.notify_refreshing_device_list()
self.show_volume_list()
self.refreshing_devices = False
def slot_refresh_volume_list(self, device):
#FIX doesnt work if hal is used in system
if not self.formatter.formatting:
if str(device).find("UDisks") >= 0:
self.refresh_volume_list()
def hide_notifier(self):
self.notifier.animate(start=MIDCENTER, stop=MIDCENTER, direction=OUT)
def generate_file_system_list(self):
self.ui.fileSystem.clear()
# Temporary sapce for file system list
self.tempFileSystems = []
# Get file system list
for fs in FILE_SYSTEMS:
self.tempFileSystems.append(fs)
# Sort file system list
self.tempFileSystems.sort()
self.sortedFileSystems = self.tempFileSystems
# Display file system list in combobox
for fs in self.sortedFileSystems:
self.ui.fileSystem.addItem(fs)
def __find_key(self, dic, val):
"""return the key of a value of a dictionary"""
return [k for k, v in dic.iteritems() if v == val][0]
def set_info(self):
""" Displays the selected volume info on Quickformat.ui """
currentIndex = self.ui.volumeName.currentIndex()
item = self.ui.listWidget.item(currentIndex)
# Get item data (QVariant) convert to Python Object
volume = item.data(32).toPyObject()
# Display volume file system on UI if supported
try:
# find fileSystem index from list
fsIndex = self.ui.fileSystem.findText(
self.__find_key(FILE_SYSTEMS, volume.file_system))
# select fileSystem type
self.ui.fileSystem.setCurrentIndex(fsIndex)
except:
self.ui.fileSystem.setCurrentIndex(0)
print "Cannot match file system. Selecting Ext2 as default."
self.ui.volumeLabel.setText(volume.name)
self.ui.icon.setPixmap(volume.icon)
# Set selected volume as the volume to format
self.volume_to_format = volume
def _prepare_selection_text(self, volume):
if volume.name != "":
return "%s - %s" % (volume.name, volume.path)
return "Volume (%s) - %s" % (volume.file_system, volume.path)
def add_volume_to_list(self, volume):
# Create custom widget
volume_item_widget = VolumeUiItem(volume, self.ui.listWidget)
# Create an empty list widget item
# First parameter is the text shown on the combobox when a selection is made
selectionText = self._prepare_selection_text(volume)
item = QtGui.QListWidgetItem(selectionText, self.ui.listWidget)
# Add whole volume as the item data
item.setData(32, QVariant(volume))
# Set the item's widget to custom widget and append to list
self.ui.listWidget.setItemWidget(item, volume_item_widget)
item.setSizeHint(QSize(200, 70))
class QuickFormat(QtGui.QWidget):
def __init__(self, parent = None, args = None):
QtGui.QWidget.__init__(self, parent)
self.__sysargs = args
self.ui = Ui_QuickFormat()
self.ui.setupUi(self)
self.formatter = Formatter()
self.first_run = True
# Initial selection is the first partition found (see next comment)
self.initial_selection = 0
# If a partition path is given as argument, intial selection becomes the parition given
self.__process_args__()
# Connect Qt Signals
self.__init_signals__()
# Initialize custom widgets such as the new combobox
self.__set_custom_widgets__()
# Initialize the fancy notification widget (Derrived from PDS by Gokmen Goksel)
self.__init_notifier__()
self.show_volume_list()
self.generate_file_system_list()
self.volume_to_format = ""
self.__make_initial_selection__(self.initial_selection)
self.refreshing_devices = False
# Monitor USB ports for any new devices
notifier = Solid.DeviceNotifier().instance()
self.connect(notifier, SIGNAL("deviceAdded(const QString&)"), self.slot_refresh_volume_list)
self.connect(notifier, SIGNAL("deviceRemoved(const QString&)"), self.slot_refresh_volume_list)
def __make_initial_selection__(self, index):
self.ui.volumeName.setCurrentIndex(index)
self.set_info()
def __init_notifier__(self):
self.notifier = Notifier(self)
self.notifier.enableOverlay()
self.notifier.busy.busy()
self.notifier.setStyleSheet("color:white")
self.notifier.adjustSize()
self.notifier.label.adjustSize()
def __set_custom_widgets__(self):
self.ui.listWidget = QtGui.QListWidget(self)
self.ui.volumeName.setModel(self.ui.listWidget.model())
self.ui.volumeName.setView(self.ui.listWidget)
def __process_args__(self):
self.volumePathArg = ""
if len(self.__sysargs) == 2:
if not self.__sysargs[1].startswith("-"):
self.volumePathArg = self.__sysargs[1]
def __init_signals__(self):
self.connect(self.ui.volumeName, SIGNAL("activated(int)"), self.set_info)
self.connect(self.ui.btn_format, SIGNAL("clicked()"), self.format_device)
self.connect(self.ui.btn_cancel, SIGNAL("clicked()"), self.close)
# Formatter signals
self.connect(self.formatter, SIGNAL("format_started()"), self.slot_format_started)
self.connect(self.formatter, SIGNAL("format_successful()"), self.slot_format_successful)
self.connect(self.formatter, SIGNAL("format_failed()"), self.slot_format_failed)
self.connect(self.formatter, SIGNAL("partition_table_error()"), self.slot_partition_table_error)
def filter_file_system(self, volume):
if volume.has_accepted_bus() and volume.has_accepted_drivetype():
return True
def get_volumes(self):
volumes = []
# Get volumes
for v in Solid.Device.listFromType(Solid.StorageDrive.StorageVolume):
# Apply filter
volume = Volume(v)
if self.filter_file_system(volume):
volumes.append(volume)
return volumes
def show_volume_list(self):
self.ui.listWidget.clear()
self.ui.volumeName.hidePopup()
selectedIndex = 0
currentIndex = 0
volumes = self.get_volumes()
if not volumes:
self.no_device_notification()
else:
for volume in volumes:
self.add_volume_to_list(volume)
# select the appropriate volume from list
self.ui.volumeName.setCurrentIndex(selectedIndex)
# Auto-hide notifier
if not self.formatter.formatting and not self.first_run and self.notifier.okButton.isHidden():
QtCore.QTimer.singleShot(2000, self.hide_notifier)
self.set_info()
self.first_run = False
def format_device(self):
""" Starts the formatting operation """
selected_file_system = FILE_SYSTEMS[str(self.ui.fileSystem.currentText())]
selected_label = self.ui.volumeLabel.text()
self.formatter.set_volume_to_format(self.volume_to_format, selected_file_system, selected_label)
self.formatter.start()
def slot_format_started(self):
self.notifier.notify(FORMAT_STARTED)
def slot_format_successful(self):
self.notifier.notify(FORMAT_SUCCESSFUL)
self.refresh_volume_list(notify=False)
def slot_format_failed(self):
self.notifier.notify(FORMAT_FAILED)
def slot_partition_table_error(self):
self.notifier.notify(PARTITION_TABLE_ERROR)
def no_device_notification(self):
if self.first_run:
msgBox = QtGui.QMessageBox(1, i18n("QuickFormat"), i18n("There aren't any removable devices."))
msgBox.exec_()
sys.exit()
else:
self.notifier.notify(NO_DEVICE)
def notify_refreshing_device_list(self):
if not self.refreshing_devices and not self.formatter.formatting:
self.refreshing_devices = True
self.notifier.notify(LOADING_DEVICES)
def refresh_volume_list(self, notify=True):
if notify:
self.notify_refreshing_device_list()
self.show_volume_list()
self.refreshing_devices = False
def slot_refresh_volume_list(self, device):
#FIX doesnt work if hal is used in system
if not self.formatter.formatting:
if str(device).find("UDisks")>=0:
self.refresh_volume_list()
def hide_notifier(self):
self.notifier.animate(start=MIDCENTER, stop=MIDCENTER, direction=OUT)
def generate_file_system_list(self):
self.ui.fileSystem.clear()
# Temporary sapce for file system list
self.tempFileSystems = []
# Get file system list
for fs in FILE_SYSTEMS:
self.tempFileSystems.append(fs)
# Sort file system list
self.tempFileSystems.sort()
self.sortedFileSystems = self.tempFileSystems
# Display file system list in combobox
for fs in self.sortedFileSystems:
self.ui.fileSystem.addItem(fs)
def __find_key(self, dic, val):
"""return the key of a value of a dictionary"""
return [k for k, v in dic.iteritems() if v == val][0]
def set_info(self):
""" Displays the selected volume info on Quickformat.ui """
currentIndex = self.ui.volumeName.currentIndex()
item = self.ui.listWidget.item(currentIndex)
# Get item data (QVariant) convert to Python Object
volume = item.data(32).toPyObject()
# Display volume file system on UI if supported
try:
# find fileSystem index from list
fsIndex = self.ui.fileSystem.findText(
self.__find_key(FILE_SYSTEMS, volume.file_system))
# select fileSystem type
self.ui.fileSystem.setCurrentIndex(fsIndex)
except:
self.ui.fileSystem.setCurrentIndex(0)
print "Cannot match file system. Selecting Ext2 as default."
self.ui.volumeLabel.setText(volume.name)
self.ui.icon.setPixmap(volume.icon)
# Set selected volume as the volume to format
self.volume_to_format = volume
def _prepare_selection_text(self, volume):
if volume.name != "":
return "%s - %s" % (volume.name, volume.path)
return "%s - %s" % (device.name, volume.path)
def add_volume_to_list(self, volume):
# Create custom widget
volume_item_widget = VolumeUiItem(volume, self.ui.listWidget)
# Create an empty list widget item
# First parameter is the text shown on the combobox when a selection is made
selectionText = self._prepare_selection_text(volume)
item = QtGui.QListWidgetItem(selectionText, self.ui.listWidget)
# Add whole volume as the item data
item.setData(32, QVariant(volume))
# Set the item's widget to custom widget and append to list
self.ui.listWidget.setItemWidget(item, volume_item_widget)
item.setSizeHint(QSize(200,70))
if volume.path == self.volumePathArg:
self.initial_selection = self.ui.listWidget.count() - 1