Append .py when saving files

* Fixes issue #41
This commit is contained in:
iabdalkader 2015-07-09 06:36:26 +02:00
parent 9d51051ba8
commit 80e025dcd7

View File

@ -550,7 +550,14 @@ class OMVGtk:
self.save_button.set_sensitive(False) self.save_button.set_sensitive(False)
self._update_title() self._update_title()
self.update_recent_files() self.update_recent_files()
with open(dialog.get_filename(), "w") as file:
# append .py
filename = dialog.get_filename()
if not filename.endswith(".py"):
filename += ".py"
# save file
with open(filename, "w") as file:
file.write(self.buffer.get_text(self.buffer.get_start_iter(), self.buffer.get_end_iter())) file.write(self.buffer.get_text(self.buffer.get_start_iter(), self.buffer.get_end_iter()))
dialog.destroy() dialog.destroy()