#!/usr/bin/env python
# coding: utf-8

# In[1]:


get_ipython().run_line_magic('load_ext', 'nb_black')
import pandas as pd
import numpy as np
import altair as alt


# In[2]:


count_df = pd.read_excel("./output/seasonal_popularity.xlsx", index_col=0).fillna("")
count_df["Year"] = count_df["season_readable"].apply(lambda x: x.split()[-1])
count_df


# In[3]:


popularity_chart = (
    alt.Chart(count_df)
    .mark_bar()
    .transform_calculate(combined_comment="datum.comment + ', ' + datum.comment_2")
    .encode(
        x=alt.X("season_readable", title="Season", sort=list(count_df.season_readable)),
        y=alt.Y("pro_players_per_day", title="Popularity (# Pro Player/Season Length)"),
        color="Year",
        tooltip=[
            alt.Tooltip("pro_players_per_day", title="Popularity", format=".2f"),
            alt.Tooltip("length", title="Season Length"),
            alt.Tooltip("combined_comment:N", title="Info"),
        ],
    )
)
popularity_chart.properties(width="container").save("./output/popularity_chart.json")
popularity_chart


# In[4]:


player_chart = (
    alt.Chart(count_df)
    .mark_bar()
    .transform_calculate(combined_comment="datum.comment + ', ' + datum.comment_2")
    .encode(
        x=alt.X("season_readable", title="Season", sort=list(count_df.season_readable)),
        y=alt.Y("players_total", title="Pro Ranked Players"),
        color="Year",
        tooltip=[
            alt.Tooltip("players_total", title="# Pro Players"),
            alt.Tooltip("length", title="Season Length"),
            alt.Tooltip("combined_comment:N", title="Info"),
        ],
    )
)
player_chart.properties(width="container").save("./output/pro_player_chart.json")
player_chart