# Bokeh libraries
from bokeh.io import output_file
from bokeh.models import ColumnDataSource
from bokeh.plotting import figure, show

# Output to file
output_file(
    "west-top-2-standings-race.html",
    title="Western Conference Top 2 Teams Wins Race",
)

# Isolate the data for the Rockets and Warriors
rockets_data = west_top_2[west_top_2["teamAbbr"] == "HOU"]  # noqa
warriors_data = west_top_2[west_top_2["teamAbbr"] == "GS"]  # noqa

# Create a ColumnDataSource object for each team
rockets_cds = ColumnDataSource(rockets_data)
warriors_cds = ColumnDataSource(warriors_data)

# Create and configure the figure
fig = figure(
    x_axis_type="datetime",
    plot_height=300,
    plot_width=600,
    title="Western Conference Top 2 Teams Wins Race, 2017-18",
    x_axis_label="Date",
    y_axis_label="Wins",
    toolbar_location=None,
)

# Render the race as step lines
fig.step(
    "stDate", "gameWon", color="#CE1141", legend="Rockets", source=rockets_cds
)
fig.step(
    "stDate",
    "gameWon",
    color="#006BB6",
    legend="Warriors",
    source=warriors_cds,
)

# Move the legend to the upper left corner
fig.legend.location = "top_left"

# Show the plot
show(fig)