grapher.py

import requests
import seaborn as sns
import pandas as pd
import os
import time
from lib.mn_count import get_cur_mn_count

nexus_hash = "7877338b947d7fe25d4dcc80979ee1f918736596677188ebadc79e5b9009d847"


def get_votes(p_hash="2f09a8b36477c1f7be2b8e3153b32a67f8438172ae0b0fd6370ea00c6352a762"):
    query = "/governanceproposals/votes?testnet=0&proposalhash={}".format(p_hash)

    dn_base_url = os.environ.get("DN_API_URL")
    url = dn_base_url + query

    response = requests.get(url)
    data = response.json()['data']['governanceproposalvotes']

    for vote in data:
        if vote['VoteValue'] == "NO":
            vote['Math'] = -1
        if vote['VoteValue'] == "YES":
            vote['Math'] = 1

    return data


def graph_votes(vote_data_df):
    # Just have to graph it now

    g = sns.relplot(x="Date", y="Cumulative Votes", kind="line", data=vote_data_df)
    g.fig.autofmt_xdate()
    g.fig.set_size_inches(w=6, h=3)

    try:
        g.savefig(fname="test.png")
        return True
    except Exception as e:
        print(e)
        return False


def convert_to_dataframe(vote_data_dict):
    df = pd.DataFrame.from_records(vote_data_dict)
    df['Date'] = pd.to_datetime(df['VoteTime'], origin='unix', unit='s')
    df.sort_values(by=['VoteTime'], ascending=True, inplace=True)
    df['Cumulative Votes'] = df.Math.cumsum()
    # print(df.head())

    return df


if __name__ == "__main__":
    vote_data = get_votes(p_hash=nexus_hash)
    vote_df = convert_to_dataframe(vote_data)
    graph = graph_votes(vote_df)