verify-clickhouse-custom-tld.py

import publicsuffix2
import clickhouse_driver

ch = clickhouse_driver.Client(host='127.1')
pls = publicsuffix2.PublicSuffixList(idna=False)

def check_tld(tld):
    print(tld)
    host = 'foo.' + tld
    ch_tld = ch.execute("SELECT cutToFirstSignificantSubdomainCustom(%(host)s, 'public_suffix_list')", params={
        'host': host,
    })[0][0].strip()
    py_tld = pls.get_public_suffix(host)
    assert ch_tld == py_tld, f'ch_tld={ch_tld}, py_tld={py_tld}, host={host}'

tlds = pls.tlds
any(map(check_tld, tlds))