（python）域名查询服务（whois）程序+检查5位以内域名到期时间邮件报警抢注域名

前言：这个程序写了挺长时间，主要的加‘-d’选项+域名（只支持.com域名）查询特定域名，不带任何参数后台运行（可以设定cron）脚本检查域名到期时间，发现域名还有5天以内到期报警提示，方便抢注呵呵！

缺点：1  有些域名是没有域名信息的，比如a.com，p.com  所有就过滤掉了

2  程序执行时间不会对负载有影响，但是可能出现socket超时，我设置了socket连接三次，但是可能会出现一些未知错误break，大家发现BUG留言我再修改

#!/bin/env python
import sys
import string
import socket
import optparse
import smtplib
import re
import time
import itertools
reload(sys)
sys.setdefaultencoding('utf8') 


class Host(object):



    INTERHOST           ="whois.internic.net"
    NICHOST             = "whois.crsnic.net"
    INICHOST            = "whois.networksolutions.com"
    ip_whois = [ INTERHOST, NICHOST ,INICHOST ]



class Whois(object) :



    WHOIS_SERVER_ID     = "Whois Server:"
    WHOIS_ORG_SERVER_ID = "Registrant Street1:Whois Server:"
    WHOIS_RECURSE       = 0x01
    WHOIS_QUICK         = 0x02



    def whois(self, query, hostname, flags):
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        fails =0
        while True:
            try:
                if fails <=3:
                    s.connect((hostname, 43))
            except socket.error:
                fails +=1
            else:
                break
                print 'Socket Connection timed out'
        queryBytes = None
        queryBytes = (query  + "\r\n").encode()
        s.send(queryBytes)
        response = ''
        while True:
            d = s.recv(4096)
            response += d
            if not d:
                break
        s.close()
        nhost = None
        nhost = self.findwhois_server(response.decode(), hostname)
        if (nhost != None):
            response += self.whois(query, nhost, 0)
        return response.decode()
    def findwhois_server(self, buf, hostname):
        nhost = None
        parts_index = 1
        start = buf.find(Whois.WHOIS_SERVER_ID)
        if (start == -1):
            start = buf.find(Whois.WHOIS_ORG_SERVER_ID)
            parts_index = 2



        if (start > -1):
            end = buf[start:].find('\n')
            whois_line = buf[start:end+start]
            whois_parts = whois_line.split(':')
            nhost = whois_parts[parts_index].strip()
        else:
            for nichost in Host.ip_whois:
                if (buf.find(nichost) != -1):
                    nhost = nichost
                    break
        return nhost



    def whois_lookup(self, options, query_arg, flags):
        nichost = None
        if (options == None):
            options = {}



        options['whoishost'] = Host.NICHOST
        if ( not (flags & Whois.WHOIS_QUICK)):
            flags |= Whois.WHOIS_RECURSE
        return self.whois(query_arg, options['whoishost'], flags)   



class WhoisEntry(object):
    _regex = {
        'domain_name':      'Domain Name:\s?(.+)',
        'registrar':        'Registrar:\s?(.+)',
        'whois_server':     'Whois Server:\s?(.+)',
        'referral_url':     'Referral URL:\s?(.+)',
        'updated_date':     'Updated Date:\s?(.+)',
        'creation_date':    'Creation Date:\s?(.+)',
        'expiration_date':  'Expiration Date:\s?(.+)',
        'name_servers':     'Name Server:\s?(.+)',
        'status':           'Status:\s?(.+)',
        'emails':           '[\w.-]+@[\w.-]+\.[\w]{2,4}',
    }
    def __init__(self, domain, text, regex=None):
        self.domain = domain
        self.text = text
        if regex is not None:
            self._regex = regex



    def __getattr__(self, attr):
        whois_regex = self._regex.get(attr)
        if whois_regex:
            setattr(self, attr, re.findall(whois_regex, self.text))
            return getattr(self, attr)



        else:
            raise KeyError('Unknown attribute: %s' % attr)



    def __str__(self):
        return '\n'.join('%s: %s' % (attr, str(getattr(self, attr))) for attr in self.attrs())



    def attrs(self):
        return sorted(self._regex.keys())



    @staticmethod
    def load(domain, text):
        if text.strip() == 'No whois server is known for this kind of object.':
            pass
        return WhoisCom(domain, text)
class WhoisCom(WhoisEntry):
    def __init__(self, domain, text):
        if 'No match for "' in text:
            pass
        else:
            WhoisEntry.__init__(self, domain, text)
class Email(object):
    def __init__(self,body):
        mail_server = 'mail.dongwm.com'
        mail_server_port = 25
        from_addr='whois_check@dongwm.com'
        to_addr ='dongwm@dongwm.com'
        from_header = 'From: %s\r\n' % from_addr#    to_header = 'To: %s\r\n\r\n' % to_addr
        to_header = 'To: %s\r\n\r\n' % to_addr
        subject_header = 'Subject: Yuming check'
        email_message = '%s\n%s\n%s\n\n%s' % (from_header, to_header, subject_header, body)
        s = smtplib.SMTP(mail_server, mail_server_port)
        s.sendmail(from_addr, to_addr, email_message)
        s.quit()



def parse_command_line(argv):
    usage = "usage: %prog [options] name"
    parser = optparse.OptionParser(add_help_option=False, usage=usage)
    parser.add_option("-d", "--domain", action="store_const",
                      const=Host.NICHOST, dest="whoishost",
                      help="Query for a specific domain name, please use: XXX.com format")
    parser.add_option("-h", "--help", action="help")
    return parser.parse_args(argv)



if __name__ == "__main__":
     whoisclient = Whois()
     (options, args) = parse_command_line(sys.argv)
     if len(sys.argv)==3 and re.sub('^.*://', '', args[1]).split('/')[0].lower().split('.')[1]  in 'com':
             text = whoisclient.whois_lookup(options.__dict__,args[1], 0)
             print  WhoisEntry.load(args[1], text)



     elif len(sys.argv)==1:
         #for i in range(97,123):
         #   print chr(i)
         len = 1
         while (len<6):
             for i in list(itertools.permutations(string.lowercase,len)):
                 if whoisclient.whois_lookup(options.__dict__,''.join(i)+'.com', 0).encode('utf8').find('Expiration Date: ') != -1:
                     sa = whoisclient.whois_lookup(options.__dict__,''.join(i)+'.com', 0).encode('utf8').split('Expiration Date: ')[1].split(r'>>> Last')[0][:-2]
                     wtime = time.strftime('%d-%b-%Y',time.localtime(time.time()))
                     if wtime.lower()[3:6] == sa[3:6] and wtime[7:11] == sa[7:11] and int(sa[0:2]) - int(wtime[0:2]) <=5 and int(sa[0:2]) - int(wtime[0:2]) >=0:
                         tt = '%s is %s' % (''.join(i)+'.com',sa)
                         Email(tt)
                         time.sleep(1)
             len +=1
     else:
             print "Please use the '-h' or '--help' parameter"