def __init__(self, package='', **kwargs):
super(UnpackTab, self).__init__(**kwargs)
layout = QtGui.QGridLayout()
layout.addWidget(QtGui.QLabel("RPZ package:"), 0, 0)
self.package_widget = QtGui.QLineEdit(package, enabled=False)
layout.addWidget(self.package_widget, 0, 1)
browse_pkg = QtGui.QPushButton("Browse")
browse_pkg.clicked.connect(self._browse_pkg)
layout.addWidget(browse_pkg, 0, 2)
layout.addWidget(QtGui.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
ulayout = QtGui.QVBoxLayout()
self.unpackers = QtGui.QButtonGroup()
for i, name in enumerate(n for n, c in self.UNPACKERS):
radio = QtGui.QRadioButton(name)
self.unpackers.addButton(radio, i)
ulayout.addWidget(radio)
layout.addLayout(ulayout, 1, 1, 1, 2)
group = QtGui.QGroupBox(title="Unpacker options")
group_layout = QtGui.QVBoxLayout()
self.unpacker_options = ResizableStack()
self.unpackers.buttonClicked[int].connect(
self.unpacker_options.setCurrentIndex)
scroll = QtGui.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 2, 0, 1, 3)
layout.setRowStretch(2, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(QtGui.QLabel("Select an unpacker to "
"display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
layout.addWidget(QtGui.QLabel("Destination directory:"), 3, 0)
self.directory_widget = QtGui.QLineEdit()
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 3, 1)
browse_dir = QtGui.QPushButton("Browse")
browse_dir.clicked.connect(self._browse_dir)
layout.addWidget(browse_dir, 3, 2)
buttons = QtGui.QHBoxLayout()
buttons.addStretch(1)
self.unpack_widget = QtGui.QPushButton("Unpack experiment",
enabled=False)
self.unpack_widget.clicked.connect(self._unpack)
buttons.addWidget(self.unpack_widget)
layout.addLayout(buttons, 4, 0, 1, 3)
self.setLayout(layout)
self._package_changed()
class UnpackTab(QtWidgets.QWidget):
"""The unpack window, that sets up a .RPZ file in a directory.
"""
UNPACKERS = [
('directory', DirectoryOptions),
('chroot', ChrootOptions),
('docker', DockerOptions),
('vagrant', VagrantOptions),
]
unpacked = QtCore.Signal(str, object)
def __init__(self, package='', **kwargs):
super(UnpackTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("RPZ package:"), 0, 0)
self.package_widget = QtWidgets.QLineEdit(package, enabled=False)
layout.addWidget(self.package_widget, 0, 1)
browse_pkg = QtWidgets.QPushButton("Browse")
browse_pkg.clicked.connect(self._browse_pkg)
layout.addWidget(browse_pkg, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
ulayout = QtWidgets.QVBoxLayout()
self.unpackers = QtWidgets.QButtonGroup()
for i, name in enumerate(n for n, c in self.UNPACKERS):
radio = QtWidgets.QRadioButton(name)
radio.unpacker = name
self.unpackers.addButton(radio, i)
ulayout.addWidget(radio)
layout.addLayout(ulayout, 1, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
self.unpackers.buttonClicked[int].connect(
self.unpacker_options.setCurrentIndex)
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 2, 0, 1, 3)
layout.setRowStretch(2, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select an unpacker to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
layout.addWidget(QtWidgets.QLabel("Destination directory:"), 3, 0)
self.directory_widget = QtWidgets.QLineEdit()
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 3, 1)
browse_dir = QtWidgets.QPushButton("Browse")
browse_dir.clicked.connect(self._browse_dir)
layout.addWidget(browse_dir, 3, 2)
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.unpack_widget = QtWidgets.QPushButton("Unpack experiment",
enabled=False)
self.unpack_widget.clicked.connect(self._unpack)
buttons.addWidget(self.unpack_widget)
layout.addLayout(buttons, 4, 0, 1, 3)
self.setLayout(layout)
self._package_changed()
def replaceable(self):
return not self.package_widget.text()
def _browse_pkg(self):
picked, _ = QtWidgets.QFileDialog.getOpenFileName(
self, "Pick package file", QtCore.QDir.currentPath(),
"ReproZip Packages (*.rpz)")
if picked:
record_usage(browse_pkg=True)
self.package_widget.setText(picked)
self._package_changed()
def _package_changed(self, new_pkg=None):
package = self.package_widget.text()
if package.lower().endswith('.rpz'):
self.directory_widget.setText(package[:-4])
self._directory_changed()
def _browse_dir(self):
picked, _ = QtWidgets.QFileDialog.getSaveFileName(
self, "Pick directory", QtCore.QDir.currentPath())
if picked:
if os.path.exists(picked):
error_msg(self, "This directory already exists", 'warning')
else:
self.directory_widget.setText(picked)
self._directory_changed()
def _directory_changed(self, new_dir=None):
self.unpack_widget.setEnabled(bool(self.directory_widget.text()))
def _unpack(self):
directory = self.directory_widget.text()
if not directory:
return
unpacker = self.unpackers.checkedButton()
if unpacker:
record_usage(unpacker=unpacker.text())
options = self.unpacker_options.currentWidget().options()
if options is None:
return
if handle_error(
self,
reprounzip.unpack(self.package_widget.text(),
unpacker.unpacker, directory, options)):
self.unpacked.emit(os.path.abspath(directory),
options.get('root'))
else:
error_msg(self, "No unpacker selected", 'warning')
def __init__(self, package='', **kwargs):
super(UnpackTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("RPZ package:"), 0, 0)
self.package_widget = QtWidgets.QLineEdit(package, enabled=False)
layout.addWidget(self.package_widget, 0, 1)
browse_pkg = QtWidgets.QPushButton("Browse")
browse_pkg.clicked.connect(self._browse_pkg)
layout.addWidget(browse_pkg, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
ulayout = QtWidgets.QVBoxLayout()
self.unpackers = QtWidgets.QButtonGroup()
for i, name in enumerate(n for n, c in self.UNPACKERS):
radio = QtWidgets.QRadioButton(name)
radio.unpacker = name
self.unpackers.addButton(radio, i)
ulayout.addWidget(radio)
layout.addLayout(ulayout, 1, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
self.unpackers.buttonClicked[int].connect(
self.unpacker_options.setCurrentIndex)
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 2, 0, 1, 3)
layout.setRowStretch(2, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select an unpacker to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
layout.addWidget(QtWidgets.QLabel("Destination directory:"), 3, 0)
self.directory_widget = QtWidgets.QLineEdit()
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 3, 1)
browse_dir = QtWidgets.QPushButton("Browse")
browse_dir.clicked.connect(self._browse_dir)
layout.addWidget(browse_dir, 3, 2)
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.unpack_widget = QtWidgets.QPushButton("Unpack experiment",
enabled=False)
self.unpack_widget.clicked.connect(self._unpack)
buttons.addWidget(self.unpack_widget)
layout.addLayout(buttons, 4, 0, 1, 3)
self.setLayout(layout)
self._package_changed()
class UnpackTab(QtWidgets.QWidget):
"""The unpack window, that sets up a .RPZ file in a directory.
"""
UNPACKERS = [
('directory', DirectoryOptions),
('chroot', ChrootOptions),
('docker', DockerOptions),
('vagrant', VagrantOptions),
]
unpacked = QtCore.Signal(str, object)
def __init__(self, package='', **kwargs):
super(UnpackTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("RPZ package:"), 0, 0)
self.package_widget = QtWidgets.QLineEdit(package, enabled=False)
layout.addWidget(self.package_widget, 0, 1)
browse_pkg = QtWidgets.QPushButton("Browse")
browse_pkg.clicked.connect(self._browse_pkg)
layout.addWidget(browse_pkg, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
ulayout = QtWidgets.QVBoxLayout()
self.unpackers = QtWidgets.QButtonGroup()
for i, name in enumerate(n for n, c in self.UNPACKERS):
radio = QtWidgets.QRadioButton(name)
radio.unpacker = name
self.unpackers.addButton(radio, i)
ulayout.addWidget(radio)
layout.addLayout(ulayout, 1, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
self.unpackers.buttonClicked[int].connect(
self.unpacker_options.setCurrentIndex)
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 2, 0, 1, 3)
layout.setRowStretch(2, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select an unpacker to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
layout.addWidget(QtWidgets.QLabel("Destination directory:"), 3, 0)
self.directory_widget = QtWidgets.QLineEdit()
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 3, 1)
browse_dir = QtWidgets.QPushButton("Browse")
browse_dir.clicked.connect(self._browse_dir)
layout.addWidget(browse_dir, 3, 2)
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.unpack_widget = QtWidgets.QPushButton("Unpack experiment",
enabled=False)
self.unpack_widget.clicked.connect(self._unpack)
buttons.addWidget(self.unpack_widget)
layout.addLayout(buttons, 4, 0, 1, 3)
self.setLayout(layout)
self._package_changed()
def replaceable(self):
return not self.package_widget.text()
def _browse_pkg(self):
picked, _ = QtWidgets.QFileDialog.getOpenFileName(
self, "Pick package file",
QtCore.QDir.currentPath(), "ReproZip Packages (*.rpz)")
if picked:
record_usage(browse_pkg=True)
self.package_widget.setText(picked)
self._package_changed()
def _package_changed(self, new_pkg=None):
package = self.package_widget.text()
if package.lower().endswith('.rpz'):
self.directory_widget.setText(package[:-4])
self._directory_changed()
def _browse_dir(self):
picked, _ = QtWidgets.QFileDialog.getSaveFileName(
self, "Pick directory",
QtCore.QDir.currentPath())
if picked:
if os.path.exists(picked):
error_msg(self, "This directory already exists", 'warning')
else:
self.directory_widget.setText(picked)
self._directory_changed()
def _directory_changed(self, new_dir=None):
self.unpack_widget.setEnabled(bool(self.directory_widget.text()))
def _unpack(self):
directory = self.directory_widget.text()
if not directory:
return
unpacker = self.unpackers.checkedButton()
if unpacker:
record_usage(unpacker=unpacker.text())
options = self.unpacker_options.currentWidget().options()
if options is None:
return
if handle_error(self, reprounzip.unpack(
self.package_widget.text(),
unpacker.unpacker,
directory,
options)):
self.unpacked.emit(os.path.abspath(directory),
options.get('root'))
else:
error_msg(self, "No unpacker selected", 'warning')
def __init__(self, unpacked_directory='', **kwargs):
super(RunTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("Experiment directory:"), 0, 0)
self.directory_widget = QtWidgets.QLineEdit(unpacked_directory)
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 0, 1)
browse = QtWidgets.QPushButton("Browse")
browse.clicked.connect(self._browse)
layout.addWidget(browse, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
self.unpacker_widget = QtWidgets.QLabel("-")
layout.addWidget(self.unpacker_widget, 1, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Input/output files:"), 2, 0,
QtCore.Qt.AlignTop)
self.files_button = QtWidgets.QPushButton("Manage files",
enabled=False)
self.files_button.clicked.connect(self._open_files_manager)
layout.addWidget(self.files_button, 2, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Runs:"), 3, 0, QtCore.Qt.AlignTop)
self.runs_widget = QtWidgets.QListWidget(
selectionMode=QtWidgets.QListWidget.MultiSelection)
layout.addWidget(self.runs_widget, 3, 1, 3, 1)
select_all = QtWidgets.QPushButton("Select All")
select_all.clicked.connect(self.runs_widget.selectAll)
layout.addWidget(select_all, 3, 2)
deselect_all = QtWidgets.QPushButton("Deselect All")
deselect_all.clicked.connect(self.runs_widget.clearSelection)
layout.addWidget(deselect_all, 4, 2)
layout.addWidget(QtWidgets.QLabel("Elevate privileges:"), 6, 0)
self.root = QtWidgets.QComboBox(editable=False)
self.root.addItems(ROOT.TEXT)
layout.addWidget(self.root, 6, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Jupyter integration:"), 7, 0)
self.run_jupyter_notebook = QtWidgets.QCheckBox("Run notebook server",
checked=False,
enabled=False)
layout.addWidget(self.run_jupyter_notebook, 7, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 8, 0, 1, 3)
layout.setRowStretch(8, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select a directory to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.run_widget = QtWidgets.QPushButton("Run experiment")
self.run_widget.clicked.connect(self._run)
buttons.addWidget(self.run_widget)
self.destroy_widget = QtWidgets.QPushButton("Destroy unpacked "
"experiment")
self.destroy_widget.clicked.connect(self._destroy)
buttons.addWidget(self.destroy_widget)
layout.addLayout(buttons, 9, 0, 1, 3)
self.setLayout(layout)
self._directory_changed()
class RunTab(QtWidgets.QWidget):
"""The main window, that allows you to run/change an unpacked experiment.
"""
UNPACKERS = [
('directory', DirectoryOptions),
('chroot', ChrootOptions),
('docker', DockerOptions),
('vagrant', VagrantOptions),
]
directory = None
unpacker = None
def __init__(self, unpacked_directory='', **kwargs):
super(RunTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("Experiment directory:"), 0, 0)
self.directory_widget = QtWidgets.QLineEdit(unpacked_directory)
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 0, 1)
browse = QtWidgets.QPushButton("Browse")
browse.clicked.connect(self._browse)
layout.addWidget(browse, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
self.unpacker_widget = QtWidgets.QLabel("-")
layout.addWidget(self.unpacker_widget, 1, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Input/output files:"), 2, 0,
QtCore.Qt.AlignTop)
self.files_button = QtWidgets.QPushButton("Manage files",
enabled=False)
self.files_button.clicked.connect(self._open_files_manager)
layout.addWidget(self.files_button, 2, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Runs:"), 3, 0, QtCore.Qt.AlignTop)
self.runs_widget = QtWidgets.QListWidget(
selectionMode=QtWidgets.QListWidget.MultiSelection)
layout.addWidget(self.runs_widget, 3, 1, 3, 1)
select_all = QtWidgets.QPushButton("Select All")
select_all.clicked.connect(self.runs_widget.selectAll)
layout.addWidget(select_all, 3, 2)
deselect_all = QtWidgets.QPushButton("Deselect All")
deselect_all.clicked.connect(self.runs_widget.clearSelection)
layout.addWidget(deselect_all, 4, 2)
layout.addWidget(QtWidgets.QLabel("Elevate privileges:"), 6, 0)
self.root = QtWidgets.QComboBox(editable=False)
self.root.addItems(ROOT.TEXT)
layout.addWidget(self.root, 6, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Jupyter integration:"), 7, 0)
self.run_jupyter_notebook = QtWidgets.QCheckBox("Run notebook server",
checked=False,
enabled=False)
layout.addWidget(self.run_jupyter_notebook, 7, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 8, 0, 1, 3)
layout.setRowStretch(8, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select a directory to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.run_widget = QtWidgets.QPushButton("Run experiment")
self.run_widget.clicked.connect(self._run)
buttons.addWidget(self.run_widget)
self.destroy_widget = QtWidgets.QPushButton("Destroy unpacked "
"experiment")
self.destroy_widget.clicked.connect(self._destroy)
buttons.addWidget(self.destroy_widget)
layout.addLayout(buttons, 9, 0, 1, 3)
self.setLayout(layout)
self._directory_changed()
def _browse(self):
picked = QtWidgets.QFileDialog.getExistingDirectory(
self, "Pick directory", QtCore.QDir.currentPath())
if picked:
record_usage(browse_unpacked=True)
self.directory_widget.setText(picked)
self._directory_changed()
def _directory_changed(self, new_dir=None, force=False):
if not force and self.directory_widget.text() == self.directory:
return
self.directory = self.directory_widget.text()
unpacker = reprounzip.check_directory(self.directory)
self.run_jupyter_notebook.setChecked(False)
self.run_jupyter_notebook.setEnabled(False)
self.runs_widget.clear()
if unpacker is not None:
with open(self.directory + '/config.yml') as fp:
self.config = yaml.safe_load(fp)
self.run_widget.setEnabled(True)
self.destroy_widget.setEnabled(True)
self.files_button.setEnabled(True)
self.unpacker = unpacker
self.unpacker_widget.setText(unpacker)
for run in self.config['runs']:
self.runs_widget.addItem(' '.join(
reprounzip.shell_escape(arg) for arg in run['argv']))
self.runs_widget.selectAll()
self.unpacker_options.setCurrentIndex(
dict((n, i)
for i, (n,
w) in enumerate(self.UNPACKERS)).get(unpacker, 4))
if (unpacker == 'docker'
and reprounzip.find_command('reprozip-jupyter') is not None
and reprounzip.is_jupyter(self.directory)):
self.run_jupyter_notebook.setEnabled(True)
self.run_jupyter_notebook.setChecked(True)
else:
self.run_widget.setEnabled(False)
self.destroy_widget.setEnabled(False)
self.files_button.setEnabled(False)
self.unpacker = None
self.unpacker_widget.setText("-")
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
def _run(self):
options = self.unpacker_options.currentWidget().options()
if options is None:
return
runs = sorted(i.row() for i in self.runs_widget.selectedIndexes())
if not runs:
error_msg(self, "No run selected", 'warning')
return
record_usage(run='%d/%d' % (len(runs), self.runs_widget.count()))
handle_error(
self,
reprounzip.run(self.directory,
runs=runs,
unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()],
jupyter=self.run_jupyter_notebook.isChecked(),
**options))
def _destroy(self):
handle_error(
self,
reprounzip.destroy(
self.directory,
unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()]))
self._directory_changed(force=True)
def _open_files_manager(self):
manager = FilesManager(
parent=self,
directory=self.directory_widget.text(),
unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()])
manager.exec_()
def set_directory(self, directory, root=None):
self.root.setCurrentIndex(ROOT.OPTION_TO_INDEX[root])
self.directory_widget.setText(directory)
self._directory_changed(force=True)
def should_exit(self):
if self.unpacker:
r = QtWidgets.QMessageBox.question(
self, "Close Confirmation",
"The experiment is still unpacked with '%s'. Are you sure you "
"want to exit without removing it?" % self.unpacker,
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No)
if r == QtWidgets.QMessageBox.Yes:
record_usage(leave_unpacked=True)
return True
else:
return False
else:
return True
def replaceable(self):
return not self.unpacker
def __init__(self, unpacked_directory='', **kwargs):
super(RunTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("Experiment directory:"), 0, 0)
self.directory_widget = QtWidgets.QLineEdit(unpacked_directory)
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 0, 1)
browse = QtWidgets.QPushButton("Browse")
browse.clicked.connect(self._browse)
layout.addWidget(browse, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
self.unpacker_widget = QtWidgets.QLabel("-")
layout.addWidget(self.unpacker_widget, 1, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Input/output files:"), 2, 0,
QtCore.Qt.AlignTop)
self.files_button = QtWidgets.QPushButton("Manage files",
enabled=False)
self.files_button.clicked.connect(self._open_files_manager)
layout.addWidget(self.files_button, 2, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Runs:"), 3, 0,
QtCore.Qt.AlignTop)
self.runs_widget = QtWidgets.QListWidget(
selectionMode=QtWidgets.QListWidget.MultiSelection)
layout.addWidget(self.runs_widget, 3, 1, 3, 1)
select_all = QtWidgets.QPushButton("Select All")
select_all.clicked.connect(self.runs_widget.selectAll)
layout.addWidget(select_all, 3, 2)
deselect_all = QtWidgets.QPushButton("Deselect All")
deselect_all.clicked.connect(self.runs_widget.clearSelection)
layout.addWidget(deselect_all, 4, 2)
layout.addWidget(QtWidgets.QLabel("Elevate privileges:"), 6, 0)
self.root = QtWidgets.QComboBox(editable=False)
self.root.addItems(ROOT.TEXT)
layout.addWidget(self.root, 6, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Jupyter integration:"),
7, 0)
self.run_jupyter_notebook = QtWidgets.QCheckBox("Run notebook server",
checked=False,
enabled=False)
layout.addWidget(self.run_jupyter_notebook, 7, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 8, 0, 1, 3)
layout.setRowStretch(8, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select a directory to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.run_widget = QtWidgets.QPushButton("Run experiment")
self.run_widget.clicked.connect(self._run)
buttons.addWidget(self.run_widget)
self.destroy_widget = QtWidgets.QPushButton("Destroy unpacked "
"experiment")
self.destroy_widget.clicked.connect(self._destroy)
buttons.addWidget(self.destroy_widget)
layout.addLayout(buttons, 9, 0, 1, 3)
self.setLayout(layout)
self._directory_changed()
class RunTab(QtWidgets.QWidget):
"""The main window, that allows you to run/change an unpacked experiment.
"""
UNPACKERS = [
('directory', DirectoryOptions),
('chroot', ChrootOptions),
('docker', DockerOptions),
('vagrant', VagrantOptions),
]
directory = None
unpacker = None
def __init__(self, unpacked_directory='', **kwargs):
super(RunTab, self).__init__(**kwargs)
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QLabel("Experiment directory:"), 0, 0)
self.directory_widget = QtWidgets.QLineEdit(unpacked_directory)
self.directory_widget.editingFinished.connect(self._directory_changed)
layout.addWidget(self.directory_widget, 0, 1)
browse = QtWidgets.QPushButton("Browse")
browse.clicked.connect(self._browse)
layout.addWidget(browse, 0, 2)
layout.addWidget(QtWidgets.QLabel("Unpacker:"), 1, 0,
QtCore.Qt.AlignTop)
self.unpacker_widget = QtWidgets.QLabel("-")
layout.addWidget(self.unpacker_widget, 1, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Input/output files:"), 2, 0,
QtCore.Qt.AlignTop)
self.files_button = QtWidgets.QPushButton("Manage files",
enabled=False)
self.files_button.clicked.connect(self._open_files_manager)
layout.addWidget(self.files_button, 2, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Runs:"), 3, 0,
QtCore.Qt.AlignTop)
self.runs_widget = QtWidgets.QListWidget(
selectionMode=QtWidgets.QListWidget.MultiSelection)
layout.addWidget(self.runs_widget, 3, 1, 3, 1)
select_all = QtWidgets.QPushButton("Select All")
select_all.clicked.connect(self.runs_widget.selectAll)
layout.addWidget(select_all, 3, 2)
deselect_all = QtWidgets.QPushButton("Deselect All")
deselect_all.clicked.connect(self.runs_widget.clearSelection)
layout.addWidget(deselect_all, 4, 2)
layout.addWidget(QtWidgets.QLabel("Elevate privileges:"), 6, 0)
self.root = QtWidgets.QComboBox(editable=False)
self.root.addItems(ROOT.TEXT)
layout.addWidget(self.root, 6, 1, 1, 2)
layout.addWidget(QtWidgets.QLabel("Jupyter integration:"),
7, 0)
self.run_jupyter_notebook = QtWidgets.QCheckBox("Run notebook server",
checked=False,
enabled=False)
layout.addWidget(self.run_jupyter_notebook, 7, 1, 1, 2)
group = QtWidgets.QGroupBox(title="Unpacker options")
group_layout = QtWidgets.QVBoxLayout()
self.unpacker_options = ResizableStack()
scroll = QtWidgets.QScrollArea(widgetResizable=True)
scroll.setWidget(self.unpacker_options)
group_layout.addWidget(scroll)
group.setLayout(group_layout)
layout.addWidget(group, 8, 0, 1, 3)
layout.setRowStretch(8, 1)
for i, (name, WidgetClass) in enumerate(self.UNPACKERS):
widget = WidgetClass()
self.unpacker_options.addWidget(widget)
self.unpacker_options.addWidget(
QtWidgets.QLabel("Select a directory to display options..."))
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
buttons = QtWidgets.QHBoxLayout()
buttons.addStretch(1)
self.run_widget = QtWidgets.QPushButton("Run experiment")
self.run_widget.clicked.connect(self._run)
buttons.addWidget(self.run_widget)
self.destroy_widget = QtWidgets.QPushButton("Destroy unpacked "
"experiment")
self.destroy_widget.clicked.connect(self._destroy)
buttons.addWidget(self.destroy_widget)
layout.addLayout(buttons, 9, 0, 1, 3)
self.setLayout(layout)
self._directory_changed()
def _browse(self):
picked = QtWidgets.QFileDialog.getExistingDirectory(
self, "Pick directory",
QtCore.QDir.currentPath())
if picked:
record_usage(browse_unpacked=True)
self.directory_widget.setText(picked)
self._directory_changed()
def _directory_changed(self, new_dir=None, force=False):
if not force and self.directory_widget.text() == self.directory:
return
self.directory = self.directory_widget.text()
unpacker = reprounzip.check_directory(self.directory)
self.run_jupyter_notebook.setChecked(False)
self.run_jupyter_notebook.setEnabled(False)
self.runs_widget.clear()
if unpacker is not None:
with open(self.directory + '/config.yml') as fp:
self.config = yaml.safe_load(fp)
self.run_widget.setEnabled(True)
self.destroy_widget.setEnabled(True)
self.files_button.setEnabled(True)
self.unpacker = unpacker
self.unpacker_widget.setText(unpacker)
for run in self.config['runs']:
self.runs_widget.addItem(' '.join(reprounzip.shell_escape(arg)
for arg in run['argv']))
self.runs_widget.selectAll()
self.unpacker_options.setCurrentIndex(
dict((n, i) for i, (n, w) in enumerate(self.UNPACKERS))
.get(unpacker, 4))
if (unpacker == 'docker' and
reprounzip.find_command('reprozip-jupyter') is not None and
reprounzip.is_jupyter(self.directory)):
self.run_jupyter_notebook.setEnabled(True)
self.run_jupyter_notebook.setChecked(True)
else:
self.run_widget.setEnabled(False)
self.destroy_widget.setEnabled(False)
self.files_button.setEnabled(False)
self.unpacker = None
self.unpacker_widget.setText("-")
self.unpacker_options.setCurrentIndex(len(self.UNPACKERS))
def _run(self):
options = self.unpacker_options.currentWidget().options()
if options is None:
return
runs = sorted(i.row() for i in self.runs_widget.selectedIndexes())
if not runs:
error_msg(self, "No run selected", 'warning')
return
record_usage(run='%d/%d' % (len(runs), self.runs_widget.count()))
handle_error(self, reprounzip.run(
self.directory, runs=runs,
unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()],
jupyter=self.run_jupyter_notebook.isChecked(),
**options))
def _destroy(self):
handle_error(self, reprounzip.destroy(
self.directory, unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()]))
self._directory_changed(force=True)
def _open_files_manager(self):
manager = FilesManager(
parent=self,
directory=self.directory_widget.text(),
unpacker=self.unpacker,
root=ROOT.INDEX_TO_OPTION[self.root.currentIndex()])
manager.exec_()
def set_directory(self, directory, root=None):
self.root.setCurrentIndex(ROOT.OPTION_TO_INDEX[root])
self.directory_widget.setText(directory)
self._directory_changed(force=True)
def should_exit(self):
if self.unpacker:
r = QtWidgets.QMessageBox.question(
self, "Close Confirmation",
"The experiment is still unpacked with '%s'. Are you sure you "
"want to exit without removing it?" % self.unpacker,
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No)
if r == QtWidgets.QMessageBox.Yes:
record_usage(leave_unpacked=True)
return True
else:
return False
else:
return True
def replaceable(self):
return not self.unpacker