main.py

import os
import numpy as np
import pandas as pd
import requests
from datetime import datetime as dt
from datetime import timedelta

# plotly imports
import plotly.offline as plo
import plotly.graph_objs as go

# dash imports
import dash
import dash_auth
import dash_core_components as dcc
import dash_html_components as html
import dash_table as dtbl

from dash.dependencies import Input, Output, State

# Authentication token
token = os.getenv('STREAMHOSTER')

# Credentials
username_secret = os.getenv('STREAMHOSTER_USER')
password_secret = os.getenv('STREAMHOSTER_PASS')
USERNAME_PASSWORD_PAIRS = [[username_secret,password_secret]]

# Initialize app
app = dash.Dash(__name__, static_folder='static')
auth = dash_auth.BasicAuth(app,USERNAME_PASSWORD_PAIRS)
server = app.server

# Function returns a pandas data frame of Streamhoster data acquired through API 
def __StreamhosterDataFetcher__(start_time, end_time, media_key=None):
    TOKEN = token
    URL = 'https://api.streamhoster.com/reports/service-segments'
    payload = {'startTime': start_time, 'endTime': end_time, 'segmentNames': 'm'}
    r = requests.get(URL, params=payload, headers={'Authorization': 'Basic %s' %  TOKEN})
    data = r.json()
    df = pd.DataFrame({'mediakey': [], 'views': [], 'uniques': [], 'datatransferGB': []})
    
    for x in data['services']:
        if "vnr" in x['serviceName']:
            for i in x['m']:
                df = df.append({'mediakey': i['key'], 'views': i['totals']['views'], 'uniques': i['totals']['uniques'], 'datatransferGB': i['totals']['dataTransferGB']}, ignore_index=True)

    return df.sort_values(by=['views'], ascending=False)

today = dt.today().date()
tenDaysAgo = dt.today().date() - timedelta(days=1)

df = __StreamhosterDataFetcher__(tenDaysAgo.strftime('%Y%m%d'),today.strftime('%Y%m%d'))

data_views = [go.Bar(
            x=df['mediakey'].tolist(),
            y=df['views']
)]

data_uniques = [go.Bar(
            x=df['mediakey'].tolist(),
            y=df['uniques']
)]

app.layout = html.Div([

        html.H1('VNR1 Dashboard'),

        html.Div([
        html.Label('Media Key: '),        
        dcc.Input(id='mediakeyinput',
                value='',
                type='text'),
        ], style={'padding':10, 'display':'inline-block'}
        ),
        
        html.Div([
        html.Label('Date Range: '),
        dcc.DatePickerRange(id='daterangeinput',
                        start_date=tenDaysAgo,
                        end_date=today,
                        max_date_allowed=dt.today()),
        ], style={'padding':10}
        ),

        html.Button(id='submit-button',
            n_clicks=0,
            children='Submit',
            style={'fontSize':18, 
                    'padding':15}
        ),

        html.Div([
        
        dcc.Graph(id='views',
            figure={
                    'data':data_views,
                    'layout':{'height':600, 
                                'width':1000,
                                'title':'Media Views',
                                'yaxis':{'title':'Count of Media Views'},
                                'margin':{'b':200}    
                    }
            }
            , style={'width':'50%'
                            ,'overflowX':'scroll'
                            ,'float':'left'
                    }
                ),

        dcc.Graph(id='uniques',
            figure={'data':data_uniques,
                    'layout':{'height':600, 
                                'width':1000,
                                'title':'Unique Viewers',
                                'yaxis':{'title':'Count of Unique Viewers'},
                                'margin':{'b':200},
                                'images':dict(source='VNR1.jpg',         
                                                xref="paper", yref="paper",
                                                x=1, y=1.05,
                                                sizex=0.2, sizey=0.2,
                                                xanchor="right", yanchor="bottom")
                    }
                    }, style={'width':'50%',
                                'overflowX':'scroll'})
        ]),

        html.Div([
            
            dtbl.DataTable(
                id='table',
                columns=[{"name": i, "id": i} for i in df.columns],
                data=df.to_dict("rows")
            )
            
        ], style={'marginBottom': 50, 'marginTop':50})

])

