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

seolpyo_mplchart/_utils/theme/__init__.py

@@ -0,0 +1,15 @@
+from typing import Literal
+
+from ..._config import ConfigData, SliderConfigData
+
+from . import light, dark
+
+
+def set_theme(config: ConfigData|SliderConfigData, theme: Literal['light', 'dark']='dark'):
+    if theme == 'light':
+        config = light.set_theme(config)
+    elif theme == 'dark':
+        config = dark.set_theme(config)
+
+    return config
+

seolpyo_mplchart/_utils/theme/dark.py

@@ -0,0 +1,57 @@
+from ..._config import ConfigData, SliderConfigData
+
+
+def set_theme(config: ConfigData|SliderConfigData):
+    # axes
+    config.FIGURE.facecolor = '#0f0f0f'
+    config.AX.facecolor = '#0f0f0f'
+    config.AX.TICK.edgecolor = '#dbdbdb'
+    config.AX.TICK.fontcolor = '#dbdbdb'
+    config.AX.GRID.color = '#1c1c1c'
+
+    # candle
+    config.CANDLE.line_color = 'w'
+    config.CANDLE.FACECOLOR.bull_rise = '#089981'
+    config.CANDLE.FACECOLOR.bull_fall = '#0f0f0f'
+    config.CANDLE.FACECOLOR.bear_fall = '#f23645'
+    config.CANDLE.FACECOLOR.bear_rise = '#0f0f0f'
+
+    config.CANDLE.EDGECOLOR.bull_rise = '#089981'
+    config.CANDLE.EDGECOLOR.bull_fall = '#089981'
+    config.CANDLE.EDGECOLOR.bear_fall = '#f23645'
+    config.CANDLE.EDGECOLOR.bear_rise = '#f23645'
+    config.CANDLE.EDGECOLOR.doji = 'w'
+
+    # volume
+    config.VOLUME.FACECOLOR.rise = '#2A8076'
+    config.VOLUME.FACECOLOR.fall = '#BE4F58'
+    config.VOLUME.FACECOLOR.doji = '#82828A'
+
+    config.VOLUME.EDGECOLOR.rise = '#2A8076'
+    config.VOLUME.EDGECOLOR.fall = '#BE4F58'
+    config.VOLUME.EDGECOLOR.doji = '#82828A'
+
+    # ma
+    config.MA.color_default = 'w'
+    config.MA.color_list = ['#FFB300', '#03A9F4', '#AB47BC', '#8BC34A', '#EF5350']
+
+    # text
+    config.CURSOR.TEXT.BBOX.facecolor = '#3d3d3d'
+    config.CURSOR.TEXT.BBOX.edgecolor = '#ffffff'
+    config.CURSOR.TEXT.color = '#ffffff'
+
+    # box
+    config.CURSOR.BOX.edgecolor = 'w'
+
+    # line
+    config.CURSOR.CROSSLINE.edgecolor = '#9c9c9c'
+
+    # wartermark
+    config.FIGURE.WATERMARK.color = 'w'
+
+    if getattr(config, 'SLIDER', None):
+        config.SLIDER.NAV.edgecolor = "#00A6FF"
+        config.SLIDER.NAV.facecolor = '#FFFFFF4D'
+
+    return config
+

seolpyo_mplchart/_utils/theme/light.py

@@ -0,0 +1,56 @@
+from ..._config import ConfigData, SliderConfigData
+
+
+def set_theme(config: ConfigData|SliderConfigData):
+    # axes
+    config.FIGURE.facecolor = '#fafafa'
+    config.AX.facecolor = '#fafafa'
+    config.AX.TICK.edgecolor = 'k'
+    config.AX.TICK.fontcolor = 'k'
+    config.AX.GRID.color = '#d0d0d0'
+
+    # candle
+    config.CANDLE.line_color = 'k'
+    config.CANDLE.FACECOLOR.bull_rise = '#FF2400'
+    config.CANDLE.FACECOLOR.bull_fall = 'w'
+    config.CANDLE.FACECOLOR.bear_fall = '#1E90FF'
+    config.CANDLE.FACECOLOR.bear_rise = 'w'
+
+    config.CANDLE.EDGECOLOR.bull_rise = '#FF2400'
+    config.CANDLE.EDGECOLOR.bull_fall = '#FF2400'
+    config.CANDLE.EDGECOLOR.bear_fall = '#1E90FF'
+    config.CANDLE.EDGECOLOR.bear_rise = '#1E90FF'
+    config.CANDLE.EDGECOLOR.doji = 'k'
+
+    # volume
+    config.VOLUME.FACECOLOR.rise = '#F27663'
+    config.VOLUME.FACECOLOR.fall = '#70B5F2'
+    config.VOLUME.FACECOLOR.doji = '#BEBEBE'
+
+    config.VOLUME.EDGECOLOR.rise = '#F27663'
+    config.VOLUME.EDGECOLOR.fall = '#70B5F2'
+    config.VOLUME.EDGECOLOR.doji = '#BEBEBE'
+
+    # ma
+    config.MA.color_default = 'k'
+    config.MA.color_list = ['#8B00FF', '#008000', '#A0522D', '#008B8B', '#FF0080']
+    # text
+    config.CURSOR.TEXT.BBOX.facecolor = 'w'
+    config.CURSOR.TEXT.BBOX.edgecolor = 'k'
+    config.CURSOR.TEXT.color = 'k'
+
+    # box
+    config.CURSOR.BOX.edgecolor = 'k'
+
+    # line
+    config.CURSOR.CROSSLINE.edgecolor = 'k'
+
+    # wartermark
+    config.FIGURE.WATERMARK.color = 'k'
+
+    if getattr(config, 'SLIDER', None):
+        config.SLIDER.NAV.edgecolor = '#2962FF'
+        config.SLIDER.NAV.facecolor = '#0000002E'
+
+    return config
+

