feat(auth): add created_at and updated_at in OAuthToken

resolve #78
This commit is contained in:
MingxuanGame
2025-11-23 13:56:28 +00:00
parent e049056534
commit 7b34bd8b0b
3 changed files with 58 additions and 24 deletions

View File

@@ -16,6 +16,7 @@ from sqlmodel import (
Relationship,
SQLModel,
Text,
text,
)
if TYPE_CHECKING:
@@ -40,14 +41,20 @@ class OAuthToken(UTCBaseModel, SQLModel, table=True):
login_session: LoginSession | None = Relationship(back_populates="token", passive_deletes=True)
class OAuthClient(SQLModel, table=True):
class OAuthClient(UTCBaseModel, SQLModel, table=True):
__tablename__: str = "oauth_clients"
name: str = Field(max_length=100, index=True)
description: str = Field(sa_column=Column(Text), default="")
client_id: int | None = Field(default=None, primary_key=True, index=True)
client_secret: str = Field(default_factory=secrets.token_hex, index=True)
client_secret: str = Field(default_factory=secrets.token_hex, index=True, exclude=True)
redirect_uris: list[str] = Field(default_factory=list, sa_column=Column(JSON))
owner_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), index=True))
owner_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), index=True), exclude=True)
created_at: datetime = Field(default_factory=utcnow, sa_column=Column(DateTime))
updated_at: datetime = Field(
default_factory=utcnow,
sa_column=Column(DateTime, onupdate=text("CURRENT_TIMESTAMP")),
)
class V1APIKeys(SQLModel, table=True):

View File

@@ -46,9 +46,8 @@ async def create_oauth_app(
await session.commit()
await session.refresh(oauth_client)
return {
"client_id": oauth_client.client_id,
"client_secret": oauth_client.client_secret,
"redirect_uris": oauth_client.redirect_uris,
**oauth_client.model_dump(exclude={"client_secret"}),
}
@@ -57,6 +56,7 @@ async def create_oauth_app(
name="获取 OAuth 应用信息",
description="通过客户端 ID 获取 OAuth 应用的详细信息",
tags=["osu! OAuth 认证", "g0v0 API"],
response_model=OAuthClient,
)
async def get_oauth_app(
session: Database,
@@ -66,12 +66,7 @@ async def get_oauth_app(
oauth_app = await session.get(OAuthClient, client_id)
if not oauth_app:
raise HTTPException(status_code=404, detail="OAuth app not found")
return {
"name": oauth_app.name,
"description": oauth_app.description,
"redirect_uris": oauth_app.redirect_uris,
"client_id": oauth_app.client_id,
}
return oauth_app
@router.get(
@@ -79,21 +74,14 @@ async def get_oauth_app(
name="获取用户的 OAuth 应用列表",
description="获取当前用户创建的所有 OAuth 应用程序",
tags=["osu! OAuth 认证", "g0v0 API"],
response_model=list[OAuthClient],
)
async def get_user_oauth_apps(
session: Database,
current_user: ClientUser,
):
oauth_apps = await session.exec(select(OAuthClient).where(OAuthClient.owner_id == current_user.id))
return [
{
"name": app.name,
"description": app.description,
"redirect_uris": app.redirect_uris,
"client_id": app.client_id,
}
for app in oauth_apps
]
return oauth_apps.all()
@router.delete(
@@ -150,9 +138,8 @@ async def update_oauth_app(
await session.refresh(oauth_client)
return {
"client_id": oauth_client.client_id,
"client_secret": oauth_client.client_secret,
"redirect_uris": oauth_client.redirect_uris,
**oauth_client.model_dump(exclude={"client_secret"}),
}
@@ -182,9 +169,8 @@ async def refresh_secret(
await session.refresh(oauth_client)
return {
"client_id": oauth_client.client_id,
"client_secret": oauth_client.client_secret,
"redirect_uris": oauth_client.redirect_uris,
**oauth_client.model_dump(exclude={"client_secret"}),
}