lol_time_tracker_lambda.py

import json
from datetime import datetime, timedelta
import requests, bs4, re, time
import plotly.graph_objects as go
import traceback

API_KEY = "RGAPI-c16c2668-0913-4123-9416-113f700d30f0"
HEADER = {
    # "Origin": "https://developer.riotgames.com",
    # "Accept-Charset": "application/x-www-form-urlencoded; charset=UTF-8",
    "X-Riot-Token": API_KEY,
    # "Accept-Language": "en-us",
    # "User-Agent": ""
}


def get_account_id(user_name):
    response = requests.get("https://na1.api.riotgames.com/lol/summoner/v4/summoners/by-name/" + user_name,
                            headers=HEADER)
    if response.status_code != 200:
        raise Exception("Cannot get accountID via API" + str(response.status_code) + json.dumps(response.json()))
    return response.json()['accountId']


def get_match_ids(beginTime_stamp, accountId):
    beginTime = str("{:.3f}".format(beginTime_stamp)).replace(".", "")
    response = requests.get(
        "https://na1.api.riotgames.com/lol/match/v4/matchlists/by-account/" + accountId,
        params={
            "beginTime": beginTime,
        },
        headers=HEADER)
    if response.status_code == 404:
        return []
    if response.status_code != 200:
        raise Exception("API call did not go through " + json.dumps(response.json()))
    match_list = response.json()["matches"]
    match_ids = [match['gameId'] for match in match_list]
    return match_ids


def get_matches_data(match_ids):
    resp = dict()
    matches_data = {}
    for match_id in match_ids:
        response = requests.get("https://na1.api.riotgames.com/lol/match/v4/matches/" + str(match_id),
                                headers=HEADER)
        while response.status_code == 429:
            print('rate limited')
            time.sleep(0.05)
            response = requests.get("https://na1.api.riotgames.com/lol/match/v4/matches/" + str(match_id),
                                    headers=HEADER)
        resp[match_id] = response

    matches_data = {match_id: match.json() for match_id, match in resp.items()}

    return matches_data


def get_time_dict(matches_data, beginTime_stamp):
    time_data = {}

    cur_time = datetime.fromtimestamp(beginTime_stamp)
    today_pt = now

    while cur_time <= now + timedelta(hours=1):
        cur_pt = cur_time.strftime("%Y/%m/%d")
        time_data[cur_pt] = 0
        cur_time += timedelta(days=1)

    for match_id, match in matches_data.items():
        cur_pt = datetime.fromtimestamp(match['gameCreation'] / 1000).strftime("%Y/%m/%d")
        time_data[cur_pt] += match['gameDuration'] / 3600

    for time, duration in time_data.items():
        time_data[time] = str("{:.2f}".format(duration))

    return time_data


def transform_data(time_data):
    total_time_x = []
    total_time_list = []

    for date, total_time in sorted(time_data.items()):
        total_time_x.append(date)
        total_time_list.append(total_time)
    return total_time_x, total_time_list


def render_data(total_time_x, total_time_list, userName, days):
    fig = go.Figure(data=go.Bar(x=total_time_x, y=total_time_list, text=total_time_list,
                                textposition='auto', ))
    fig.update_layout(
        title='Amount of Hours "' + userName + '" Spends on League of Legends in the past ' + str(days) + ' days',
        xaxis_tickfont_size=14,
        yaxis=dict(
            title='Hours',
            titlefont_size=16,
            tickfont_size=14,
        ),
    )
    return fig.to_html()


def lambda_handler(event, context):
    try:
        global user_name, days, beginTime_stamp
        user_name = event['queryStringParameters']['user_name']
        days = int(event['queryStringParameters']['days'])
        beginTime_stamp = datetime.timestamp(now - timedelta(days=days))
        print(beginTime_stamp)
    except Exception as e:
        return {
            'statusCode': 400,
            'headers': {
                'Content-Type': 'text/html',
            },
            'body': "error in initializing"
        }
    try:
        accountId = get_account_id(user_name)
        match_ids = get_match_ids(beginTime_stamp, accountId)
        matches_data = get_matches_data(match_ids)
        print("here are match data " + json.dumps(matches_data))
        time_data = get_time_dict(matches_data, beginTime_stamp)
        print("here are the time_data" + json.dumps(time_data))
        total_time_x, total_time_list = transform_data(time_data)
        html = render_data(total_time_x, total_time_list, user_name, days) or "html broke"
        return {
            'statusCode': 200,
            'headers': {
                'Content-Type': 'text/html',
            },
            'body': html
        }
    except Exception as e:
        return {
            'statusCode': 400,
            'headers': {
                'Content-Type': 'text/html',
            },
            'body': str(traceback.format_exc())
        }


try:
    user_name = 'nostest'
    days = 0

    now = datetime.now()
    beginTime_stamp = datetime.timestamp(now - timedelta(days=days))
    print(beginTime_stamp)
except Exception as e:
    print("error outside of main handler")