def __open_or_create_test_file(self, style):
if self.__cur_po_class:
folder = self.GetTopLevelParent().get_root_folder()
if not folder:
folder = os.path.dirname(self.__cur_po_class.file_path)
elif RootFolder.TESTS_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.TESTS_FOLDER)
dialog = FileDialog(self,
defaultDir=folder,
style=style,
wildcard='*.py')
if dialog.ShowModal() == ID_OK:
test_file = dialog.GetPath()
load_file = style == FD_OPEN
filename = os.path.basename(test_file)
if StringUtils.is_test_file_name_correct(test_file):
test_file_ui = TestFileUI(self.tabs, test_file,
self.__cur_po_class, load_file)
self.tabs.AddPage(test_file_ui, filename)
self.tabs.SetSelection(self.tabs.GetPageCount() - 1)
if style == FD_SAVE:
test_file_ui.set_file_was_changed()
else:
show_dialog_bad_name(self, filename, 'my_first_test.py')
else:
show_dialog(self, u'Please select class file.',
u'Class file was not opened')
def onFileInsertProject(self, event: CommandEvent):
"""
Insert a project into this one
Args:
event:
"""
PyutUtils.displayWarning(_("The project insert is experimental, "
"use it at your own risk.\n"
"You risk a shapes ID duplicate with "
"unexpected results !"), parent=self)
if (self._treeNotebookHandler.getCurrentProject()) is None:
PyutUtils.displayError(_("No project to insert this file into !"))
return
# Ask which project to insert
defaultDirectory: str = self._currentDirectoryHandler.currentDirectory
dlg = FileDialog(self._parent, _("Choose a project"), defaultDirectory, "", "*.put", FD_OPEN)
if dlg.ShowModal() != ID_OK:
dlg.Destroy()
return False
self._currentDirectoryHandler.currentDirectory = dlg.GetPath()
filename = dlg.GetPath()
dlg.Destroy()
self.logger.warning(f'inserting file: {filename}')
# Insert the specified files
try:
self._treeNotebookHandler.insertFile(filename)
except (ValueError, Exception) as e:
PyutUtils.displayError(_(f"An error occurred while loading the project! {e}"))
def choose_file(parent, message, d=None):
folder = FileDialog(parent,
message=message,
defaultFile='sample.txt',
style=FD_SAVE)
if folder.ShowModal() == ID_OK:
return folder.GetPath()
def __open_class(self, evt):
folder = self.GetTopLevelParent().get_root_folder()
if folder:
if RootFolder.PO_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.PO_FOLDER)
dialog = FileDialog(self, defaultDir=folder, wildcard=u'*.py')
if dialog.ShowModal() == ID_OK:
self.__load_po_class(dialog.GetPath())
else:
show_dialog_path_doesnt_exist(self, folder)
def getFilePath(wildcard, title):
app = App(None)
style = FD_OPEN | FD_FILE_MUST_EXIST
dialog = FileDialog(None, title, wildcard=wildcard, style=style)
if dialog.ShowModal() == ID_OK:
path = dialog.GetPath()
else:
path = None
dialog.Destroy()
return path
def __load_tests_from_files(self, evt):
folder = self.__get_safe_path_from_root_folder(RootFolder.TESTS_FOLDER)
if folder:
dialog = FileDialog(self,
defaultDir=folder,
style=FD_OPEN | FD_FILE_MUST_EXIST
| FD_MULTIPLE,
wildcard=u'*.py')
if dialog.ShowModal() == ID_OK:
self.__load_tests_to_tree(file_paths=dialog.GetPaths())
else:
show_dialog_path_doesnt_exist(self, folder)
def OnSave(self, event):
file_choices = "PNG (*.png)|*.png"
dlg = FileDialog(self,
message="Zapisz wykres jako...",
defaultDir=getcwd(),
defaultFile="plot.png",
wildcard=file_choices,
style=SAVE)
if dlg.ShowModal() == ID_OK:
path = dlg.GetPath()
self.canvas.print_figure(path, dpi=100)
def __on_select_file(self, evt):
folder = self.__get_safe_path_from_root_folder(RootFolder.REPORTS)
obj = evt.GetEventObject()
txt_ctrl = None
if obj == self.btn_select_html:
wildcard = u'*.html'
txt_ctrl = self.txt_html_report
elif obj == self.btn_select_xml:
wildcard = u'*.xml'
txt_ctrl = self.txt_xml_report
else:
wildcard = u'*.*'
dialog = FileDialog(self,
defaultDir=folder,
style=FD_SAVE | FD_OVERWRITE_PROMPT,
wildcard=wildcard)
if dialog.ShowModal() == ID_OK and txt_ctrl:
txt_ctrl.SetValue(dialog.GetPath())
def OnAdd(self, event):
'''
dialog = AddDialog("Add Option", None)
if dialog.ShowModal() == wx.ID_OK:
print dialog.GetOption()
self.listCtrl.AppendAndEnsureVisible(dialog.GetOption())
'''
wildcard = "All Files (*.*) | *.* |" \
"source (*.py) | *.py |" \
"Doc Files (*.doc) | *.doc"
dialog = FileDialog(self, "Attach files", os.getcwd(), "", "*.*", FileDialog.OPEN | FileDialog.MULTIPLE)
if dialog.ShowModal() == wx.ID_OK:
fullList = dialog.GetPaths()
self.attachments.copy(fullList)
for item in self.attachments.GetFilenames():
self.listCtrl.Append(item)
dialog.Destroy()
def _askForFileImport(self,
multiSelect: bool = False,
startDirectory: str = None):
"""
Called by plugin to ask which file must be imported
Args:
multiSelect: True to allow multi-file selection
Returns:
filename or "" for multiple=False, fileNames[] or
[] else
"",
[] indicates that the user pressed the cancel button
"""
inputFormat: PyutPlugin.INPUT_FORMAT_TYPE = self.getInputFormat()
defaultDir: str = startDirectory
if defaultDir is None:
defaultDir = self._ctrl.getCurrentDir()
if multiSelect:
dlg = FileDialog(self._umlFrame,
"Choose files to import",
wildcard=inputFormat[0] + " (*." +
inputFormat[1] + ")|*." + inputFormat[1],
defaultDir=defaultDir,
style=FD_OPEN | FD_FILE_MUST_EXIST | FD_MULTIPLE
| FD_CHANGE_DIR)
dlg.ShowModal()
if dlg.GetReturnCode() == ID_CANCEL:
return [], ""
return dlg.GetFilenames(), dlg.GetDirectory()
else:
file = FileSelector("Choose a file to import",
wildcard=inputFormat[0] + " (*." +
inputFormat[1] + ")|*." + inputFormat[1],
flags=FD_OPEN | FD_FILE_MUST_EXIST
| FD_CHANGE_DIR)
return file
def loadFile(self, filename: str = ""):
"""
Load the specified filename; This is externally available so that
we can open a file from the command line
Args:
filename: Its name
"""
# Make a list to be compatible with multi-files loading
fileNames = [filename]
currentDir: str = self._mediator.getCurrentDir()
# TODO This is bad practice to do something different based on input
if filename == "":
dlg = FileDialog(self._parent, _("Choose a file"), currentDir, "", "*.put", FD_OPEN | FD_MULTIPLE)
if dlg.ShowModal() != ID_OK:
dlg.Destroy()
return False
fileNames = dlg.GetPaths()
self._currentDirectoryHandler.currentDirectory = fileNames[0]
dlg.Destroy()
self.logger.info(f"loading file(s) {filename}")
# Open the specified files
for filename in fileNames:
try:
if self._treeNotebookHandler.openFile(filename):
# Add to last opened files list
self._preferences.addNewLastOpenedFilesEntry(filename)
self.setLastOpenedFilesItems()
self._mediator.updateTitle()
except (ValueError, Exception) as e:
PyutUtils.displayError(_("An error occurred while loading the project !"))
self.logger.error(f'{e}')
def _loadFile(self, filename=""):
"""
Load the specified filename
Args:
filename: Its name
"""
# Make a list to be compatible with multi-files loading
fileNames = [filename]
# Ask which filename to load ?
if filename == "":
dlg = FileDialog(self, _("Choose a file"), self._lastDir, "",
"*.put", FD_OPEN | FD_MULTIPLE)
if dlg.ShowModal() != ID_OK:
dlg.Destroy()
return False
self.updateCurrentDir(dlg.GetPath())
fileNames = dlg.GetPaths()
dlg.Destroy()
self.logger.info(f"loading file(s) {filename}")
# Open the specified files
for filename in fileNames:
try:
if self._mainFileHandlingUI.openFile(filename):
# Add to last opened files list
self._prefs.addNewLastOpenedFilesEntry(filename)
self.__setLastOpenedFilesItems()
self._ctrl.updateTitle()
except (ValueError, Exception) as e:
PyutUtils.displayError(
_("An error occurred while loading the project !"),
parent=self)
self.logger.error(f'{e}')
def _OnMnuFileInsertProject(self, event: CommandEvent):
"""
Insert a project into this one
Args:
event:
"""
PyutUtils.displayWarning(_("The project insert is experimental, "
"use it at your own risk.\n"
"You risk a shapes ID duplicate with "
"unexpected results !"),
parent=self)
if (self._mainFileHandlingUI.getCurrentProject()) is None:
PyutUtils.displayError(_("No project to insert this file into !"),
parent=self)
return
# Ask which project to insert
dlg = FileDialog(self, _("Choose a file"), self._lastDir, "", "*.put",
FD_OPEN)
if dlg.ShowModal() != ID_OK:
dlg.Destroy()
return False
self.updateCurrentDir(dlg.GetPath())
filename = dlg.GetPath()
dlg.Destroy()
print(("inserting file", str(filename)))
# Insert the specified files
try:
self._mainFileHandlingUI.insertFile(filename)
except (ValueError, Exception) as e:
PyutUtils.displayError(
_(f"An error occurred while loading the project! {e}"),
parent=self)
def saveFileAs(self):
"""
Ask for a filename and save the diagram data
Returns:
`True` if the save succeeds else `False`
"""
if self._mediator.isInScriptMode():
PyutUtils.displayError(_("Save File As is not accessible in script mode !"))
return
# Test if no diagram exists
if self._mediator.getDiagram() is None:
PyutUtils.displayError(_("No diagram to save !"), _("Error"))
return
# Ask for filename
filenameOK = False
# TODO revisit this to figure out how to get rid of Pycharm warning 'dlg referenced before assignment'
# Bad thing is dlg can be either a FileDialog or a MessageDialog
dlg: DialogType = cast(DialogType, None)
while not filenameOK:
dlg = FileDialog(self.__parent,
defaultDir=self.__parent.getCurrentDir(),
wildcard=_("Pyut file (*.put)|*.put"),
style=FD_SAVE | FD_OVERWRITE_PROMPT)
# Return False if canceled
if dlg.ShowModal() != ID_OK:
dlg.Destroy()
return False
# Find if a specified filename is already opened
filename = dlg.GetPath()
if len([project for project in self._projects if project.getFilename() == filename]) > 0:
dlg = MessageDialog(self.__parent,
_("Error ! The filename '%s" + "' correspond to a project which is currently opened !" +
" Please choose another filename !") % str(filename),
_("Save change, filename error"), OK | ICON_ERROR)
dlg.ShowModal()
dlg.Destroy()
return
filenameOK = True
project = self._currentProject
project.setFilename(dlg.GetPath())
project.saveXmlPyut()
# Modify notebook text
for i in range(self.__notebook.GetPageCount()):
frame = self.__notebook.GetPage(i)
document = [document for document in project.getDocuments() if document.getFrame() is frame]
if len(document) > 0:
document = document[0]
if frame in project.getFrames():
diagramTitle: str = document.getTitle()
shortName: str = self.shortenNotebookPageFileName(diagramTitle)
self.__notebook.SetPageText(i, shortName)
else:
self.logger.info("Not updating notebook in FileHandling")
self.__parent.updateCurrentDir(dlg.GetPath())
project.setModified(False)
dlg.Destroy()
return True