Merge pull request #16 from toshiakiasakura/folded_context

cplt.Multiple can remove usage of mul from arguments.
toshiakiasakura · May 4, 2022 · f5bf3ef · f5bf3ef
2 parents 1c22784 + b1560c5
commit f5bf3ef
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/contextplt/__version__.py b/contextplt/__version__.py
@@ -1,2 +1,2 @@
-__version_info__ = (0,2,8)
+__version_info__ = (0,2,9)
 __version__ = ".".join(map(str, __version_info__))
diff --git a/contextplt/main.py b/contextplt/main.py
@@ -1,10 +1,12 @@
 from dataclasses import dataclass, field
-from typing import Union, Optional, List, Dict, Tuple, Any
+from typing import Union, Optional, List, Dict, Tuple, Any, ClassVar
+import copy
 
 import matplotlib.pyplot as plt
 from matplotlib.lines import Line2D
 import matplotlib.dates as mdates
 
+
 @dataclass(repr=True)
 class Single():
     """Create one figure. Adjust figure with arguments. 
@@ -170,6 +172,9 @@ def __post_init__(self):
         self.fig = plt.figure(figsize=self.figsize,dpi=self.dpi)
         plt.suptitle(self.suptitle)
 
+        self.Single = MulSingle
+        self.Single.mul = self
+
     def set_ax(self,index,xlim=None, ylim=None, 
                xlabel="", ylabel="",title="", rotation : int = 0):
         """Return axis object. 
@@ -201,15 +206,17 @@ def __exit__(self,exc_type, exc_value, exc_traceback):
         plt.savefig(self.save_path, **self.savefig_kargs) if self.save_path else None
         plt.show() if self.show else None
 
+        self.Single.mul = None
+
     def option(self):
         """This method is for additional graphic setting. 
         See DatePlot for example."""
         pass
 
 @dataclass(repr=True)
 class MulSingle():
-    mul : Multiple 
     index : int 
+    mul : ClassVar[Multiple] 
     sharex : Optional[plt.Axes] = None
     sharey : Optional[plt.Axes] = None
     xlim : Optional[List[float]] = None