封装plot chart

chart.py

@@ -179,25 +179,13 @@ def resample_data(df: pd.DataFrame, timeframe: str) -> pd.DataFrame:
     return resampled
 
 
-if __name__ == "__main__":
-    code = "399986"
-    timeframe = "1W"
-
-    df = pd.read_csv(
-        "/Users/aszer/Documents/vscode/etf/data/index_all_stock.csv",
-        encoding="utf-8-sig",
-    )
-    name = df.loc[df["代码"] == code, "名称"].values[0]
-
-    df = get_kline(code=code)
-    df = resample_data(df, timeframe)
+def plot_chart(symbol: str, name: str, timeframe: str):
     chart = Chart(toolbox=True, inner_height=0.7, maximize=True)
 
-    chart.topbar.textbox("symbol", code)
+    chart.topbar.textbox("symbol", symbol)
     chart.topbar.textbox("name", name)
     chart.topbar.textbox("timeframe", timeframe)
 
-    # chart.time_scale(visible=False) # 将主图的时间轴隐藏
     chart.legend(visible=True)
 
     chart.set(df)
@@ -210,3 +198,18 @@ if __name__ == "__main__":
     add_TD(df, chart)
 
     chart.show(block=True)
+
+
+if __name__ == "__main__":
+    symbol = "931357"
+    timeframe = "1W"
+
+    df = pd.read_csv(
+        "/Users/aszer/Documents/vscode/etf/data/index_all_stock.csv",
+        encoding="utf-8-sig",
+    )
+    name = df.loc[df["代码"] == symbol, "名称"].values[0]
+
+    df = get_kline(code=symbol)
+    df = resample_data(df, timeframe)
+    plot_chart(symbol=symbol, name=name, timeframe=timeframe)