lwzm
5/13/2016 - 5:40 AM

domain_name.py

import functools

top_level_domains = {
    "com", "co", "net", "org",
}


@functools.lru_cache(10000)
def get_2_3_level(domain):
    fs = domain.split(".")
    j= ".".join
    if fs[-1] not in top_level_domains and fs[-2] in top_level_domains:
        m, n = -3, -4
    else:
        m, n = -2, -3
    return j(fs[m:]), j(fs[n:])