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

seolpyo_mplchart/test.py

@@ -1,38 +1,187 @@
-import json
+import sys
 from pathlib import Path
-from typing import Literal
+name_pkg = 'seolpyo_mplchart'
+path_pkg = Path(__file__)
+while path_pkg.name != name_pkg:
+    path_pkg = path_pkg.parent
+sys.path = [path_pkg.parent.__str__()] + sys.path
+
+import json
+import tkinter as tk
+from tkinter import filedialog
 
-import matplotlib.pyplot as plt
 import pandas as pd
+import matplotlib.pyplot as plt
+from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
 
+import seolpyo_mplchart as mc
+from seolpyo_mplchart.xl_to_dict import convert
 
-from seolpyo_mplchart import Chart
 
+path_file = path_pkg / 'sample' / 'apple.txt'
+with open(path_file, 'r', encoding='utf-8') as txt:
+    data = json.load(txt)
 
-_name = {'samsung', 'apple'}
-def sample(name: Literal['samsung', 'apple']='samsung'):
-    if name not in _name:
-        print('name should be either samsung or apple.')
-        return
-    file = Path(__file__).parent / f'data/{name}.txt'
-    with open(file, 'r', encoding='utf-8') as txt:
-        data = json.load(txt)
-    data = data
-    df = pd.DataFrame(data)
-
-    c = Chart()
-    if name == 'apple':
-        c.unit_price = '$'
-        c.unit_volume = 'vol'
-        c.digit_price = 3
-        c.label_ma = 'ma{}'
-        c.candleformat = '{}\n\nclose:   {}\nrate:    {}\ncompare: {}\nopen:    {}({})\nhigh:    {}({})\nlow:     {}({})\nvolume:  {}({})'
-        c.volumeformat = '{}\n\nvolume:      {}\nvolume rate: {}'
+df = pd.DataFrame(data[:])
+
+
+def test():
+    import seolpyo_mplchart as mc
+    class C(mc.CursorChart):
+        # fraction = True
+        # watermark = 0
+        theme = 'light'
+        limit_wick = 2000
+        # limit_ma = None
+        # candle_on_ma = False
+        # slider_top = False
+        def on_click(self, e):
+            super().on_click(e)
+            # print(f'{e.button=}')
+            # print(f'{e.button.__str__()=}')
+            if e.button.__str__() == '3':
+                # print('refresh')
+                if self.theme == 'light':
+                    self.theme = 'dark'
+                    self.CONFIG = mc.set_theme(mc.SLIDERCONFIG_EN, theme=self.theme)
+                    # self.CONFIG.FIGURE.RATIO.price = 18
+                    # self.CONFIG.FIGURE.RATIO.volume = 4
+                else:
+                    self.theme = 'light'
+                    self.CONFIG = mc.set_theme(mc.SLIDERCONFIG, theme=self.theme)
+                    # self.CONFIG.FIGURE.RATIO.slider = 9
+                    # self.CONFIG.FIGURE.RATIO.price = 9
+                    # self.CONFIG.FIGURE.RATIO.volume = 9
+                    self.CONFIG.UNIT.digit = 2
+                self.refresh()
+            return
+        # def set_segments(self):
+        #     super().set_segments()
+        #     self.collection_candle.set_linewidth(1.3)
+        #     return
+        # def get_candle_segment(self, *, is_up, x, left, right, top, bottom, high, low):
+        #     if is_up:
+        #         return [
+        #             (x, bottom),
+        #             (x, low),
+        #             (x, bottom),
+        #             (left, bottom),
+        #             (x, bottom),
+        #             (x, top),
+        #             (right, top),
+        #             (x, top),
+        #             (x, high),
+        #         ]
+        #     else:
+        #         return [
+        #             (x, bottom),
+        #             (x, low),
+        #             (x, bottom),
+        #             (right, bottom),
+        #             (x, bottom),
+        #             (x, top),
+        #             (left, top),
+        #             (x, top),
+        #             (x, high),
+        #         ]
+    c = C()
+    c.key_date = '기준일'
+    c.key_open = '시가'
+    c.key_high = '고가'
+    c.key_low = '저가'
+    c.key_close = '종가'
+    c.key_volume = '거래량'
+    # c.volume = None
     c.set_data(df)
+    
     plt.show()
+    return
+
+
+class TkChart:
+    chart = None
+
+    def __init__(self, window: tk.Tk):
+        window.wm_title('seolpyo tk chart')
+        self.window = window
+        # self.window.option_add('맑은고딕 14')  # 모든 위젯 기본 폰트 크기 설정
+        window.protocol('WM_DELETE_WINDOW', lambda *_: (mc.close('all'), window.destroy()))
+
+        self.add_entry()
+        return
+
+    def open_file(self):
+        path_file = filedialog.askopenfilename(
+            title="Select a file",
+            filetypes=(("Xlsx Files", "*.xlsx"), ("All Files", "*.*"),)
+        )
+        # print(f'{path_file=}')
+        if path_file:
+            self.filname.config(state="normal")   # 잠깐 풀고
+            self.filname.delete(0, tk.END)
+            self.filname.insert(0, path_file)
+            self.filname.config(state="readonly") # 다시 잠금
+            
+            if not self.chart:
+                self.add_chart()
+            self.set_chart(path_file)
+            return path_file
+        return
 
