Skip to content

Commit 9e7702d

Browse files
committed
fix: 修复 ruff lint 和格式化问题
- 修复 E712: 使用 .is_(True) 替代 == True - 修复 N802: CORS_ORIGINS 改为 cors_origins - 修复 UP035: 从 collections.abc 导入 - 修复 I001: 导入排序 - 格式化 33 个文件
1 parent 8338b67 commit 9e7702d

33 files changed

+204
-118
lines changed

apps/api/alembic/env.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
"""Alembic 环境配置"""
2+
23
import asyncio
34
from logging.config import fileConfig
45

5-
from alembic import context
66
from sqlalchemy import pool
77
from sqlalchemy.engine import Connection
88
from sqlalchemy.ext.asyncio import async_engine_from_config
99

10+
from alembic import context
1011
from app.core.config import settings
1112
from app.db.base import Base
1213
from app.db.tables import * # noqa: F401, F403 - 导入所有表

apps/api/app/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
"""QueryGPT API Server"""
2+
23
__version__ = "2.0.0"

apps/api/app/api/deps.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""API 依赖注入"""
2+
23
from uuid import UUID
34

45
from fastapi import Depends, HTTPException, status

apps/api/app/api/v1/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""API v1 路由"""
2+
23
from fastapi import APIRouter
34

45
from app.api.v1 import auth, chat, connections, history, models, user_config

apps/api/app/api/v1/auth.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,18 @@
11
"""认证 API"""
2+
23
from uuid import UUID
34

45
from fastapi import APIRouter, Depends, HTTPException, status
56
from sqlalchemy import select
67
from sqlalchemy.ext.asyncio import AsyncSession
78

9+
from app.api.deps import get_current_user
810
from app.core import create_access_token, create_refresh_token, get_password_hash, verify_password
911
from app.core.config import settings
1012
from app.core.demo_db import get_demo_db_path
1113
from app.db import get_db
1214
from app.db.tables import Connection, User
1315
from app.models import APIResponse, Token, UserCreate, UserLogin, UserResponse
14-
from app.api.deps import get_current_user
1516

1617
router = APIRouter()
1718

apps/api/app/api/v1/chat.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
"""聊天 API"""
2+
23
import asyncio
3-
from typing import AsyncGenerator
4-
from uuid import UUID, uuid4
4+
from collections.abc import AsyncGenerator
5+
from uuid import UUID
56

67
from fastapi import APIRouter, Depends, HTTPException, Query, status
78
from sqlalchemy import select
8-
from sse_starlette.sse import EventSourceResponse
99
from sqlalchemy.ext.asyncio import AsyncSession
10+
from sse_starlette.sse import EventSourceResponse
1011

1112
from app.api.deps import get_current_user
1213
from app.core import decode_token

apps/api/app/api/v1/connections.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""数据库连接管理 API"""
2+
23
from uuid import UUID
34

45
from fastapi import APIRouter, Depends, HTTPException, status
@@ -40,9 +41,7 @@ async def create_connection(
4041
# 如果设为默认,取消其他默认
4142
if conn_in.is_default:
4243
result = await db.execute(
43-
select(Connection).where(
44-
Connection.user_id == current_user.id, Connection.is_default == True
45-
)
44+
select(Connection).where(Connection.user_id == current_user.id, Connection.is_default)
4645
)
4746
for c in result.scalars():
4847
c.is_default = False
@@ -137,9 +136,7 @@ async def update_connection(
137136
# 如果设为默认,取消其他默认
138137
if conn_in.is_default and not connection.is_default:
139138
other_result = await db.execute(
140-
select(Connection).where(
141-
Connection.user_id == current_user.id, Connection.is_default == True
142-
)
139+
select(Connection).where(Connection.user_id == current_user.id, Connection.is_default)
143140
)
144141
for c in other_result.scalars():
145142
c.is_default = False

apps/api/app/api/v1/history.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""历史记录 API"""
2+
23
from uuid import UUID
34

45
from fastapi import APIRouter, Depends, HTTPException, Query, status
@@ -40,7 +41,7 @@ async def list_conversations(
4041
)
4142

4243
if favorites:
43-
query = query.where(Conversation.is_favorite == True)
44+
query = query.where(Conversation.is_favorite)
4445

4546
if q:
4647
query = query.where(Conversation.title.ilike(f"%{q}%"))

apps/api/app/api/v1/models.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""模型配置管理 API"""
2+
23
import time
34
from uuid import UUID
45

@@ -38,12 +39,9 @@ async def create_model(
3839
"""添加模型配置"""
3940
# 如果设为默认,取消其他默认
4041
if model_in.is_default:
41-
await db.execute(
42-
select(Model)
43-
.where(Model.user_id == current_user.id, Model.is_default == True)
44-
)
42+
await db.execute(select(Model).where(Model.user_id == current_user.id, Model.is_default))
4543
result = await db.execute(
46-
select(Model).where(Model.user_id == current_user.id, Model.is_default == True)
44+
select(Model).where(Model.user_id == current_user.id, Model.is_default)
4745
)
4846
for m in result.scalars():
4947
m.is_default = False
@@ -89,7 +87,7 @@ async def update_model(
8987
# 如果设为默认,取消其他默认
9088
if model_in.is_default and not model.is_default:
9189
other_result = await db.execute(
92-
select(Model).where(Model.user_id == current_user.id, Model.is_default == True)
90+
select(Model).where(Model.user_id == current_user.id, Model.is_default)
9391
)
9492
for m in other_result.scalars():
9593
m.is_default = False

apps/api/app/api/v1/user_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""用户配置 API"""
2+
23
from fastapi import APIRouter, Depends
34
from sqlalchemy.ext.asyncio import AsyncSession
45

0 commit comments

Comments
 (0)