Mobile proxies for Telegram

import csv
import time
from urllib.parse import urlencode

import requests

SOURCE_FILE = "mobihub_proxies.csv"
TELEGRAM_API = "https://api.telegram.org"
LINKS_FILE = "telegram_links.txt"
SLOW_MS = 1500


def check(row):
    proxy = f"http://{row['login']}:{row['password']}@{row['host']}:{row['port']}"
    started = time.time()
    try:
        requests.get(TELEGRAM_API, proxies={"http": proxy, "https": proxy}, timeout=15)
    except requests.RequestException as error:
        return None, type(error).__name__
    return round((time.time() - started) * 1000), "ok"


def socks_link(row):
    query = urlencode(
        {"server": row["host"], "port": row["port"], "user": row["login"], "pass": row["password"]}
    )
    return f"https://t.me/socks?{query}"


def main():
    with open(SOURCE_FILE, newline="") as f:
        rows = list(csv.DictReader(f))

    links = []
    for row in rows:
        latency, state = check(row)
        if latency is None:
            print(f"{row['name']:<12} DOWN   {state}")
            continue
        note = "slow, pick a closer server in the proxy settings" if latency > SLOW_MS else ""
        print(f"{row['name']:<12} {latency:>5} ms  {note}")
        links.append(f"{row['name']}\t{socks_link(row)}")

    with open(LINKS_FILE, "w") as f:
        f.write("\n".join(links) + "\n")
    print(f"{len(links)} of {len(rows)} links saved to {LINKS_FILE}")


if __name__ == "__main__":
    main()