# callback functions to update graphs when inputs are changed

@app.callback(Output('table','data'),
            [Input('submit-button','n_clicks')],
            [State('mediakeyinput','value'),
                State('daterangeinput','start_date'),
                State('daterangeinput','end_date')])
def update_table_data(n_clicks,media_key,daterangestart,daterangeend):
    # strip date from datetime and convert to string format
    daterangestart_input = dt.strptime(daterangestart,'%Y-%m-%d')
    daterangestart_string = dt.strftime(daterangestart_input,'%Y%m%d')

    daterangeend_input = dt.strptime(daterangeend,'%Y-%m-%d')
    daterangeend_string = dt.strftime(daterangeend_input,'%Y%m%d')    

    # call StreamhosterDataFetcher for updated data
    new_data = __StreamhosterDataFetcher__(daterangestart_string, daterangeend_string)

    # filter to media key wildcard search
    if media_key == '':
        filtered_df = new_data
    else:
        filtered_df = new_data[new_data['mediakey'].str.contains(media_key)]

    return filtered_df.to_dict("rows")


@app.callback(Output('views','figure'),
            [Input('submit-button','n_clicks')],
            [State('mediakeyinput','value'),
                State('daterangeinput','start_date'),
                State('daterangeinput','end_date')])
def update_figure_views(n_clicks,media_key,daterangestart,daterangeend):
    # strip date from datetime and convert to string format
    daterangestart_input = dt.strptime(daterangestart,'%Y-%m-%d')
    daterangestart_string = dt.strftime(daterangestart_input,'%Y%m%d')

    daterangeend_input = dt.strptime(daterangeend,'%Y-%m-%d')
    daterangeend_string = dt.strftime(daterangeend_input,'%Y%m%d')

    # call StreamhosterDataFetcher for updated data
    new_data = __StreamhosterDataFetcher__(daterangestart_string, daterangeend_string)

    # filter to media key wildcard search
    if media_key == '':
        filtered_df = new_data
    else:
        filtered_df = new_data[new_data['mediakey'].str.contains(media_key)]

    traces = []

    traces.append(go.Bar(
            x = filtered_df['mediakey'].tolist(),
            y = filtered_df['views']
    ))

    layout = {'height':600, 
                'width':1000,
                'title':'Media Views',
                'yaxis':{'title':'Count of Media Views'},
                'margin':{'b':200}
            }

    return {'data': traces,'layout':layout}

@app.callback(Output('uniques','figure'),
            [Input('submit-button','n_clicks')],
            [State('mediakeyinput','value'),
                State('daterangeinput','start_date'),
                State('daterangeinput','end_date')])
def update_figure_unique_viewers(n_clicks,media_key,daterangestart,daterangeend):
    daterangestart_input = dt.strptime(daterangestart,'%Y-%m-%d')
    daterangestart_string = dt.strftime(daterangestart_input,'%Y%m%d')

    daterangeend_input = dt.strptime(daterangeend,'%Y-%m-%d')
    daterangeend_string = dt.strftime(daterangeend_input,'%Y%m%d')

    new_data = __StreamhosterDataFetcher__(daterangestart_string, daterangeend_string)

    if media_key == '':
        filtered_df = new_data
    else:
        filtered_df = new_data[new_data['mediakey'].str.contains(media_key)]

    traces = []

    traces.append(go.Bar(
            x = filtered_df['mediakey'].tolist(),
            y = filtered_df['uniques']
    ))

    layout = {'height':600, 
                'width':1000, 
                'yaxis':{'title':'Count of Unique Viewers'},
                'title':'Unique Viewers',
                'margin':{'b':200}
    }

    return {'data': traces, 'layout': layout}

if __name__ == '__main__':
    app.run_server()