From 1994ba5917dbef5dcafc38c82d3548b2b4e9efd0 Mon Sep 17 00:00:00 2001 From: lan Date: Wed, 14 Dec 2022 10:21:11 +0800 Subject: [PATCH] =?UTF-8?q?update:=E5=85=B3=E9=97=ADdocs=EF=BC=8Credoc?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- depends.py | 6 +++--- main.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) 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)