Parser script for Tor Snowflake Metrics API

metrics.py

import re
import requests

# Replace with your actual metrics endpoint
url = "http://ubuntu:1080/internal/metrics"
url2 = "http://192.168.1.2:1080/internal/metrics"

# Download metrics text
resp = None
try:
    resp = requests.get(url)
except:
    resp = requests.get(url2)
metrics = resp.text

# Parse relevant metrics
timeouts = 0
total_connections = 0
connections_by_country = {}
inbound = 0
outbound = 0

for line in metrics.splitlines():
    if line.startswith("tor_snowflake_proxy_connection_timeouts_total"):
        timeouts = int(float(line.split()[-1]))
    elif line.startswith("tor_snowflake_proxy_connections_total"):
        match = re.search(r'country="([A-Z]{2})"', line)
        country = match.group(1) if match else "?"
        count = int(float(line.split()[-1]))
        connections_by_country[country] = connections_by_country.get(country, 0) + count
        total_connections += count
    elif line.startswith("tor_snowflake_proxy_traffic_inbound_bytes_total"):
        inbound = int(float(line.split()[-1]))
    elif line.startswith("tor_snowflake_proxy_traffic_outbound_bytes_total"):
        outbound = int(float(line.split()[-1]))

# Manual ISO country name + emoji flag
country_map = {
    "?": "Unknown [?]",
    "AW": "Aruba 🇦🇼",
    "AF": "Afghanistan 🇦🇫",
    "AO": "Angola 🇦🇴",
    "AI": "Anguilla 🇦🇮",
    "AX": "Åland Islands 🇦🇽",
    "AL": "Albania 🇦🇱",
    "AD": "Andorra 🇦🇩",
    "AE": "United Arab Emirates 🇦🇪",
    "AR": "Argentina 🇦🇷",
    "AM": "Armenia 🇦🇲",
    "AS": "American Samoa 🇦🇸",
    "AQ": "Antarctica 🇦🇶",
    "TF": "French Southern Territories 🇹🇫",
    "AG": "Antigua and Barbuda 🇦🇬",
    "AU": "Australia 🇦🇺",
    "AT": "Austria 🇦🇹",
    "AZ": "Azerbaijan 🇦🇿",
    "BI": "Burundi 🇧🇮",
    "BE": "Belgium 🇧🇪",
    "BJ": "Benin 🇧🇯",
    "BF": "Burkina Faso 🇧🇫",
    "BD": "Bangladesh 🇧🇩",
    "BG": "Bulgaria 🇧🇬",
    "BH": "Bahrain 🇧🇭",
    "BS": "Bahamas 🇧🇸",
    "BA": "Bosnia and Herzegovina 🇧🇦",
    "BY": "Belarus 🇧🇾",
    "BZ": "Belize 🇧🇿",
    "BM": "Bermuda 🇧🇲",
    "BO": "Bolivia 🇧🇴",
    "BR": "Brazil 🇧🇷",
    "BB": "Barbados 🇧🇧",
    "BN": "Brunei Darussalam 🇧🇳",
    "BT": "Bhutan 🇧🇹",
    "BV": "Bouvet Island 🇧🇻",
    "BW": "Botswana 🇧🇼",
    "CF": "Central African Republic 🇨🇫",
    "CA": "Canada 🇨🇦",
    "CC": "Cocos (Keeling) Islands 🇨🇨",
    "CH": "Switzerland 🇨🇭",
    "CL": "Chile 🇨🇱",
    "CN": "China 🇨🇳",
    "CI": "Côte d'Ivoire 🇨🇮",
    "CM": "Cameroon 🇨🇲",
    "CD": "Congo, The Democratic Republic of the 🇨🇩",
    "CG": "Congo 🇨🇬",
    "CK": "Cook Islands 🇨🇰",
    "CO": "Colombia 🇨🇴",
    "KM": "Comoros 🇰🇲",
    "CV": "Cabo Verde 🇨🇻",
    "CR": "Costa Rica 🇨🇷",
    "CU": "Cuba 🇨🇺",
    "CW": "Curaçao 🇨🇼",
    "CX": "Christmas Island 🇨🇽",
    "CY": "Cyprus 🇨🇾",
    "CZ": "Czechia 🇨🇿",
    "DE": "Germany 🇩🇪",
    "DJ": "Djibouti 🇩🇯",
    "DM": "Dominica 🇩🇲",
    "DK": "Denmark 🇩🇰",
    "DO": "Dominican Republic 🇩🇴",
    "DZ": "Algeria 🇩🇿",
    "EC": "Ecuador 🇪🇨",
    "EG": "Egypt 🇪🇬",
    "ER": "Eritrea 🇪🇷",
    "EH": "Western Sahara 🇪🇭",
    "ES": "Spain 🇪🇸",
    "EE": "Estonia 🇪🇪",
    "ET": "Ethiopia 🇪🇹",
    "FI": "Finland 🇫🇮",
    "FJ": "Fiji 🇫🇯",
    "FK": "Falkland Islands 🇫🇰",
    "FM": "Micronesia 🇫🇲",
    "FO": "Faroe Islands 🇫🇴",
    "FR": "France 🇫🇷",
    "GA": "Gabon 🇬🇦",
    "GB": "United Kingdom 🇬🇧",
    "GE": "Georgia 🇬🇪",
    "GG": "Guernsey 🇬🇬",
    "GH": "Ghana 🇬🇭",
    "GI": "Gibraltar 🇬🇮",
    "GN": "Guinea 🇬🇳",
    "GP": "Guadeloupe 🇬🇵",
    "GM": "Gambia 🇬🇲",
    "GW": "Guinea-Bissau 🇬🇼",
    "GQ": "Equatorial Guinea 🇬🇶",
    "GR": "Greece 🇬🇷",
    "GD": "Grenada 🇬🇩",
    "GL": "Greenland 🇬🇱",
    "GT": "Guatemala 🇬🇹",
    "GF": "French Guiana 🇬🇫",
    "GU": "Guam 🇬🇺",
    "GY": "Guyana 🇬🇾",
    "HK": "Hong Kong 🇭🇰",
    "HM": "Heard Island and McDonald Islands 🇭🇲",
    "HN": "Honduras 🇭🇳",
    "HR": "Croatia 🇭🇷",
    "HT": "Haiti 🇭🇹",
    "HU": "Hungary 🇭🇺",
    "ID": "Indonesia 🇮🇩",
    "IM": "Isle of Man 🇮🇲",
    "IN": "India 🇮🇳",
    "IO": "British Indian Ocean Territory 🇮🇴",
    "IE": "Ireland 🇮🇪",
    "IR": "Iran 🇮🇷",
    "IQ": "Iraq 🇮🇶",
    "IS": "Iceland 🇮🇸",
    "IL": "Israel 🇮🇱",
    "IT": "Italy 🇮🇹",
    "JM": "Jamaica 🇯🇲",
    "JE": "Jersey 🇯🇪",
    "JO": "Jordan 🇯🇴",
    "JP": "Japan 🇯🇵",
    "KZ": "Kazakhstan 🇰🇿",
    "KE": "Kenya 🇰🇪",
    "KG": "Kyrgyzstan 🇰🇬",
    "KH": "Cambodia 🇰🇭",
    "KI": "Kiribati 🇰🇮",
    "KN": "Saint Kitts and Nevis 🇰🇳",
    "KR": "Korea, Republic of 🇰🇷",
    "KW": "Kuwait 🇰🇼",
    "LA": "Lao People's Democratic Republic 🇱🇦",
    "LB": "Lebanon 🇱🇧",
    "LR": "Liberia 🇱🇷",
    "LY": "Libya 🇱🇾",
    "LC": "Saint Lucia 🇱🇨",
    "LI": "Liechtenstein 🇱🇮",
    "LK": "Sri Lanka 🇱🇰",
    "LS": "Lesotho 🇱🇸",
    "LT": "Lithuania 🇱🇹",
    "LU": "Luxembourg 🇱🇺",
    "LV": "Latvia 🇱🇻",
    "MO": "Macao 🇲🇴",
    "MF": "Saint Martin 🇲🇫",
    "MA": "Morocco 🇲🇦",
    "MC": "Monaco 🇲🇨",
    "MD": "Moldova 🇲🇩",
    "MG": "Madagascar 🇲🇬",
    "MV": "Maldives 🇲🇻",
    "MX": "Mexico 🇲🇽",
    "MH": "Marshall Islands 🇲🇭",
    "MK": "North Macedonia 🇲🇰",
    "ML": "Mali 🇲🇱",
    "MT": "Malta 🇲🇹",
    "MM": "Myanmar 🇲🇲",
    "ME": "Montenegro 🇲🇪",
    "MN": "Mongolia 🇲🇳",
    "MP": "Northern Mariana Islands 🇲🇵",
    "MZ": "Mozambique 🇲🇿",
    "MR": "Mauritania 🇲🇷",
    "MS": "Montserrat 🇲🇸",
    "MQ": "Martinique 🇲🇶",
    "MU": "Mauritius 🇲🇺",
    "MW": "Malawi 🇲🇼",
    "MY": "Malaysia 🇲🇾",
    "YT": "Mayotte 🇾🇹",
    "NA": "Namibia 🇳🇦",
    "NC": "New Caledonia 🇳🇨",
    "NE": "Niger 🇳🇪",
    "NF": "Norfolk Island 🇳🇫",
    "NG": "Nigeria 🇳🇬",
    "NI": "Nicaragua 🇳🇮",
    "NL": "Netherlands 🇳🇱",
    "NO": "Norway 🇳🇴",
    "NP": "Nepal 🇳🇵",
    "NR": "Nauru 🇳🇷",
    "NU": "Niue 🇳🇺",
    "NZ": "New Zealand 🇳🇿",
    "OM": "Oman 🇴🇲",
    "PK": "Pakistan 🇵🇰",
    "PA": "Panama 🇵🇦",
    "PN": "Pitcairn 🇵🇳",
    "PE": "Peru 🇵🇪",
    "PH": "Philippines 🇵🇭",
    "PW": "Palau 🇵🇼",
    "PG": "Papua New Guinea 🇵🇬",
    "PL": "Poland 🇵🇱",
    "PR": "Puerto Rico 🇵🇷",
    "KP": "Korea, Democratic People's Republic of 🇰🇵",
    "PT": "Portugal 🇵🇹",
    "PY": "Paraguay 🇵🇾",
    "PS": "Palestine, State of 🇵🇸",
    "QA": "Qatar 🇶🇦",
    "RE": "Réunion 🇷🇪",
    "RO": "Romania 🇷🇴",
    "RU": "Russia 🇷🇺",
    "RW": "Rwanda 🇷🇼",
    "SA": "Saudi Arabia 🇸🇦",
    "SD": "Sudan 🇸🇩",
    "SN": "Senegal 🇸🇳",
    "SG": "Singapore 🇸🇬",
    "GS": "South Georgia and the South Sandwich Islands 🇬🇸",
    "SH": "Saint Helena, Ascension and Tristan da Cunha 🇸🇭",
    "SJ": "Svalbard and Jan Mayen 🇸🇯",
    "SB": "Solomon Islands 🇸🇧",
    "SL": "Sierra Leone 🇸🇱",
    "SV": "El Salvador 🇸🇻",
    "SM": "San Marino 🇸🇲",
    "SO": "Somalia 🇸🇴",
    "PM": "Saint Pierre and Miquelon 🇵🇲",
    "RS": "Serbia 🇷🇸",
    "SS": "South Sudan 🇸🇸",
    "ST": "Sao Tome and Principe 🇸🇹",
    "SR": "Suriname 🇸🇷",
    "SK": "Slovakia 🇸🇰",
    "SI": "Slovenia 🇸🇮",
    "SE": "Sweden 🇸🇪",
    "SZ": "Eswatini 🇸🇿",
    "SX": "Sint Maarten 🇸🇽",
    "SC": "Seychelles 🇸🇨",
    "SY": "Syrian Arab Republic 🇸🇾",
    "TC": "Turks and Caicos Islands 🇹🇨",
    "TD": "Chad 🇹🇩",
    "TG": "Togo 🇹🇬",
    "TH": "Thailand 🇹🇭",
    "TJ": "Tajikistan 🇹🇯",
    "TK": "Tokelau 🇹🇰",
    "TM": "Turkmenistan 🇹🇲",
    "TL": "Timor-Leste 🇹🇱",
    "TO": "Tonga 🇹🇴",
    "TT": "Trinidad and Tobago 🇹🇹",
    "TN": "Tunisia 🇹🇳",
    "TR": "Türkiye 🇹🇷",
    "TW": "Taiwan 🇹🇼",
    "TV": "Tuvalu 🇹🇻",
    "TZ": "Tanzania 🇹🇿",
    "UG": "Uganda 🇺🇬",
    "UA": "Ukraine 🇺🇦",
    "UM": "United States Minor Outlying Islands 🇺🇲",
    "UY": "Uruguay 🇺🇾",
    "US": "United States 🇺🇸",
    "UZ": "Uzbekistan 🇺🇿",
    "VA": "Holy See 🇻🇦",
    "VC": "Saint Vincent and the Grenadines 🇻🇨",
    "VE": "Venezuela 🇻🇪",
    "VG": "Virgin Islands, British 🇻🇬",
    "VI": "Virgin Islands, U.S. 🇻🇮",
    "VN": "Vietnam 🇻🇳",
    "VU": "Vanuatu 🇻🇺",
    "WF": "Wallis and Futuna 🇼🇫",
    "WS": "Samoa 🇼🇸",
    "YE": "Yemen 🇾🇪",
    "ZA": "South Africa 🇿🇦",
    "ZM": "Zambia 🇿🇲",
    "ZW": "Zimbabwe 🇿🇼"
}

# Auto-scaling display for values in KB
def format_bytes_from_kb(kb_val):
    units = ["KB", "MB", "GB", "TB"]
    size = float(kb_val)
    for unit in units:
        if size < 1000:
            return f"{size:.2f} {unit}"
        size /= 1000
    return f"{size:.2f} PB"

# Print summary
print("\n==== Snowflake Proxy Metrics Summary ====\n")
print(f"{'Total Successful Connections':<30}: {total_connections}")
print(f"{'Connection Timeouts':<30}: {timeouts}")
print(f"{'Inbound Traffic':<30}: {format_bytes_from_kb(inbound)}")
print(f"{'Outbound Traffic':<30}: {format_bytes_from_kb(outbound)}")

# Print country-wise table
print("\n=== Connections by Country ===\n")
print(f"{'Country':<27} {'Connections':>12}")
print("-" * 40)
for code, count in sorted(connections_by_country.items(), key=lambda x: x[1], reverse=True):
    name = country_map.get(code, f"{code} [?]")
    print(f"{name:<27} {count:>12}")