domaincheck.py

#!/usr/bin/env python

from __future__ import print_function

import sys

try:
    import whois
    # python-whois from pypi, not whois from pypi or python-whois from debian
    # https://bitbucket.org/richardpenman/pywhois
except:
    print("Couldn't import whois. Try: pip3 install python-whois")
    sys.exit(1)

import socket     # for the socket.timeout exception

import datetime
from dateutil.relativedelta import relativedelta

format="%25s   %10s %3s %s"
RETRIES = 1

def get_domain(domainname):
    for i in range(RETRIES):
        try:
            domain = whois.whois(name)
            return domain
        except socket.timeout:
            print("%s timed out; retrying" % domainname)
    print("Giving up on %s after %d timeouts" % (domainname, RETRIES))
    return None

if __name__ == '__main__':
    domainlist = []
    for name in sys.argv[1:]:
        domain = get_domain(name)
        if not domain:
            print("Can't get info for %s" % name)
            continue
        if not domain["expiration_date"]:
            print("Can't get expiration date for %s" % name)
            continue
        elif hasattr(domain["expiration_date"], "__len__"):
            # Sometimes python-whois returns a list of dates,
            # for unknown reasons.
            # Check whether they're all the same.
            expdate = domain["expiration_date"][0].date()
            for e in domain["expiration_date"][1:]:
                if e.date() != expdate:
                    print("Yikes, %s != %s" % (str(e), str(expdate)))
        else:
            expdate = domain["expiration_date"].date()

        if domain:
            domainlist.append((name, expdate, domain.registrar))

    domainlist.sort(key = lambda a: a[1])

    two_months_from_now = datetime.datetime.today() + relativedelta(months=2)
    two_months_from_now = two_months_from_now.date()
    print(format % ("Domain", "Expires", "", "Registrar"))
    for d in domainlist:
        if d[1] < two_months_from_now:
            alert = "***"
        else:
            alert = ""
        print(format % (d[0], d[1].strftime('%Y-%m-%d'), alert, d[2]))