查域名有效期

import whois
import datetime

def check_domain_expiration(domain):
    try:
        w = whois.whois(domain)
        expiration_date = w.expiration_date
        if isinstance(expiration_date, list):
            expiration_date = expiration_date[0]
        current_date = datetime.datetime.now()
        days_remaining = (expiration_date - current_date).days
        return days_remaining
    except Exception as e:
        return str(e)

domain = "example.com"
days_remaining = check_domain_expiration(domain)
print(f"Domain {domain} will expire in {days_remaining} days.")