54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.responses import Response
|
|
|
|
from favicon_app.routes import favicon_router
|
|
from favicon_app.utils.file_util import FileUtil
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 获取当前所在目录
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.append(os.path.dirname(current_dir))
|
|
# 站点的 favicon.ico 图标
|
|
favicon_icon_file = FileUtil.read_file(os.path.join(current_dir, 'favicon.ico'), mode='rb')
|
|
# 默认的站点图标
|
|
default_icon_file = FileUtil.read_file(os.path.join(current_dir, 'favicon.png'), mode='rb')
|
|
|
|
# 定义referer日志文件路径
|
|
referer_log_file = os.path.join(current_dir, 'data', 'referer.txt')
|
|
|
|
# fastapi
|
|
app = FastAPI(title="Favicon API", description="获取网站favicon图标", version="3.0.0")
|
|
app.include_router(favicon_router)
|
|
|
|
|
|
@app.middleware("http")
|
|
async def log_referer(request: Request, call_next):
|
|
referer = request.headers.get('referrer') or request.headers.get('referer')
|
|
if referer:
|
|
FileUtil.write_file(referer_log_file, referer, mode='a', atomic=True)
|
|
response = await call_next(request)
|
|
|
|
return response
|
|
|
|
|
|
# @app.get("/")
|
|
async def root():
|
|
return {"message": "Welcome to Favicon API! Use /icon/?url=example.com to get favicon."}
|
|
|
|
|
|
@app.get("/favicon.ico", summary="favicon.ico", tags=["default"])
|
|
async def favicon_ico():
|
|
return Response(content=favicon_icon_file, media_type="image/x-icon")
|
|
|
|
|
|
@app.get("/favicon.png", summary="favicon.png", tags=["default"])
|
|
async def favicon_png():
|
|
return Response(content=default_icon_file, media_type="image/png")
|