stats.py

'''

                    -= Werewolves Data Collection Module =-

                     -= Our Lord and Saviour, The Zucc =-

@@@@@@@@@@@@@@@@@@@@@@@@@@@@%//((/((((((#((##(%&@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@@@@#///((((((////((/**/(((((((/%@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@%/((/(#(((///(/,,,***,*,***///*//@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@#*///**///*/******,......,,,*,****///(@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@(///***,*/(/*,,,,...,,*(*,.,,,,**/(*,*///#@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@#*/////*,,****,...,**//((*,,...,*/*,/*,*/*//%@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@///,,/*,,****,,..,,,.,**,,,**,,,,,,.,,,,,*//((#@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@(/*,,**,,,..****//((#####(/**/(#%#*(###(,.,***/%@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@&,*/,*/*..,****//((((###%%%%&&&&&&&&&&&&%#/**//**#@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@%,,*.//(,,*//**/((((#####%%%&&&&&&&&&&&&&&%/****//@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@#,*,,,,..*/#(#(/(((#####%%%%&&&&&&&&&&&&&&%#//****@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@%*,,,....*/((##((((##%%%%%&&&&&&&&&&&&&&&&&%#/.,,,@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@*,,.....*((%%(/****//(##%%%&&&&&&&&&&&&&&%%%(,**(@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@/*.,...,##%#**/((##(////(%%%%%%#((/((#%%%%%#*..*/#//@@@@@@@@@@@
@@@@@@@@@@@@@@@&***,.,./&%#/*****/((///**(#&&%#((((#%%&%##%#.,,,*//%@@@@@@@@@@@
@@@@@@@@@@@@@@@&,,*/**/#&%/****,*..//,***(%&&(**,,/###%%%(...,,/@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@,,**/**%%%(/(///((((((/**(#%&&&%((*.*(/#%%%(.*/,,(@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@%**,,**#%%#((###(((((/***(#%%&&#(#%&%%&&%(%##,,@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@(,,,(%#####(##%###(//***(#&%&&&&&&&&&&&&&%(/*#@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@/(#*####(((#####((/****/#&&&&&&&&&&&&&&(%%(%@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@(/((((//((###(/*****((#&&%%%%&&&&&&&&##&@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@%//((((///(((/*******/#%&&%%%#(#%&&&&&%%%&@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@&(//**//******,,,,***/(#%%%#*(#%%%%%&&@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@(/***((*********/((#%%%%%&%*/(#%##%&@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@&(/**/((/**  ******/*/((####//#%%#%&@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@&(//**/((//**,.*/#%%%%%%,/%%##%%%%&@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@&((/***/(//***//(((((((#%&&%%%%%%%@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@&(((/***////**////((##%%&&&%%%%#%@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@((((/***/(//////(((##%%%&&%%%##&@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@(((((/////(#(######%%%&&&%%##(&@@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@(((#(//////(##%%%%%&&&&&%##(#%@@@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@#((#((////**/##%%%&&&&%%#((%%%@@@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@#((##(/*******/((#####(((#%%%%&@@@@@@@@@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@@@%((##(/***,,,,,,,,,*/(##%%%%&%#%%(*//(%@@@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@@@@%###((((/********//((##%%%%&&&&%%%%#****(//%@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@@@@##(#((#((((/****///(((##%%%%%%%&&%%%%#********//(#@@@@@@@@@@@
@@@@@@@@@@@@((((((((//#(((//****//(((##%%%%%%%%%%%%%(**,**/****///(((&@@@@@@@
@@@@@@@@@%#(((((((///(//#(////****//((###%%%%%%%%%%%#*,,,**/***////////((((#%@@
@@@@@%#(*/((((((//**////%#///****//(((#####%%%%%%%*,,,,***/*******/////////(((#
@@@#((/////////((((((/**##(/*****///(((((######(,,,,,**/*****/////*******//////
@%((///////*****//(/((/**(((*******///((((((/,,,,,,***************,*/********//
@(/*///************//((**,*//*****/////*,,...,,,,*******,,,,,,,,,,,,,,,**/*****
%/*****/***********,,*//*,,,.   .........,,,,****,,,,,,,,,,,,,,,,,,,,*****,*//*
(,.,*****///******/*********,.......,,,,*,,,,,,,,,,,,,,,,,,,,,,,,,,,****,,,,,**
/, .,********//*********,,,,..,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,**,,,,,,,,,**
/.  .**********/*******,,,,,,,,.....,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,.,**

                  "We do not sell data to advertisers"
                  "Violating your privacy since 2004™"
                             - The Zucc

                           *In Zucc We Trust*
'''

import json
from config import stats_file


def get_stats():
    """Returns the current stats from the file as a JSON object"""
    try:
        with open(stats_file) as f:
            return json.load(f)
    except:
        print("WARNING: Stats file is invalid! Writing a new one.")
        write_stats({"users":{}, "current_day": {}})
def write_stats(stats):
    """Writes the stats to the file"""
    with open(stats_file, 'w') as outfile:
        json.dump(stats, outfile)

def reset_stats(confirm):
    """Overwrites all stats, ever"""
    if confirm:
        write_stats({"users":{}, "current_day": {}})
        return True
    return False

def increment_stat(stat, amount):
    """Increments {stat} by {amount}"""
    stats = get_stats()
    try:
        stats[stat] += amount
    except:
        stats[stat] = amount
    try:
        stats["current_day"][stat] += amount
    except:
        stats["current_day"][stat] = amount
    write_stats(stats)

def get_stat(stat):
    """Returns the value of {stat}"""
    try:
        return get_stats()[stat]
    except:
        return "0"

def get_stat_day(stat):
    """Returns the value of {stat} for the current d/n cycle"""
    try:
        return get_stats()["current_day"][stat]
    except:
        return "0"

def get_user_stat(user_id, stat):
    """Returns {user_id}'s {stat}"""
    try:
        user_id = str(user_id)
        return get_stats()["users"][user_id][stat]
    except:
        return "0"

def increment_user_stat(user_id, stat, value):
    """Returns {user_id}'s {stat}"""
    user_id = str(user_id)
    stats = get_stats()
    if not user_id in stats["users"]:
        stats["users"][user_id] = {} # Initialise user ID
    try:
        stats["users"][user_id][stat] += value
    except:
        stats["users"][user_id][stat] = value
    write_stats(stats)

def next_day():
    stats = get_stats()
    stats["current_day"] = {}
    write_stats(stats)