Update fofa账号查询.py

py/fofa账号查询.py

@@ -6,58 +6,40 @@ API_URL = "https://fofa.info/api/v1/search/all"  # FOFA搜索接口URL
 API_KEY = "849489858@qq.com"  # 假设的API密钥
 SECRET = "30bffdb13deccd433f6505f2dc699972"  # 假设的密文
 
-# 搜索参数
+PAGE = 1
-QUERY = "udpxy"  # 用户自定义的查询语句，例如: "protocol=\"http\""
+PAGE_SIZE = 50  # 每页条数，fofa API 限制最大 100
-PAGE = 1  # 请求的页码
-PAGE_SIZE = 50  # 每页条数，最大100
  
 def get_token(key, secret):
-    """
+    url = "https://fofa.info/api/v1/info/ip"
-    获取访问令牌的函数
-    :param key: API密钥
-    :param secret: 与API密钥对应的密文
-    :return: 访问令牌
-    """
-    url = "https://fofa.info/api/v1/info/ip"  # 获取Token的API接口URL
     data = {
-        "email": key,  # 传入API密钥
+        "email": key,
-        "key": secret  # 传入密文
+        "key": secret
     }
-    response = requests.post(url, json=data)  # 发送POST请求获取Token
+    response = requests.post(url, data=data)
-    return response.json().get("data", {}).get("Token")  # 安全地获取Token
+    return response.json()["data"]["Token"]
  
 def search_fofa(query, page, page_size, token):
-    """
+    url = API_URL
-    使用FOFA进行搜索的函数
-    :param query: 搜索查询语句
-    :param page: 当前页码
-    :param page_size: 每页条数
-    :param token: 访问令牌
-    :return: 搜索结果
-    """
-    url = API_URL  # 使用FOFA的搜索接口URL
     data = {
-        "email": API_KEY,  # 传入API密钥
+        "email": API_KEY,
-        "key": SECRET,  # 传入密文
+        "key": SECRET,
-        "qbase64": base64.b64encode(query.encode('utf-8')).decode('utf-8'),  # 对查询语句进行Base64编码
+        "qbase64": base64.b64encode(query.encode('utf-8')).decode('utf-8'),
-        "page": page,  # 页码
+        "page": page,
-        "size": page_size,  # 每页条数
+        "size": page_size,
-        "token": token  # 访问令牌
+        "token": token
     }
     headers = {
-        "Range": f"items={page}-{page*page_size}"  # 设置请求头，指定请求的数据范围
+        "Range": f"items {page}-{page_size}"
     }
-    response = requests.post(url, data=data, headers=headers)  # 发送POST请求进行搜索
+    response = requests.post(url, data=data, headers=headers)
-    return response.json()  # 返回从响应中解析出的JSON数据
+    return response.json()
  
 if __name__ == "__main__":
-    token = get_token(API_KEY, SECRET)  # 调用get_token函数获取Token
+    token = get_token(API_KEY, SECRET)
-    if token:  # 确保获取到有效的Token
+    results = search_fofa(QUERY, PAGE, PAGE_SIZE, token)
-        results = search_fofa(QUERY, PAGE, PAGE_SIZE, token)  # 调用search_fofa函数进行搜索
     # 处理结果
-        if "error" in results and results["error"]:  # 检查是否有错误信息
+    for result in results:
-            print(results["errmsg"])  # 打印错误信息
+        print(result)
-        else:
             for result in results.get("data", []):  # 安全地获取数据
                 print(result)  # 打印每个结果
     else: