https://github.com/uname-yang/pysnowball/tree/master
import pysnowball as ball
import pandas as pd
import mplfinance as mpf
import matplotlib.pyplot as plt
from datetime import datetime
ball.set_token("xq_a_token=662745a236*****;u=909119****")
symbol = 'SZ1xxxx'
offset = 1
specified_date = '2024-01-01'
specified = datetime.strptime(specified_date, '%Y-%m-%d')
current_date = datetime.now()
days_difference = (current_date - specified).days
# print(current_date,specified, days_difference)
json_data = ball.kline(symbol, days_difference) # K线数据
data = json_data["data"]
columns = data["column"]
items = data["item"]
# 转换为 DataFrame
df = pd.DataFrame(items, columns=columns)
df['Date'] = pd.to_datetime(df['timestamp'] / 1000, unit='s')
df['Date'] = df['Date'] + pd.Timedelta(days=offset)
df = df[df['Date'] >= specified]
# 选择绘制K线图所需的列
ohlc_columns = ['Date', 'open', 'high', 'low', 'close', 'volume']
df_ohlc = df[ohlc_columns]
df_ohlc.to_excel(symbol+".xlsx", index=False, engine="openpyxl")