424 lines
12 KiB
Python
424 lines
12 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
osu! API 模拟服务器的示例数据填充脚本
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
import time
|
|
|
|
from app.auth import get_password_hash
|
|
from app.database import (
|
|
DailyChallengeStats,
|
|
LazerUserAchievement,
|
|
LazerUserStatistics,
|
|
RankHistory,
|
|
User,
|
|
)
|
|
from app.dependencies.database import engine, get_db
|
|
|
|
from sqlmodel import SQLModel
|
|
|
|
# 创建所有表
|
|
SQLModel.metadata.create_all(bind=engine)
|
|
|
|
|
|
def create_sample_user():
|
|
"""创建示例用户数据"""
|
|
with next(get_db()) as db:
|
|
# 检查用户是否已存在
|
|
from sqlmodel import select
|
|
|
|
statement = select(User).where(User.name == "Googujiang")
|
|
existing_user = db.exec(statement).first()
|
|
if existing_user:
|
|
print("示例用户已存在，跳过创建")
|
|
return existing_user
|
|
|
|
# 当前时间戳
|
|
current_timestamp = int(time.time())
|
|
join_timestamp = int(datetime(2019, 11, 29, 17, 23, 13).timestamp())
|
|
last_visit_timestamp = int(datetime(2025, 7, 18, 16, 31, 29).timestamp())
|
|
|
|
# 创建用户
|
|
user = User(
|
|
name="Googujiang",
|
|
safe_name="googujiang", # 安全用户名（小写）
|
|
email="googujiang@example.com",
|
|
priv=1, # 默认权限
|
|
pw_bcrypt=get_password_hash("password123"), # 使用新的哈希方式
|
|
country="JP",
|
|
silence_end=0,
|
|
donor_end=0,
|
|
creation_time=join_timestamp,
|
|
latest_activity=last_visit_timestamp,
|
|
clan_id=0,
|
|
clan_priv=0,
|
|
preferred_mode=0, # 0 = osu!
|
|
play_style=0,
|
|
custom_badge_name=None,
|
|
custom_badge_icon=None,
|
|
userpage_content="「世界に忘れられた」",
|
|
api_key=None,
|
|
# # 兼容性字段
|
|
# avatar_url="https://a.ppy.sh/15651670?1732362658.jpeg",
|
|
# cover_url="https://assets.ppy.sh/user-profile-covers/15651670/0fc7b77adef39765a570e7f535bc383e5a848850d41a8943f8857984330b8bc6.jpeg",
|
|
# has_supported=True,
|
|
# interests="「世界に忘れられた」",
|
|
# location="咕谷国",
|
|
# website="https://gmoe.cc",
|
|
# playstyle=["mouse", "keyboard", "tablet"],
|
|
# profile_order=[
|
|
# "me",
|
|
# "recent_activity",
|
|
# "top_ranks",
|
|
# "medals",
|
|
# "historical",
|
|
# "beatmaps",
|
|
# "kudosu",
|
|
# ],
|
|
# beatmap_playcounts_count=3306,
|
|
# favourite_beatmapset_count=15,
|
|
# follower_count=98,
|
|
# graveyard_beatmapset_count=7,
|
|
# mapping_follower_count=1,
|
|
# previous_usernames=["hehejun"],
|
|
# monthly_playcounts=[
|
|
# {"start_date": "2019-11-01", "count": 43},
|
|
# {"start_date": "2020-04-01", "count": 216},
|
|
# {"start_date": "2020-05-01", "count": 656},
|
|
# {"start_date": "2020-07-01", "count": 158},
|
|
# {"start_date": "2020-08-01", "count": 174},
|
|
# {"start_date": "2020-10-01", "count": 13},
|
|
# {"start_date": "2020-11-01", "count": 52},
|
|
# {"start_date": "2020-12-01", "count": 140},
|
|
# {"start_date": "2021-01-01", "count": 359},
|
|
# {"start_date": "2021-02-01", "count": 452},
|
|
# {"start_date": "2021-03-01", "count": 77},
|
|
# {"start_date": "2021-04-01", "count": 114},
|
|
# {"start_date": "2021-05-01", "count": 270},
|
|
# {"start_date": "2021-06-01", "count": 148},
|
|
# {"start_date": "2021-07-01", "count": 246},
|
|
# {"start_date": "2021-08-01", "count": 56},
|
|
# {"start_date": "2021-09-01", "count": 136},
|
|
# {"start_date": "2021-10-01", "count": 45},
|
|
# {"start_date": "2021-11-01", "count": 98},
|
|
# {"start_date": "2021-12-01", "count": 54},
|
|
# {"start_date": "2022-01-01", "count": 88},
|
|
# {"start_date": "2022-02-01", "count": 45},
|
|
# {"start_date": "2022-03-01", "count": 6},
|
|
# {"start_date": "2022-04-01", "count": 54},
|
|
# {"start_date": "2022-05-01", "count": 105},
|
|
# {"start_date": "2022-06-01", "count": 37},
|
|
# {"start_date": "2022-07-01", "count": 88},
|
|
# {"start_date": "2022-08-01", "count": 7},
|
|
# {"start_date": "2022-09-01", "count": 9},
|
|
# {"start_date": "2022-10-01", "count": 6},
|
|
# {"start_date": "2022-11-01", "count": 2},
|
|
# {"start_date": "2022-12-01", "count": 16},
|
|
# {"start_date": "2023-01-01", "count": 7},
|
|
# {"start_date": "2023-04-01", "count": 16},
|
|
# {"start_date": "2023-05-01", "count": 3},
|
|
# {"start_date": "2023-06-01", "count": 8},
|
|
# {"start_date": "2023-07-01", "count": 23},
|
|
# {"start_date": "2023-08-01", "count": 3},
|
|
# {"start_date": "2023-09-01", "count": 1},
|
|
# {"start_date": "2023-10-01", "count": 25},
|
|
# {"start_date": "2023-11-01", "count": 160},
|
|
# {"start_date": "2023-12-01", "count": 306},
|
|
# {"start_date": "2024-01-01", "count": 735},
|
|
# {"start_date": "2024-02-01", "count": 420},
|
|
# {"start_date": "2024-03-01", "count": 549},
|
|
# {"start_date": "2024-04-01", "count": 466},
|
|
# {"start_date": "2024-05-01", "count": 333},
|
|
# {"start_date": "2024-06-01", "count": 1126},
|
|
# {"start_date": "2024-07-01", "count": 534},
|
|
# {"start_date": "2024-08-01", "count": 280},
|
|
# {"start_date": "2024-09-01", "count": 116},
|
|
# {"start_date": "2024-10-01", "count": 120},
|
|
# {"start_date": "2024-11-01", "count": 332},
|
|
# {"start_date": "2024-12-01", "count": 243},
|
|
# {"start_date": "2025-01-01", "count": 122},
|
|
# {"start_date": "2025-02-01", "count": 379},
|
|
# {"start_date": "2025-03-01", "count": 278},
|
|
# {"start_date": "2025-04-01", "count": 296},
|
|
# {"start_date": "2025-05-01", "count": 964},
|
|
# {"start_date": "2025-06-01", "count": 821},
|
|
# {"start_date": "2025-07-01", "count": 230},
|
|
# ],
|
|
)
|
|
|
|
db.add(user)
|
|
db.commit()
|
|
db.refresh(user)
|
|
|
|
# 确保用户ID存在
|
|
if user.id is None:
|
|
raise ValueError("User ID is None after saving to database")
|
|
|
|
# 创建 osu! 模式统计
|
|
osu_stats = LazerUserStatistics(
|
|
user_id=user.id,
|
|
mode="osu",
|
|
count_100=276274,
|
|
count_300=1932068,
|
|
count_50=32776,
|
|
count_miss=111064,
|
|
level_current=97,
|
|
level_progress=96,
|
|
global_rank=298026,
|
|
country_rank=11221,
|
|
pp=2826.26,
|
|
ranked_score=4415081049,
|
|
hit_accuracy=95.7168,
|
|
play_count=12711,
|
|
play_time=836529,
|
|
total_score=12390140370,
|
|
total_hits=2241118,
|
|
maximum_combo=1859,
|
|
replays_watched_by_others=0,
|
|
is_ranked=True,
|
|
grade_ss=14,
|
|
grade_ssh=3,
|
|
grade_s=322,
|
|
grade_sh=11,
|
|
grade_a=757,
|
|
rank_highest=295701,
|
|
rank_highest_updated_at=datetime(2025, 7, 2, 17, 30, 21),
|
|
)
|
|
|
|
# 创建 taiko 模式统计
|
|
taiko_stats = LazerUserStatistics(
|
|
user_id=user.id,
|
|
mode="taiko",
|
|
count_100=160,
|
|
count_300=154,
|
|
count_50=0,
|
|
count_miss=480,
|
|
level_current=2,
|
|
level_progress=49,
|
|
global_rank=None,
|
|
pp=0,
|
|
ranked_score=0,
|
|
hit_accuracy=0,
|
|
play_count=6,
|
|
play_time=217,
|
|
total_score=79301,
|
|
total_hits=314,
|
|
maximum_combo=0,
|
|
replays_watched_by_others=0,
|
|
is_ranked=False,
|
|
)
|
|
|
|
# 创建 fruits 模式统计
|
|
fruits_stats = LazerUserStatistics(
|
|
user_id=user.id,
|
|
mode="fruits",
|
|
count_100=109,
|
|
count_300=1613,
|
|
count_50=1861,
|
|
count_miss=328,
|
|
level_current=6,
|
|
level_progress=14,
|
|
global_rank=None,
|
|
pp=0,
|
|
ranked_score=343854,
|
|
hit_accuracy=89.4779,
|
|
play_count=19,
|
|
play_time=669,
|
|
total_score=1362651,
|
|
total_hits=3583,
|
|
maximum_combo=75,
|
|
replays_watched_by_others=0,
|
|
is_ranked=False,
|
|
grade_a=1,
|
|
)
|
|
|
|
# 创建 mania 模式统计
|
|
mania_stats = LazerUserStatistics(
|
|
user_id=user.id,
|
|
mode="mania",
|
|
count_100=7867,
|
|
count_300=12104,
|
|
count_50=991,
|
|
count_miss=2951,
|
|
level_current=12,
|
|
level_progress=89,
|
|
global_rank=660670,
|
|
pp=25.3784,
|
|
ranked_score=3812295,
|
|
hit_accuracy=77.9316,
|
|
play_count=85,
|
|
play_time=4834,
|
|
total_score=13454470,
|
|
total_hits=20962,
|
|
maximum_combo=573,
|
|
replays_watched_by_others=0,
|
|
is_ranked=True,
|
|
grade_a=1,
|
|
)
|
|
|
|
db.add_all([osu_stats, taiko_stats, fruits_stats, mania_stats])
|
|
|
|
# 创建每日挑战统计
|
|
daily_challenge = DailyChallengeStats(
|
|
user_id=user.id,
|
|
daily_streak_best=1,
|
|
daily_streak_current=0,
|
|
last_update=datetime(2025, 6, 21, 0, 0, 0),
|
|
last_weekly_streak=datetime(2025, 6, 19, 0, 0, 0),
|
|
playcount=1,
|
|
top_10p_placements=0,
|
|
top_50p_placements=0,
|
|
weekly_streak_best=1,
|
|
weekly_streak_current=0,
|
|
)
|
|
|
|
db.add(daily_challenge)
|
|
|
|
# 创建排名历史 (最近90天的数据)
|
|
rank_data = [
|
|
322806,
|
|
323092,
|
|
323341,
|
|
323616,
|
|
323853,
|
|
324106,
|
|
324378,
|
|
324676,
|
|
324958,
|
|
325254,
|
|
325492,
|
|
325780,
|
|
326075,
|
|
326356,
|
|
326586,
|
|
326845,
|
|
327067,
|
|
327286,
|
|
327526,
|
|
327778,
|
|
328039,
|
|
328347,
|
|
328631,
|
|
328858,
|
|
329323,
|
|
329557,
|
|
329809,
|
|
329911,
|
|
330188,
|
|
330425,
|
|
330650,
|
|
330881,
|
|
331068,
|
|
331325,
|
|
331575,
|
|
331816,
|
|
332061,
|
|
328959,
|
|
315648,
|
|
315881,
|
|
308784,
|
|
309023,
|
|
309252,
|
|
309433,
|
|
309537,
|
|
309364,
|
|
309548,
|
|
308957,
|
|
309182,
|
|
309426,
|
|
309607,
|
|
309831,
|
|
310054,
|
|
310269,
|
|
310485,
|
|
310714,
|
|
310956,
|
|
310924,
|
|
311125,
|
|
311203,
|
|
311422,
|
|
311640,
|
|
303091,
|
|
303309,
|
|
303500,
|
|
303691,
|
|
303758,
|
|
303750,
|
|
303957,
|
|
299867,
|
|
300088,
|
|
300273,
|
|
300457,
|
|
295799,
|
|
295976,
|
|
296153,
|
|
296350,
|
|
296566,
|
|
296756,
|
|
296933,
|
|
297141,
|
|
297314,
|
|
297480,
|
|
297114,
|
|
297296,
|
|
297480,
|
|
297645,
|
|
297815,
|
|
297993,
|
|
298026,
|
|
]
|
|
|
|
rank_history = RankHistory(user_id=user.id, mode="osu", rank_data=rank_data)
|
|
|
|
db.add(rank_history)
|
|
|
|
# 创建一些成就
|
|
achievements = [
|
|
LazerUserAchievement(
|
|
user_id=user.id,
|
|
achievement_id=336,
|
|
achieved_at=datetime(2025, 6, 21, 19, 6, 32),
|
|
),
|
|
LazerUserAchievement(
|
|
user_id=user.id,
|
|
achievement_id=319,
|
|
achieved_at=datetime(2025, 6, 1, 0, 52, 0),
|
|
),
|
|
LazerUserAchievement(
|
|
user_id=user.id,
|
|
achievement_id=222,
|
|
achieved_at=datetime(2025, 5, 28, 12, 24, 37),
|
|
),
|
|
LazerUserAchievement(
|
|
user_id=user.id,
|
|
achievement_id=38,
|
|
achieved_at=datetime(2024, 7, 5, 15, 43, 23),
|
|
),
|
|
LazerUserAchievement(
|
|
user_id=user.id,
|
|
achievement_id=67,
|
|
achieved_at=datetime(2024, 6, 24, 5, 6, 44),
|
|
),
|
|
]
|
|
|
|
db.add_all(achievements)
|
|
|
|
db.commit()
|
|
print(f"成功创建示例用户: {user.name} (ID: {user.id})")
|
|
print(f"安全用户名: {user.safe_name}")
|
|
print(f"邮箱: {user.email}")
|
|
print(f"国家: {user.country}")
|
|
return user
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("开始创建示例数据...")
|
|
user = create_sample_user()
|
|
print("示例数据创建完成！")
|
|
print(f"用户名: {user.name}")
|
|
print("密码: password123")
|
|
print("现在您可以使用这些凭据来测试API了。")
|