# -*- coding: utf-8 -*-
'''
域名注冊查詢
'''
__author__ = 'Jimmy'
from sqlalchemy import Column, String,Integer, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
import requests
import json
from html.parser import HTMLParser
request_failure = []
domain_available = []
def writeToText(list,fn):
  file = open(fn, 'w')
  file.write(str(list))
  file.close()
class bodyJSON(HTMLParser):
  tag = False
  def handle_starttag(self, tag, attr):
    if tag == 'body':
      self.tag = True
  def handle_endtag(self, tag):
    if tag == 'body':
      self.tag = False
  def handle_data(self, data):
    if self.tag:
      self.data = data
  def getJSON(self):
    return self.data
Base = declarative_base()
class Words(Base):
  # 表的名字:
  __tablename__ = 'words'
  # 表的結(jié)構(gòu):
  ID = Column(Integer(), primary_key=True)
  word = Column(String(100))
  exchange = Column(String(1000))
  voice = Column(String(1000))
  times = Column(Integer())
# 初始化數(shù)據(jù)庫連接:
engine = create_engine('mysql+mysqlconnector://root:846880@localhost:3306/words')
# 創(chuàng)建DBSession類型:
DBSession = sessionmaker(bind=engine)
# 創(chuàng)建Session:
session = DBSession()
# 創(chuàng)建Query查詢，filter是where條件，最后調(diào)用one()返回唯一行，如果調(diào)用all()則返回所有行:
words = session.query(Words).filter(Words.ID).all()
def searchInaaw8(words):
  length = len(words)
  print('====開始搜索...=====共%d個單詞' %length)
  for i in range(0,length):
    word = words[i]
    url = 'http://www.aaw8.com/Api/DomainApi.aspx?domain=%s.com' % word.word
    r = requests.get(url)
    if r.status_code == 200:
      if r.headers['Content-Type'] == 'text/html':
        print('第%s個請求被拒絕,url = %s' % (i, url))
      else:
        body = bodyJSON()
        body.feed(r.text)
        res = json.loads(body.getJSON())
        if res['StateID'] == 210:
          print('第%d次，%s.com 未被注冊' % (i, word.word))
          domain_available.append(word.word)
        elif res['StateID'] == 0:
          print('第%d次，%s.com 查詢接口出錯' % (i, word.word))
          request_failure.append(word.word)
        elif res['StateID'] == 211:
          pass
          print('第%d次，%s.com 已經(jīng)被注冊' % (i, word.word))
        elif res['StateID'] == 213:
          print('第%d次，%s.com 查詢超時' % (i, word.word))
          request_failure.append(word.word)
        else:
          print('其他錯誤')
          request_failure.append(word.word)
        body.close()
    else:
      print('請求失敗')
      request_failure.append(word.word)
  print('查詢結(jié)束...')
  print('查詢失敗:')
  print(request_failure)
  writeToText(request_failure,'failure.text')
  print('未注冊域名:')
  print(domain_available)
  writeToText(request_failure,'available.text')
searchInaaw8(words)