Update main.py

main.py

@@ -1,6 +1,26 @@
 import ipaddress
 import maxminddb
 from fastapi import FastAPI, Request
+import json
+import datetime
+import logging
+import sys
+
+# 配置日志
+try:
+    logging.basicConfig(
+        filename='ip_query.log',
+        level=logging.INFO,
+        format='%(message)s'
+    )
+    logging.info(json.dumps({
+        "时间": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "事件": "系统启动",
+        "状态": "成功"
+    }, ensure_ascii=False))
+except Exception as e:
+    print(f"日志初始化失败: {e}")
+    sys.exit(1)
 
 city_reader = maxminddb.open_database('GeoLite2-City.mmdb')
 asn_reader = maxminddb.open_database('GeoLite2-ASN.mmdb')
@@ -25,7 +45,7 @@ asn_map = {
     4134:"中国电信",4812:"中国电信",23724:"中国电信",136188:"中国电信",137693:"中国电信",17638:"中国电信",
     140553:"中国电信",4847:"中国电信",140061:"中国电信",136195:"中国电信",17799:"中国电信",139018:"中国电信",
     133776:"中国电信",58772:"中国电信",146966:"中国电信",63527:"中国电信",58539:"中国电信",58540:"中国电信",
-    141998:"中国电信",138169:"中国电信",139203:"中国电信",58563:"中国电信",137690:"中国电信",63838:"中国电信",
+    141998:"中国电信",138169:"中国电信",139203:"中国电信",58563:"中电信",137690:"中国电信",63838:"中国电信",
     137694:"中国电信",137698:"中国电信",136167:"中国电信",148969:"中国电信",134764:"中国电信",
     134770:"中国电信",148981:"中国电信",134774:"中国电信",136190:"中国电信",140647:"中国电信",
     132225:"中国电信",140485:"中国电信",4811:"中国电信",131285:"中国电信",137689:"中国电信",
@@ -192,16 +212,51 @@ def query():
 app = FastAPI()
 
 @app.get("/")
-def api(request: Request, ip: str = None):
-    if not ip:
-        ip = request.headers.get("x-forwarded-for") or request.headers.get("x-real-ip") or request.client.host
-    return get_ip_info(ip.strip())
+async def api(request: Request, ip: str = None):
+    # 获取请求信息
+    client_ip = request.headers.get("x-forwarded-for") or request.headers.get("x-real-ip") or request.client.host
+    query_ip = ip.strip() if ip else client_ip
+    
+    # 获取IP信息
+    result = get_ip_info(query_ip)
+    
+    # 构建日志信息
+    log_data = {
+        "时间": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "访问IP": client_ip,
+        "查询IP": query_ip,
+        "请求头": dict(request.headers),
+        "查询结果": result
+    }
+    
+    # 写入日志
+    logging.info(json.dumps(log_data, ensure_ascii=False))
+    
+    return result
 
 @app.get("/{ip}")
-def path_api(ip):
-    return get_ip_info(ip)
+async def path_api(request: Request, ip: str):
+    # 获取请求信息
+    client_ip = request.headers.get("x-forwarded-for") or request.headers.get("x-real-ip") or request.client.host
+    
+    # 获取IP信息
+    result = get_ip_info(ip)
+    
+    # 构建日志信息
+    log_data = {
+        "时间": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "访问IP": client_ip,
+        "查询IP": ip,
+        "请求头": dict(request.headers),
+        "查询结果": result
+    }
+    
+    # 写入日志
+    logging.info(json.dumps(log_data, ensure_ascii=False))
+    
+    return result
 
 if __name__ == '__main__':
     query()
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8080, server_header=False, proxy_headers=True)
+    uvicorn.run(app, host="0.0.0.0", port=8080, server_header=False, proxy_headers=True)