class BaseSaveFileDialog(BaseFileFolderDialog, object):
def __init__(self,
name='SaveFile',
title='Save File',
size=(200, 125),
fixed_size=False,
frame_less=True,
hide_title=False,
parent=None,
use_app_browser=False):
parent = parent or dcc.get_main_window()
super(BaseSaveFileDialog,
self).__init__(name=name,
title=title,
size=size,
fixed_size=fixed_size,
frame_less=frame_less,
hide_title=hide_title,
use_app_browser=use_app_browser,
parent=parent)
self._open_button.setText('Save')
size = QSize(42, 24)
self.new_directory_button = QPushButton('New')
self.new_directory_button.setToolTip('Create new directory')
self.new_directory_button.setMinimumSize(size)
self.new_directory_button.setMaximumWidth(size)
self.new_directory_button.clicked.connect(self.create_new_directory)
self.grid.itemAtPosition(0, 1).addWidget(self.new_directory_button, 0,
5)
def accept(self, *args, **kwargs):
selected_file, selected_dir, selected_filename = self.get_result()
if not os.path.isdir(selected_file):
if os.path.exists(selected_file):
message_box = QMessageBox()
message_box.setWindowTitle('Confirm File Selection')
message_box.setText(
'File "%s" exists.\nDo you want to overwrite it?' %
selected_file)
message_box.setStandardButtons(QMessageBox.Yes
| QMessageBox.No)
message_box.setDefaultButton(QMessageBox.No)
rv = message_box.exec_()
if rv == QMessageBox.Yes and not os.path.isdir(selected_file):
super(BaseSaveFileDialog, self).accept()
else:
super(BaseSaveFileDialog, self).accept()
def create_new_directory(self):
name, ok = QInputDialog.getText(self, 'New directory name', 'Name:',
QLineEdit.Normal, 'New Directory')
if ok and name:
path = os.path.join(self.directory, name)
if os.path.exists(path):
msg_box = QMessageBox(self)
msg_box.setWindowTitle('Error')
msg_box.setText('Directory already exists')
msg_box.exec_()
else:
try:
os.makedirs(path)
self.update_view()
except os.error as e:
msg_box = QMessageBox(self)
msg_box.setWindowTitle('Error')
msg_box.setText('Cannot create directory')
msg_box.exec_()
def create_flat_button(
icon=None,
icon_size=None,
name='',
text=200,
background_color=[54, 51, 51],
ui_color=68,
border_color=180,
push_col=120,
checkable=True,
w_max=None,
w_min=None,
h_max=None,
h_min=None,
policy=None,
tip=None,
flat=True,
hover=True,
destroy_flag=False,
context=None,
):
btn = QPushButton()
btn.setText(name)
btn.setCheckable(checkable)
if icon:
if isinstance(icon, QIcon):
btn.setIcon(icon)
else:
btn.setIcon(QIcon(icon))
btn.setFlat(flat)
if flat:
change_button_color(button=btn,
text_color=text,
bg_color=ui_color,
hi_color=background_color,
mode='button',
hover=hover,
destroy=destroy_flag,
ds_color=border_color)
btn.toggled.connect(
lambda: change_button_color(button=btn,
text_color=text,
bg_color=ui_color,
hi_color=background_color,
mode='button',
toggle=True,
hover=hover,
destroy=destroy_flag,
ds_color=border_color))
else:
change_button_color(button=btn,
text_color=text,
bg_color=background_color,
hi_color=push_col,
mode='button',
hover=hover,
destroy=destroy_flag,
ds_color=border_color)
if w_max:
btn.setMaximumWidth(w_max)
if w_min:
btn.setMinimumWidth(w_min)
if h_max:
btn.setMaximumHeight(h_max)
if h_min:
btn.setMinimumHeight(h_min)
if icon_size:
btn.setIconSize(QSize(*icon_size))
if policy:
btn.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Expanding)
if tip:
btn.setToolTip(tip)
if context:
btn.setContextMenuPolicy(Qt.CustomContextMenu)
btn.customContextMenuRequested.connect(context)
return btn
