feat(team): support team

app/database/__init__.py

@@ -31,6 +31,7 @@ from .lazer_user import (
 )
 from .multiplayer_event import MultiplayerEvent, MultiplayerEventResp
 from .notification import Notification, UserNotification
+from .password_reset import PasswordReset
 from .playlist_attempts import (
     ItemAttemptsCount,
     ItemAttemptsResp,
@@ -56,7 +57,7 @@ from .statistics import (
     UserStatistics,
     UserStatisticsResp,
 )
-from .team import Team, TeamMember
+from .team import Team, TeamMember, TeamRequest
 from .user_account_history import (
     UserAccountHistory,
     UserAccountHistoryResp,
@@ -81,9 +82,11 @@ __all__ = [
     "CountResp",
     "DailyChallengeStats",
     "DailyChallengeStatsResp",
+    "EmailVerification",
     "FavouriteBeatmapset",
     "ItemAttemptsCount",
     "ItemAttemptsResp",
+    "LoginSession",
     "MonthlyPlaycounts",
     "MultiplayerEvent",
     "MultiplayerEventResp",
@@ -92,6 +95,7 @@ __all__ = [
     "OAuthClient",
     "OAuthToken",
     "PPBestScore",
+    "PasswordReset",
     "Playlist",
     "PlaylistAggregateScore",
     "PlaylistBestScore",
@@ -115,6 +119,7 @@ __all__ = [
     "ScoreTokenResp",
     "Team",
     "TeamMember",
+    "TeamRequest",
     "User",
     "UserAccountHistory",
     "UserAccountHistoryResp",

app/database/team.py

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import UTC, datetime
 from typing import TYPE_CHECKING
 
 from app.models.model import UTCBaseModel
@@ -16,19 +16,21 @@ class Team(SQLModel, UTCBaseModel, table=True):
     id: int | None = Field(default=None, primary_key=True, index=True)
     name: str = Field(max_length=100)
     short_name: str = Field(max_length=10)
-    flag_url: str | None = Field(default=None, max_length=500)
-    created_at: datetime = Field(
-        default_factory=datetime.utcnow, sa_column=Column(DateTime)
-    )
+    flag_url: str | None = Field(default=None)
+    cover_url: str | None = Field(default=None)
+    created_at: datetime = Field(default=datetime.now(UTC), sa_column=Column(DateTime))
+    leader_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id")))
 
+    leader: "User" = Relationship()
     members: list["TeamMember"] = Relationship(back_populates="team")
 
 
 class TeamMember(SQLModel, UTCBaseModel, table=True):
     __tablename__ = "team_members"  # pyright: ignore[reportAssignmentType]
 
-    id: int | None = Field(default=None, primary_key=True, index=True)
-    user_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id")))
+    user_id: int = Field(
+        sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), primary_key=True)
+    )
     team_id: int = Field(foreign_key="teams.id")
     joined_at: datetime = Field(
         default_factory=datetime.utcnow, sa_column=Column(DateTime)
@@ -40,3 +42,18 @@ class TeamMember(SQLModel, UTCBaseModel, table=True):
     team: "Team" = Relationship(
         back_populates="members", sa_relationship_kwargs={"lazy": "joined"}
     )
+
+
+class TeamRequest(SQLModel, UTCBaseModel, table=True):
+    __tablename__ = "team_requests"  # pyright: ignore[reportAssignmentType]
+
+    user_id: int = Field(
+        sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), primary_key=True)
+    )
+    team_id: int = Field(foreign_key="teams.id", primary_key=True)
+    requested_at: datetime = Field(
+        default=datetime.now(UTC), sa_column=Column(DateTime)
+    )
+
+    user: "User" = Relationship(sa_relationship_kwargs={"lazy": "joined"})
+    team: "Team" = Relationship(sa_relationship_kwargs={"lazy": "joined"})