+    def add_entry(self):
+        frame = tk.Frame(self.window)
+        frame.grid(column=0, row=0, sticky='w', padx=10, pady=10)
+
+        btn = tk.Button(frame, text='파일 열기', command=lambda *_: self.open_file())
+        btn.grid(column=0, row=0)
+
+        self.filname = tk.Entry(frame, state='readonly', width=100)
+        self.filname.grid(column=1, row=0, padx=10)
+        return
+
+    def add_chart(self):
+        self.chart = mc.SliderChart()
+        self.chart.key_date = '기준일'
+        self.chart.key_open = '시가'
+        self.chart.key_high = '고가'
+        self.chart.key_low = '저가'
+        self.chart.key_close = '종가'
+        self.chart.key_volume = '거래량'
+
+        frame = tk.Frame(self.window)
+        frame.grid(column=0, row=1, sticky='ewsn')
+        self.window.columnconfigure(0, weight=1)
+        self.window.rowconfigure(1, weight=1)
+
+        self.agg = FigureCanvasTkAgg(self.chart.figure, frame)
+        widget = self.agg.get_tk_widget()
+        widget.grid(column=0, row=0, sticky='ewsn')
+        frame.columnconfigure(0, weight=1)
+        frame.rowconfigure(0, weight=1)
+
+        return
+
+    def set_chart(self, path_file):
+        data = convert(path_file)
+        print(f'{len(data)=:,}')
+        df = pd.DataFrame(data)
+        self.chart.set_data(df)
+        return
+
+
+def run():
+    root = tk.Tk()
+    _ = TkChart(root)
+    root.mainloop()
+    return
+
+
+def test_tk():
+    run()
     return
 
 
 if __name__ == '__main__':
-    sample('apple')
+    test()
+    # test_tk()
+

seolpyo_mplchart/utils.py

@@ -1,13 +1,18 @@
 from re import search
 
+
 def convert_num(num):
     if isinstance(num, float) and num % 1: return num
     return int(num)
 
 
 def float_to_str(num: float, digit=0, plus=False):
-    num.__round__(digit)
-    text = f'{num:+,.{digit}f}' if plus else f'{num:,.{digit}f}'
+    if 0 < digit:
+        num.__round__(digit)
+        text = f'{num:+,.{digit}f}' if plus else f'{num:,.{digit}f}'
+    else:
+        num = round(num, digit).__int__()
+        text = f'{num:+,}' if plus else f'{num:,}'
     return text
 
 
@@ -16,7 +21,6 @@ dict_unit = {
     '조':      1_000_000_000_000,
     '억':            100_000_000,
     '만':                 10_000,
-    '천':                  1_000,
 }
 dict_unit_en = {
     'Qd': 1_000_000_000_000_000,
@@ -26,15 +30,20 @@ dict_unit_en = {
     'K':                  1_000,
 }
 
+
 def convert_unit(value, digit=0, word='원'):
+    # print(f'{value=:,}')
     v = abs(value)
     du = dict_unit if search('[가-힣]', word) else dict_unit_en
     for unit, n in du.items():
         if n <= v:
+            # print(f'{n=:,}')
+            # print(f'{unit=}')
             num = value / n
             return f'{float_to_str(num, digit)}{unit} {word}'
     if not value % 1: value = int(value)
     text = f'{float_to_str(value, digit)}{word}'
+    # print(f'{text=}')
     return text
 
 
@@ -42,4 +51,6 @@ if __name__ == '__main__':
     a = 456.123
     print(float_to_str(a))
     print(float_to_str(a, 2))
-    print(float_to_str(a, 6))
+    print(float_to_str(a, 6))
+
+

seolpyo_mplchart/xl_to_dict.py

@@ -0,0 +1,47 @@
+from datetime import datetime, timedelta
+from re import findall
+from zipfile import ZipFile
+
+
+base_date = datetime(1899, 12, 30).date()
+
+
+def convert(path_file: str, sheet=1, row=4, date='A', Open='B', high='C', low='D', close='E', volume='F'):
+    "xlsx 파일에서 주가 정보를 추출합니다."
+    zipfile = ZipFile(path_file)
+    # print(f'{zipfile.filelist=}')
+    a = zipfile.read(f'xl/worksheets/sheet{sheet}.xml')
+    # print(f'{a=}')
+    # print(f'{type(a)=}')
+    b = a.decode('utf-8')
+    list_price: list[dict[str, str|int]] = []
+    for i in findall('<row.+?</row>', b)[row:]:
+        # print()
+        # print(f'{i=}')
+        dt = findall(f'<c r="{date}[0-9].+?<v>([0-9]+)</v>', i)
+        c = findall(f'<c r="{close}[0-9].+?<v>([0-9\.]+)</v>', i)
+        o = findall(f'<c r="{Open}[0-9].+?<v>([0-9\.]+)</v>', i)
+        h = findall(f'<c r="{high}[0-9].+?<v>([0-9\.]+)</v>', i)
+        l = findall(f'<c r="{low}[0-9].+?<v>([0-9\.]+)</v>', i)
+        v = findall(f'<c r="{volume}[0-9].+?<v>([0-9]+)</v>', i)
+        # print(f'{(dt, c, o, h, l, v)=}')
+        if not all([dt, o, h, l, c,]):
+            continue
+        try:
+            dt = base_date + timedelta(int(dt[0]))
+            c = float(c[0])
+            o = float(o[0])
+            h = float(h[0])
+            l = float(l[0])
+        except:
+            continue
+        try:
+            v = float(v[0])
+        except:
+            v = 0
+        # print(f'{(dt, c, o, h, l, v)=}')
+        # if 2020 < dt.year and dt.year < 2024: list_price.append({'기준일': f'{dt}', '종가': c, '시가': o, '고가': h, '저가': l, '거래량': v,})
+        list_price.append({'기준일': f'{dt}', '종가': c, '시가': o, '고가': h, '저가': l, '거래량': v,})
+    # for i in enumerate(list_price, 1): print(f'  {i}')
+    return sorted(list_price, key=lambda x: x['기준일'])
+