seolpyo-mplchart 0.1.3.1__py3-none-any.whl → 2.0.0.3__py3-none-any.whl

seolpyo_mplchart/base.py

@@ -1,10 +1,13 @@
+from re import search
+
 import matplotlib.pyplot as plt
 import matplotlib.style as mplstyle
 from matplotlib.axes import Axes
 from matplotlib.backends.backend_agg import FigureCanvasAgg
+from matplotlib.figure import Figure as Fig
 
 
-from .utils import convert_unit
+from .utils import dict_unit, dict_unit_en
 
 
 try: plt.switch_backend('TkAgg')
@@ -17,95 +20,95 @@ except: pass
 mplstyle.use('fast')
 
 
-class Base:
+def convert_unit(value: float, digit=0, word='원'):
+    v = value.__abs__()
+    du = dict_unit if search('[가-힣]', word) else dict_unit_en
+    for unit, n in du.items():
+        if n <= v:
+            num = (value / n).__round__(digit)
+            if not num % 1: num = int(num)
+            return f'{num:,}{unit} {word}'
+    value = value.__round__(digit)
+    if not value % 1: value = int(value)
+    elif value < 10: digit = 2
+    text = f'{value:,}{word}'
+    return text
+
+
+class Figure(Fig):
     canvas: FigureCanvasAgg
-    unit_price, unit_volume = ('원', '주')
 
-    figsize = (12, 6)
-    ratio_ax_slider, ratio_ax_legend, ratio_ax_price, ratio_ax_volume = (3, 2, 18, 5)
+
+class Base:
+    figure: Figure
+
+    figsize = (14, 7)
+    ratio_ax_legend, ratio_ax_price, ratio_ax_volume = (2, 18, 5)
     adjust = dict(
         top=0.95, bottom=0.05, left=0.01, right=0.93, # 여백
         wspace=0, hspace=0  # 플롯간 간격
     )
-    color_grid = '#d0d0d0'
-    color_background = '#fafafa'
 
-    slider_top = True
     title = 'seolpyo mplchart'
+    color_background = '#fafafa'
+    gridKwargs = {}
+    color_tick, color_tick_label = ('k', 'k')
+
+    unit_price, unit_volume = ('원', '주')
 
     def __init__(self, *args, **kwargs):
         # 기본 툴바 비활성화
         plt.rcParams['toolbar'] = 'None'
+        # plt.rcParams['figure.dpi'] = 600
 
-        self._set_plot()
+        self._get_plot()
         return
 
-    def _set_plot(self):
-        if self.slider_top:
-            fig, ax = plt.subplots(
-                4, # row 수
-                figsize=self.figsize, # 기본 크기
-                height_ratios=(self.ratio_ax_slider, self.ratio_ax_legend, self.ratio_ax_price, self.ratio_ax_volume) # row 크기 비율
-            )
-            ax: list[Axes]
-            ax_slider, ax_legend, ax_price, ax_volume = ax
-        else:
-            fig, ax = plt.subplots(
-                5, # row 수
-                figsize=self.figsize, # 기본 크기
-                height_ratios=(self.ratio_ax_legend, self.ratio_ax_price, self.ratio_ax_volume, self.ratio_ax_legend, self.ratio_ax_slider) # row 크기 비율
-            )
-            ax: list[Axes]
-            ax_legend, ax_price, ax_volume, ax_none, ax_slider = ax
-            # 사용하지 않는 axes 숨기기
-            ax_none.axis('off')
-        ax_legend.axis('off')
-
-        ax_slider.xaxis.set_animated(True)
-        ax_slider.yaxis.set_animated(True)
-
-        ax_price.xaxis.set_animated(True)
-        ax_price.yaxis.set_animated(True)
-
-        ax_volume.xaxis.set_animated(True)
-        ax_volume.yaxis.set_animated(True)
-
-        fig.canvas.manager.set_window_title(f'{self.title}')
+    def _get_plot(self):
+        self.figure, axes = plt.subplots(
+            3, # row 수
+            figsize=self.figsize, # 기본 크기
+            height_ratios=(self.ratio_ax_legend, self.ratio_ax_price, self.ratio_ax_volume) # row 크기 비율
+        )
+        axes: list[Axes]
+        self.ax_legend, self.ax_price, self.ax_volume = axes
+        self.ax_legend.set_label('legend ax')
+        self.ax_price.set_label('price ax')
+        self.ax_volume.set_label('volume ax')
+
+        self.figure.canvas.manager.set_window_title(f'{self.title}')
+        self.figure.set_facecolor(self.color_background)
 
         # 플롯간 간격 제거(Configure subplots)
-        fig.subplots_adjust(**self.adjust)
+        self.figure.subplots_adjust(**self.adjust)
+
+        self.ax_legend.set_axis_off()
 
         # y ticklabel foramt 설정
-        ax_slider.yaxis.set_major_formatter(lambda x, _: convert_unit(x, word=self.unit_price))
-        ax_price.yaxis.set_major_formatter(lambda x, _: convert_unit(x, word=self.unit_price))
-        ax_volume.yaxis.set_major_formatter(lambda x, _: convert_unit(x, word=self.unit_volume))
+        self.ax_price.yaxis.set_major_formatter(lambda x, _: convert_unit(x, word=self.unit_price, digit=2))
+        self.ax_volume.yaxis.set_major_formatter(lambda x, _: convert_unit(x, word=self.unit_volume, digit=2))
 
+        gridKwargs = {'visible': True, 'linewidth': 0.7, 'color': '#d0d0d0', 'linestyle': '-', 'dashes': (1, 0)}
+        gridKwargs.update(self.gridKwargs)
         # 공통 설정
-        for a in [ax_slider, ax_price, ax_volume]:
+        for ax in (self.ax_price, self.ax_volume):
+            ax.xaxis.set_animated(True)
+            ax.yaxis.set_animated(True)
+
+            # x tick 외부 눈금 표시하지 않기
+            ax.xaxis.set_ticks_position('none')
+            # x tick label 제거
+            ax.set_xticklabels([])
             # y tick 우측으로 이동
-            a.tick_params(left=False, right=True, labelleft=False, labelright=True)
-            # 차트 영역 배경 색상
-            a.set_facecolor(self.color_background)
-            # grid(구분선, 격자) 그리기
-            a.grid(True, color=self.color_grid, linewidth=1)
-            # x tick 제거
-            a.set_xticklabels([])
-
-        self.fig, self.canvas = (fig, fig.canvas)
-        self.ax_slider, self.ax_legend, self.ax_price, self.ax_volume = (ax_slider, ax_legend, ax_price, ax_volume)
+            ax.tick_params(left=False, right=True, labelleft=False, labelright=True, colors=self.color_tick_label)
+            # Axes 외곽선 색 변경
+            for i in ['top', 'bottom', 'left', 'right']: ax.spines[i].set_color(self.color_tick)
 
-        return self.set_plot()
+            # 차트 영역 배경 색상
+            ax.set_facecolor(self.color_background)
 
-    def set_plot(self):
-        "This function works after set plot process is done."
+            # grid(구분선, 격자) 그리기
+            # 어째서인지 grid의 zorder 값을 선언해도 1.6을 값으로 한다.
+            ax.grid(**gridKwargs)
         return
 
-
-class Chart(Base):
-    pass
-
-
-if __name__ == '__main__':
-    Base()
-    
-    plt.show()