node_correlation_index/main.py at main · orbmis/node_correlation_index · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
# main.py
import argparse
import json
from modules.transform import transform_data
from modules.collate import collate_data
from modules.miga_labs_correlations import analyze_data
from modules.node_operators_HHI import analyze_node_operators_HHI
from modules.staking_pools_HHI import analyze_staking_pools_HHI
from modules.operators_vs_clients import collate_operators_vs_clients
from modules.operators_vs_clients import calculate_average_client_percentage_by_decile
from modules.operators_vs_pools import collate_operators_vs_pools
from modules.operators_vs_relayers import collate_operators_vs_relays
from modules.operators_vs_relayers import calculate_average_relayer_percentage_by_decile

data_folder = "data/"


def transform():
    input_filename = f"{data_folder}/results.json"
    output_filename = f"{data_folder}/data.json"
    transform_data(input_filename, output_filename)


def collate():
    input_filename = f"{data_folder}/data.json"
    output_filename = f"{data_folder}/collated.json"
    collate_data(input_filename, output_filename)


def analyze_nodes():
    _, _, _, input_filename = get_filenames(data_folder, "data")
    analyze_data(input_filename)


def analyze_node_operators():
    input_filename, _, _, _ = get_filenames(data_folder, "collated")
    analyze_node_operators_HHI(input_filename)


def analyze_staking_pools():
    input_filename, _, _, _ = get_filenames(data_folder, "collated")
    analyze_staking_pools_HHI(input_filename)


def operator_vs_clients():
    _, _, json_output_file, csv_output_file = get_filenames(
        data_folder, "operators_vs_clients"
    )
    input_file = f"{data_folder}/collated.json"

    collate_operators_vs_clients(input_file, json_output_file, csv_output_file)

    with open("data/operators_vs_clients.json", "r") as json_file:
        json_data = json.load(json_file)
        calculate_average_client_percentage_by_decile(json_data)


def operator_vs_pools():
    input_file, _, json_output_file, csv_output_file = get_filenames(
        data_folder, "operators_vs_pools"
    )
    input_file = f"{data_folder}/collated.json"

    collate_operators_vs_pools(input_file, json_output_file, csv_output_file)


def operator_vs_relays():
    input_file, _, json_output_file, csv_output_file = get_filenames(
        data_folder, "operators_vs_relays"
    )
    input_file = f"{data_folder}/collated.json"

    collate_operators_vs_relays(input_file, json_output_file, csv_output_file)

    with open("data/operators_vs_relays.json", "r") as json_file:
        json_data = json.load(json_file)
        calculate_average_relayer_percentage_by_decile(json_data)


def get_filenames(data_folder, file_prefix):
    input_filename = f"{data_folder}{file_prefix}.json"
    output_filename = f"{data_folder}{file_prefix}.json"
    json_output_file = f"{data_folder}{file_prefix}.json"
    csv_output_file = f"{data_folder}{file_prefix}.csv"

    return input_filename, output_filename, json_output_file, csv_output_file


def main():
    parser = argparse.ArgumentParser(description="Your application description.")
    parser.add_argument(
        "command",
        choices=[
            "transform",
            "collate",
            "analyze-nodes",
            "analyze-node-operators",
            "analyze-staking-pools",
            "operators-vs-clients",
            "operators-vs-pools",
            "operators-vs-relays",
            "run-analysis",
        ],
        help="Choose which script to run.",
    )

    args = parser.parse_args()
    data_folder = "data/"

    commands = {
        "transform": transform,
        "collate": collate,
        "analyze-node-operators": analyze_node_operators,
        "analyze-staking-pools": analyze_staking_pools,
        "operators-vs-clients": operator_vs_clients,
        "operators-vs-pools": operator_vs_pools,
        "operators-vs-relays": operator_vs_relays,
        "analyze-nodes": analyze_nodes,
    }

    command_list = list(commands.values())

    selected_command = commands.get(args.command)

    if args.command == "run-analysis":
        for method in command_list:
            method()
            print("-" * 80)
    elif selected_command:
        selected_command()
    else:
        print("Invalid command. Please choose a valid command.")


if __name__ == "__main__":
    main()