Merge pull request #135 from minrk/splitext

use splitext to remove extension from notebook name
jupyter · Jun 9, 2015 · bb54245 · bb54245
2 parents c85fb52 + 925da82
commit bb54245
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/notebook/static/notebook/js/notebook.js b/notebook/static/notebook/js/notebook.js
@@ -1752,7 +1752,7 @@ define(function (require) {
      * @return {string} This notebook's name (excluding file extension)
      */
     Notebook.prototype.get_notebook_name = function () {
-        var nbname = this.notebook_name.substring(0,this.notebook_name.length-6);
+        var nbname = utils.splitext(this.notebook_name)[0];
         return nbname;
     };
 
@@ -2101,8 +2101,9 @@ define(function (require) {
      * Returns the filename with the appropriate extension, appending if necessary.
      */
     Notebook.prototype.ensure_extension = function (name) {
-        if (!name.match(/\.ipynb$/)) {
-            name = name + ".ipynb";
+        var ext = utils.splitext(this.notebook_path)[1];
+        if (ext.length && name.slice(-ext.length) !== ext) {
+            name = name + ext;
         }
         return name;
     };