diff --git a/depends.py b/depends.py index 0ca7b15..e92bb40 100644 --- a/depends.py +++ b/depends.py @@ -34,10 +34,10 @@ class IPRateLimit: self.ips.pop(ip) return True - def add_ip(cls, ip): - ip_info = cls.ips.get(ip, {'count': 0, 'time': datetime.now()}) + def add_ip(self, ip): + ip_info = self.ips.get(ip, {'count': 0, 'time': datetime.now()}) ip_info['count'] += 1 - cls.ips[ip] = ip_info + self.ips[ip] = ip_info return ip_info['count'] async def remove_expired_ip(self): diff --git a/main.py b/main.py index e7ab99a..4628a02 100644 --- a/main.py +++ b/main.py @@ -3,7 +3,7 @@ import uuid import asyncio from pathlib import Path -from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException, BackgroundTasks, Header +from fastapi import FastAPI, Depends, UploadFile, Form, File, HTTPException, BackgroundTasks from starlette.responses import HTMLResponse, FileResponse from starlette.staticfiles import StaticFiles @@ -16,7 +16,7 @@ from database import get_session, Codes, init_models from depends import admin_required # 实例化FastAPI -app = FastAPI(debug=settings.DEBUG) +app = FastAPI(debug=settings.DEBUG, docs_url=None, redoc_url=None) # 数据存储文件夹 DATA_ROOT = Path(settings.DATA_ROOT)