init

favicon_app/routes/favicon_routes.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+
+import logging
+import os
+from typing import Optional
+
+import urllib3
+from fastapi import APIRouter, Request, Query, BackgroundTasks
+from fastapi.responses import Response
+
+import setting
+from favicon_app.routes import favicon_service
+from favicon_app.utils.file_util import FileUtil
+
+urllib3.disable_warnings()
+logging.captureWarnings(True)
+logger = logging.getLogger(__name__)
+
+_icon_root_path = setting.icon_root_path
+_default_icon_path = setting.default_icon_path
+
+# 创建全局服务实例
+_service = favicon_service.FaviconService()
+
+# 创建FastAPI路由器
+favicon_router = APIRouter(prefix="", tags=["favicon"])
+
+
+@favicon_router.get('/icon/')
+@favicon_router.get('/icon')
+async def get_favicon(
+        request: Request,
+        bg_tasks: BackgroundTasks,
+        url: Optional[str] = Query(None, description="网址：eg. https://www.baidu.com"),
+        refresh: Optional[str] = Query(None, include_in_schema=False),
+):
+    """获取网站图标"""
+    return await _service.get_favicon_handler(request, bg_tasks, url, refresh)
+
+
+@favicon_router.get('/icon/default')
+async def get_default_icon():
+    """获取默认图标"""
+    return _service.get_default()
+
+
+@favicon_router.get('/icon/referer', include_in_schema=False)
+async def get_referrer(unique: Optional[str] = Query(None)):
+    """获取请求来源信息，带unique参数时会进行去重处理"""
+    content = 'None'
+    _path = os.path.join(_icon_root_path, 'data', 'referer.txt')
+
+    if os.path.exists(_path):
+        try:
+            content = FileUtil.read_file(_path, mode='r') or 'None'
+
+            if unique in ['true', '1']:
+                lines = [line.strip() for line in content.split('\n') if line.strip()]
+                unique_lines = list(set(lines))
+                unique_content = '\n'.join(unique_lines)
+                FileUtil.write_file(_path, unique_content, mode='w')
+                content = unique_content
+        except Exception as e:
+            logger.error(f"读取referer文件失败: {e}")
+    return Response(content=content, media_type="text/plain")