Merge pull request #2 from max-models/1-get-the-format-from-the-filename-when-saving

Remove extension from savefig

Author: max-models

src/maxplotlib/canvas/canvas.py

@@ -1,3 +1,5 @@
+import os
+
 import matplotlib.pyplot as plt
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
@@ -120,11 +122,12 @@ def add_subplot(self, **kwargs):
     def savefig(
         self,
         filename,
-        extension="pdf",
         backend="matplotlib",
         layers=None,
         layer_by_layer=False,
+        verbose=False,
     ):
+        filename_no_extension, extension = os.path.splitext(filename)
         if backend == "matplotlib":
             if layer_by_layer:
                 layers = []
@@ -133,17 +136,21 @@ def savefig(
                     fig, axs = self.plot(
                         show=False, backend="matplotlib", savefig=True, layers=layers
                     )
-                    fig.savefig(f"{filename}_{layers}.{extension}")
+                    _fn = f"{filename_no_extension}_{layers}.{extension}"
+                    fig.savefig(_fn)
+                    print(f"Saved {_fn}")
             else:
                 if layers is None:
                     layers = self.layers
-                    full_filepath = f"{filename}.{extension}"
+                    full_filepath = filename
                 else:
-                    full_filepath = f"{filename}_{layers}.{extension}"
+                    full_filepath = f"{filename_no_extension}_{layers}.{extension}"
                 fig, axs = self.plot(
                     show=False, backend="matplotlib", savefig=True, layers=layers
                 )
                 fig.savefig(full_filepath)
+                if verbose:
+                    print(f"Saved {full_filepath}")
 
     def plot(self, backend="matplotlib", show=True, savefig=False, layers=None):
         if backend == "matplotlib":