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()