seolpyo_mplchart/_utils/utils.py

@@ -0,0 +1,28 @@
+import matplotlib.pyplot as plt
+import pandas as pd
+
+
+def switch_backend(newbackend='TkAgg'):
+    "call matplotlib.pyplot.switch_backend(newbackend)"
+    return plt.switch_backend(newbackend)
+
+
+def close(fig='all'):
+    "call matplotlib.pyplot.close(fig)"
+    return plt.close(fig)
+
+def show(Close=True):
+    """
+    call matplotlib.pyplot.show()
+    ```if Close``` if True, run matplotlib.pyplot.close('all') after window closee.
+    """
+    plt.show()
+    if Close:
+        close()
+    return
+
+
+def list_to_DataFrame(item_list):
+    "return pd.DataFrame(item_list)"
+    return pd.DataFrame(item_list)
+

seolpyo_mplchart/_utils/xl/__init__.py

@@ -0,0 +1,15 @@
+from . import xlsx, csv
+
+
+def xl_to_dataList(path_file):
+    pf = str(path_file)
+    if pf.endswith('.xlsx'):
+        data = xlsx.convert(path_file)
+    elif pf.endswith('.csv'):
+        data = csv.convert(path_file)
+    else:
+        msg = '다음 형식의 파일의 데이터만 변환할 수 있습니다.'
+        msg += f'  [.xlsx, .csv]'
+        raise ValueError(msg)
+    return data
+

seolpyo_mplchart/_utils/xl/csv.py

@@ -0,0 +1,46 @@
+from datetime import datetime
+import json
+
+
+base_date = datetime(1899, 12, 30).date()
+
+
+def convert(path_file: str, row=1, index_date=0, index_open=1, index_high=2, index_low=3, index_close=4, index_volume=5):
+    "csv 파일에서 주가 정보를 추출합니다."
+    item_list: list[dict[str, str|float]] = []
+
+    with open(path_file, 'r', encoding='utf-8') as txt:
+        j = json.load(txt)
+
+    if len(j[1]) < 6:
+        index_volume = None
+    for i in j[row:]:
+        if index_volume:
+            dt = i[index_date]
+            o = i[index_open]
+            h = i[index_high]
+            l = i[index_low]
+            c = i[index_close]
+            v = None
+        else:
+            dt = i[index_date]
+            o = i[index_open]
+            h = i[index_high]
+            l = i[index_low]
+            c = i[index_close]
+            v = i[index_volume]
+
+        o = float(o)
+        h = float(h)
+        l = float(l)
+        c = float(c)
+        if v:
+            v = float(v)
+            item = {'기준일': dt, '시가': o, '고가': h, '저가': l, '종가': c, '거래량': v}
+        else:
+            item = {'기준일': dt, '시가': o, '고가': h, '저가': l, '종가': c}
+
+        item_list.append(item)
+
+    return sorted(item_list, key=lambda x: x['기준일'])
+

seolpyo_mplchart/_utils/xl/xlsx.py

@@ -0,0 +1,49 @@
+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 파일에서 주가 정보를 추출합니다."
+    list_price: list[dict[str, str|float]] = []
+
+    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')
+    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({'기준일': str(dt), '종가': c, '시가': o, '고가': h, '저가': l, '거래량': v,})
+    # for i in enumerate(list_price, 1): print(f'  {i}')
+
+    return sorted(list_price, key=lambda x: x['기준일'])
+