use name of TA project as suggestion for file name of exported code

1 files changed, 7 insertions, 2 deletions

diff --git a/turtleblocks.py b/turtleblocks.py
index 365d7af..c0e18d4 100755
--- a/turtleblocks.py
+++ b/turtleblocks.py
@@ -551,10 +551,15 @@ Would you like to save before quitting?'))
             return
         if not pythoncode:
             return
+        # use name of TA project if it has been saved already
+        default_name = self.tw.save_file_name
+        if default_name is None:
+            default_name = _("myproject")
+        elif default_name.endswith(".ta") or default_name.endswith(".tb"):
+            default_name = default_name[:-3]
         save_type = '.py'
-        # TODO use name of TA project if it has been saved already
         (filename, self.tw.load_save_folder) = get_save_name(
-            save_type, self.tw.load_save_folder, 'myproject')
+            save_type, self.tw.load_save_folder, default_name)
         if isinstance(filename, unicode):
             filename = filename.encode('ascii', 'replace')
         if filename is not None: