from jqdatasdk import *
import numpy as np
import pandas as pd
from bokeh.models import ColumnDataSource, HoverTool
from bokeh.plotting import figure, output_file, show


auth('17799730122', 'F117xxxd')  # ID是申请时所填写的手机号；Password为聚宽官网登录密码

count = get_query_count()
print(count)
# 'pre', 'post', 'none'
data = get_price(security='000001.XSHE', start_date='2022-07-01', end_date='2022-09-08', frequency='daily',
                  fields=None, skip_paused=False, fq='pre',
                  count=None, panel=True, fill_paused=True)

print(data)

# load data
# data = load_data(order_book_id.value, start_date.value, end_date.value)  # 根据默认值提取数据
source = ColumnDataSource(data)
print(source)
inc = data['close'] >= data['open']
dec = data['open'] > data['close']

inc_source = ColumnDataSource(data.loc[inc])
dec_source = ColumnDataSource(data.loc[dec])

p = figure(plot_height=400, plot_width=800)
hover = HoverTool(tooltips=[('date', '@date'),
                            ('open', '@open'),
                            ('high', '@high'),
                            ('low', '@low'),
                            ('close', '@close')])

p.segment(x0='index', y0='high', x1='index', y1='low', color='red', source=inc_source)
p.segment(x0='index', y0='high', x1='index', y1='low', color='green', source=dec_source)
p.vbar('index', 1, 'open', 'close', fill_color='red', line_color='red', source=inc_source, hover_fill_alpha=0.5)
p.vbar('index',1, 'open', 'close', fill_color='green', line_color='green', source=dec_source, hover_fill_alpha=0.5)

# add hover tool
p.add_tools(hover)

show(p)