Class Quake

api/quake.py:12–64

class Quake:
    def __init__(self):
        self.headers = {
            "User-Agent": random.choice(user_agents),
            "X-QuakeToken": QuakeKey
        }
        if QuakeKey == "":
            logging.warning("请先在config/config.py文件中配置quake的api")
            exit(0)
        
        try:
            logging.info("[QUAKE Example]domain:example.com\n")
            while 1:
                self.keywords = input("请输入查询关键词:").strip()
                self.size = input("请输入查询数量:").strip()
                if self.keywords == "" or self.size == "":
                    logging.error("\n关键字或查询数量不能为空！")
                elif self.size.isdigit() != True:
                    logging.error("\n查询数量非整数！")
                else:
                    break
            self.run()
        except KeyboardInterrupt:
            logging.error("\n用户取消输入！直接退出。")
            exit(0)
    

    def run(self):
        logging.info("正在使用使用360 Quake进行资产收集。。。")
        logging.info("查询关键词为:{0},查询数量为:{1}".format(self.keywords, self.size))
        self.data = {
            "query": self.keywords,
            "start": 0,
            "size": self.size
        }
        try:
            response = requests.post(url="https://quake.360.cn/api/v3/search/quake_service", headers=self.headers,
                                     json=self.data, timeout=10)
            datas = json.loads(response.text)
            if len(datas['data']) >= 1 and datas['code'] == 0:
                for data in datas['data']:
                    port = "" if data['port'] == 80 or data["port"] == 443 else ":{}".format(str(data['port']))
                    if 'http/ssl' == data['service']['name']:
                        url = 'https://' + data['service']['http']['host'] + port
                        logging.info(url)
                        Urls.url.append(url)
                    elif 'http' == data['service']['name']:
                        url = 'http://' + data['service']['http']['host'] + port
                        logging.info(url)
                        Urls.url.append(url)
        except Exception as e:
            logging.error("获取失败")
            pass