1527e23b43
* chore(deps): add pyotp * feat(auth): implement TOTP verification feat(auth): implement TOTP verification and email verification services - Added TOTP keys management with a new database model `TotpKeys`. - Introduced `EmailVerification` and `LoginSession` models for email verification. - Created `verification_service` to handle email verification logic and TOTP processes. - Updated user response models to include session verification methods. - Implemented routes for TOTP creation, verification, and fallback to email verification. - Enhanced login session management to support new location checks and verification methods. - Added migration script to create `totp_keys` table in the database. * feat(config): update config example * docs(totp): complete creating TOTP flow * refactor(totp): resolve review * feat(api): forbid unverified request * fix(totp): trace session by token id to avoid other sessions are forbidden * chore(linter): make pyright happy * fix(totp): only mark sessions with a specified token id
150 lines
3.6 KiB
Python
150 lines
3.6 KiB
Python
from .achievement import UserAchievement, UserAchievementResp
|
|
from .auth import OAuthClient, OAuthToken, TotpKeys, V1APIKeys
|
|
from .beatmap import (
|
|
Beatmap,
|
|
BeatmapResp,
|
|
)
|
|
from .beatmap_playcounts import BeatmapPlaycounts, BeatmapPlaycountsResp
|
|
from .beatmap_tags import BeatmapTagVote
|
|
from .beatmapset import (
|
|
Beatmapset,
|
|
BeatmapsetResp,
|
|
)
|
|
from .beatmapset_ratings import BeatmapRating
|
|
from .best_score import BestScore
|
|
from .chat import (
|
|
ChannelType,
|
|
ChatChannel,
|
|
ChatChannelResp,
|
|
ChatMessage,
|
|
ChatMessageResp,
|
|
)
|
|
from .counts import (
|
|
CountResp,
|
|
MonthlyPlaycounts,
|
|
ReplayWatchedCount,
|
|
)
|
|
from .daily_challenge import DailyChallengeStats, DailyChallengeStatsResp
|
|
from .events import Event
|
|
from .favourite_beatmapset import FavouriteBeatmapset
|
|
from .lazer_user import (
|
|
MeResp,
|
|
User,
|
|
UserResp,
|
|
)
|
|
from .multiplayer_event import MultiplayerEvent, MultiplayerEventResp
|
|
from .notification import Notification, UserNotification
|
|
from .password_reset import PasswordReset
|
|
from .playlist_attempts import (
|
|
ItemAttemptsCount,
|
|
ItemAttemptsResp,
|
|
PlaylistAggregateScore,
|
|
)
|
|
from .playlist_best_score import PlaylistBestScore
|
|
from .playlists import Playlist, PlaylistResp
|
|
from .pp_best_score import PPBestScore
|
|
from .rank_history import RankHistory, RankHistoryResp, RankTop
|
|
from .relationship import Relationship, RelationshipResp, RelationshipType
|
|
from .room import APIUploadedRoom, Room, RoomResp
|
|
from .room_participated_user import RoomParticipatedUser
|
|
from .score import (
|
|
MultiplayerScores,
|
|
Score,
|
|
ScoreAround,
|
|
ScoreBase,
|
|
ScoreResp,
|
|
ScoreStatistics,
|
|
)
|
|
from .score_token import ScoreToken, ScoreTokenResp
|
|
from .statistics import (
|
|
UserStatistics,
|
|
UserStatisticsResp,
|
|
)
|
|
from .team import Team, TeamMember, TeamRequest
|
|
from .user_account_history import (
|
|
UserAccountHistory,
|
|
UserAccountHistoryResp,
|
|
UserAccountHistoryType,
|
|
)
|
|
from .user_login_log import UserLoginLog
|
|
from .verification import EmailVerification, LoginSession
|
|
|
|
__all__ = [
|
|
"APIUploadedRoom",
|
|
"Beatmap",
|
|
"BeatmapPlaycounts",
|
|
"BeatmapPlaycountsResp",
|
|
"BeatmapRating",
|
|
"BeatmapResp",
|
|
"BeatmapTagVote",
|
|
"Beatmapset",
|
|
"BeatmapsetResp",
|
|
"BestScore",
|
|
"ChannelType",
|
|
"ChatChannel",
|
|
"ChatChannelResp",
|
|
"ChatMessage",
|
|
"ChatMessageResp",
|
|
"CountResp",
|
|
"DailyChallengeStats",
|
|
"DailyChallengeStatsResp",
|
|
"EmailVerification",
|
|
"Event",
|
|
"FavouriteBeatmapset",
|
|
"ItemAttemptsCount",
|
|
"ItemAttemptsResp",
|
|
"LoginSession",
|
|
"MeResp",
|
|
"MonthlyPlaycounts",
|
|
"MultiplayerEvent",
|
|
"MultiplayerEventResp",
|
|
"MultiplayerScores",
|
|
"Notification",
|
|
"OAuthClient",
|
|
"OAuthToken",
|
|
"PPBestScore",
|
|
"PasswordReset",
|
|
"Playlist",
|
|
"PlaylistAggregateScore",
|
|
"PlaylistBestScore",
|
|
"PlaylistResp",
|
|
"RankHistory",
|
|
"RankHistoryResp",
|
|
"RankTop",
|
|
"Relationship",
|
|
"RelationshipResp",
|
|
"RelationshipType",
|
|
"ReplayWatchedCount",
|
|
"Room",
|
|
"RoomParticipatedUser",
|
|
"RoomResp",
|
|
"Score",
|
|
"ScoreAround",
|
|
"ScoreBase",
|
|
"ScoreResp",
|
|
"ScoreStatistics",
|
|
"ScoreToken",
|
|
"ScoreTokenResp",
|
|
"Team",
|
|
"TeamMember",
|
|
"TeamRequest",
|
|
"TotpKeys",
|
|
"User",
|
|
"UserAccountHistory",
|
|
"UserAccountHistoryResp",
|
|
"UserAccountHistoryType",
|
|
"UserAchievement",
|
|
"UserAchievement",
|
|
"UserAchievementResp",
|
|
"UserLoginLog",
|
|
"UserNotification",
|
|
"UserResp",
|
|
"UserStatistics",
|
|
"UserStatisticsResp",
|
|
"V1APIKeys",
|
|
]
|
|
|
|
for i in __all__:
|
|
if i.endswith("Resp"):
|
|
globals()[i].model_rebuild() # type: ignore[call-arg]
|