def test_is_save_file(self): """ Filename indicates Whyteboard save type """ self.assertTrue(is_save_file("blah.WTbd")) self.assertTrue(is_save_file("blah.wtbd")) self.assertFalse(is_save_file("blahwtbd")) self.assertFalse(is_save_file("/blah.png")) self.assertFalse(is_save_file("wtbd"))
def open_file(self, filename): if filename: if is_save_file(filename): logger.debug("File open with unsaved changes, prompting for save") self.prompt_for_save(self.do_open, args=[filename]) else: self.do_open(filename)
def do_open(self, filename): """ Updates the appropriate variables in the utility class and loads the selected file. """ logger.info("Opening file [%s]", filename) self.filehistory.AddFileToHistory(filename) self.filehistory.Save(self.config) self.config.Flush() self.util.save_last_path(filename) if is_save_file(filename): self.util.load_wtbd(filename) else: self.util.temp_file = filename self.util.load_file()
def on_save_as(self, event=None): """ Prompts for the filename and location to save to. """ wildcard = _("Whyteboard file ") + u"(*.wtbd)|*.wtbd" _dir = Config().get('last_opened_dir') or u"" _file = self.util.filename if not _file: _file = u"" name = file_dialog(self, _("Save Whyteboard As..."), wx.SAVE | wx.OVERWRITE_PROMPT, wildcard, _dir, _file) if name: if not os.path.splitext(name)[1]: # no file extension name += u'.wtbd' if is_save_file(name): self.util.filename = name self.on_save()