class InputDialog(DialogBase):
"""Input dialog."""
def __init__(self, title='', text='', value=None, value_type=None):
"""Base message box dialog."""
super(InputDialog, self).__init__()
# Widgets
self.label = LabelBase(text)
self.text = LineEditBase()
self.button_ok = ButtonPrimary('Ok')
self.button_cancel = ButtonNormal('Cancel')
# Widget setup
self.setWindowTitle(to_text_string(title))
if value:
self.text.setText(str(value))
# Layouts
layout = QVBoxLayout()
layout_text = QHBoxLayout()
layout_text.addWidget(self.label)
layout_text.addWidget(SpacerHorizontal())
layout_text.addWidget(self.text)
layout_buttons = QHBoxLayout()
layout_buttons.addStretch()
layout_buttons.addWidget(self.button_cancel)
layout_buttons.addWidget(SpacerHorizontal())
layout_buttons.addWidget(self.button_ok)
layout.addLayout(layout_text)
layout.addWidget(SpacerVertical())
layout.addWidget(SpacerVertical())
layout.addLayout(layout_buttons)
self.setLayout(layout)
# Signals
self.button_ok.clicked.connect(self.accept)
self.button_cancel.clicked.connect(self.reject)
class ImportDialog(EnvironmentActionsDialog):
"""Import environment from environment specification dialog."""
CONDA_ENV_FILES = 'Conda environment files (*.yaml *.yml)'
CONDA_SPEC_FILES = 'Conda explicit specification files (*.txt)'
PIP_REQUIREMENT_FILES = 'Pip requirement files (*.txt)'
def __init__(self, parent=None):
"""Import environment from environment specification dialog."""
super(ImportDialog, self).__init__(parent=parent)
self.environments = None
self.env_dirs = None
self.selected_file_filter = None
# Widgets
self.label_name = LabelBase("Name:")
self.label_location = LabelBase("Location:")
self.label_path = LabelBase("Specification File")
self.text_name = LineEditBase()
self.label_prefix = LabelBase("")
self.text_path = LineEditBase()
self.button_path = ButtonNormal("")
self.button_cancel = ButtonNormal('Cancel')
self.button_ok = ButtonPrimary('Import')
# Widgets setup
self.align_labels(
[self.label_name, self.label_location, self.label_path])
self.label_prefix.setObjectName('environment-location')
self.button_path.setObjectName('import')
self.button_ok.setDefault(True)
self.text_path.setPlaceholderText("File to import from")
self.text_name.setPlaceholderText("New environment name")
self.setMinimumWidth(self.BASE_DIALOG_WIDTH)
self.setWindowTitle("Import new environment")
self.text_name.setValidator(self.get_regex_validator())
# Layouts
layout_infile = QHBoxLayout()
layout_infile.addWidget(self.text_path)
layout_infile.addWidget(SpacerHorizontal())
layout_infile.addWidget(self.button_path)
layout_grid = QGridLayout()
layout_grid.addWidget(self.label_name, 0, 0)
layout_grid.addWidget(SpacerHorizontal(), 0, 1)
layout_grid.addWidget(self.text_name, 0, 2)
layout_grid.addWidget(SpacerVertical(), 1, 0)
layout_grid.addWidget(self.label_location, 2, 0)
layout_grid.addWidget(SpacerHorizontal(), 2, 1)
layout_grid.addWidget(self.label_prefix, 2, 2)
layout_grid.addWidget(SpacerVertical(), 3, 0)
layout_grid.addWidget(self.label_path, 4, 0)
layout_grid.addWidget(SpacerHorizontal(), 4, 1)
layout_grid.addLayout(layout_infile, 4, 2)
layout_buttons = QHBoxLayout()
layout_buttons.addStretch()
layout_buttons.addWidget(self.button_cancel)
layout_buttons.addWidget(SpacerHorizontal())
layout_buttons.addWidget(self.button_ok)
layout = QVBoxLayout()
layout.addLayout(layout_grid)
layout.addWidget(SpacerVertical())
layout.addWidget(SpacerVertical())
layout.addLayout(layout_buttons)
self.setLayout(layout)
# Signals
self.button_ok.clicked.connect(self.accept)
self.button_cancel.clicked.connect(self.reject)
self.button_path.clicked.connect(self.choose)
self.text_path.textChanged.connect(self.refresh)
self.text_name.textChanged.connect(self.refresh)
# Setup
self.text_name.setFocus()
self.refresh()
def choose(self):
"""Display file dialog to select environment specification."""
path, selected_filter = getopenfilename(
caption="Import Environment",
basedir=os.path.expanduser('~'),
parent=None,
filters="{0};;{1};;{2}".format(self.CONDA_ENV_FILES,
self.CONDA_SPEC_FILES,
self.PIP_REQUIREMENT_FILES))
if path:
name = self.name
self.selected_file_filter = selected_filter
self.text_path.setText(path)
self.refresh(path)
# Try to get the name key of the file if an environment.yaml file
if selected_filter == self.CONDA_ENV_FILES:
try:
with open(path, 'r') as f:
raw = f.read()
data = yaml.load(raw)
name = data.get('name', name)
except Exception:
pass
self.text_name.setText(name)
def refresh(self, text=''):
"""Update the status of buttons based data entered."""
name = self.name
path = self.path
self.update_location()
if (name and path and os.path.exists(self.path)
and self.is_valid_env_name(name)):
self.button_ok.setDisabled(False)
self.button_ok.setDefault(True)
else:
self.button_ok.setDisabled(True)
self.button_cancel.setDefault(True)
@property
def path(self):
"""Return the content of the selected path if it exists."""
path = None
if os.path.isfile(self.text_path.text()):
path = self.text_path.text()
return path