vpn_servers.py

import json
import requests

response = requests.get('https://api.protonmail.ch/vpn/logicals')
servers_dict = json.loads(response.text)


# Get rid of the LogicalServers wrapper
servers_list = servers_dict['LogicalServers']

# Compile all server names into single list
server_names = []
for x in range(len(servers_list)):
    server_names.append(servers_list[x]['Name'])

# Compile all server load values into single list
server_load = []
for x in range(len(servers_list)):
    server_load.append(servers_list[x]['Load'])

# Compile all server tier values into single list
server_tier = []
for x in range(len(servers_list)):
    server_tier.append(servers_list[x]['Tier'])

    
def find_a_server_in(location):
    draw = 0
    # Look for servers in given location
    index = []
    for i,x in enumerate(server_names):
        if location in x and "-" not in x:
            index.append(i)

    # Look for tier 1 and 2 servers in given location
    temp = []
    for i in index:
        if server_tier[i] == 1:
            temp.append(i)
        if server_tier[i] == 2:
            temp.append(i)

    index = temp

    # Compile list of server load values
    # for tier 1 and 2 servers in given location
    load_list = []
    for y in index:
        load_list.append(server_load[y])


    # Get index of lowest load
    index_lowest = [i for i, j in enumerate(load_list) if j == min(load_list)]

    if len(index_lowest) == 1:
        index_lowest = index_lowest[0]
    elif len(index_lowest) > 1:
        draw = 1
        index_lowest_2 = index_lowest[1]
        index_lowest = index_lowest[0]
        
    # Get index of server with lowest load
    #print(index[index_lowest])
    
    # Get the name of the server with lowest load
    ideal_server = server_names[index[index_lowest]]
    lowest_load = load_list[index_lowest]

    print(ideal_server + ": "+ str(lowest_load) + "%")
    
    # If there's a draw, print out the other contending server
    if draw == 1:
        ideal_server_2 = server_names[index[index_lowest_2]]
        lowest_load_2 = load_list[index_lowest_2]
        print(ideal_server_2 + ": "+ str(lowest_load_2) + "%")
#