mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-04 21:47:28 +08:00
260 lines
8.8 KiB
Python
260 lines
8.8 KiB
Python
from server.sql import Connect
|
||
import server.arcworld
|
||
import server.arcpurchase
|
||
import server.arcdownload
|
||
import server.character
|
||
import server.item
|
||
import time
|
||
from setting import Config
|
||
|
||
MAX_STAMINA = 12
|
||
STAMINA_RECOVER_TICK = 1800000
|
||
CORE_EXP = 250
|
||
|
||
|
||
def int2b(x):
|
||
# int与布尔值转换
|
||
if x is None or x == 0:
|
||
return False
|
||
else:
|
||
return True
|
||
|
||
|
||
def get_recent_score(c, user_id):
|
||
# 得到用户最近一次的成绩，返回列表
|
||
c.execute('''select * from user where user_id = :x''', {'x': user_id})
|
||
x = c.fetchone()
|
||
if x is not None:
|
||
if x[11] is not None:
|
||
c.execute('''select best_clear_type from best_score where user_id=:u and song_id=:s and difficulty=:d''', {
|
||
'u': user_id, 's': x[11], 'd': x[12]})
|
||
y = c.fetchone()
|
||
if y is not None:
|
||
best_clear_type = y[0]
|
||
else:
|
||
best_clear_type = x[21]
|
||
|
||
return [{
|
||
"rating": x[22],
|
||
"modifier": x[19],
|
||
"time_played": x[20],
|
||
"health": x[18],
|
||
"best_clear_type": best_clear_type,
|
||
"clear_type": x[21],
|
||
"miss_count": x[17],
|
||
"near_count": x[16],
|
||
"perfect_count": x[15],
|
||
"shiny_perfect_count": x[14],
|
||
"score": x[13],
|
||
"difficulty": x[12],
|
||
"song_id": x[11]
|
||
}]
|
||
return []
|
||
|
||
|
||
def get_user_friend(c, user_id):
|
||
# 得到用户的朋友列表，返回列表
|
||
c.execute('''select user_id_other from friend where user_id_me = :user_id''', {
|
||
'user_id': user_id})
|
||
x = c.fetchall()
|
||
s = []
|
||
if x != [] and x[0][0] is not None:
|
||
|
||
for i in x:
|
||
c.execute('''select exists(select * from friend where user_id_me = :x and user_id_other = :y)''',
|
||
{'x': i[0], 'y': user_id})
|
||
if c.fetchone() == (1,):
|
||
is_mutual = True
|
||
else:
|
||
is_mutual = False
|
||
|
||
c.execute('''select * from user where user_id = :x''', {'x': i[0]})
|
||
y = c.fetchone()
|
||
if y is not None:
|
||
character = y[6]
|
||
is_char_uncapped = int2b(y[8])
|
||
is_char_uncapped_override = int2b(y[9])
|
||
if y[23] != -1:
|
||
character = y[23]
|
||
if not Config.CHARACTER_FULL_UNLOCK:
|
||
c.execute('''select is_uncapped, is_uncapped_override from user_char where user_id=:a and character_id=:b''', {
|
||
'a': i[0], 'b': character})
|
||
else:
|
||
c.execute('''select is_uncapped, is_uncapped_override from user_char_full where user_id=:a and character_id=:b''', {
|
||
'a': i[0], 'b': character})
|
||
z = c.fetchone()
|
||
if z:
|
||
is_char_uncapped = int2b(z[0])
|
||
is_char_uncapped_override = int2b(z[1])
|
||
|
||
rating = y[5]
|
||
if int2b(y[10]):
|
||
rating = -1
|
||
|
||
s.append({
|
||
"is_mutual": is_mutual,
|
||
"is_char_uncapped_override": is_char_uncapped_override,
|
||
"is_char_uncapped": is_char_uncapped,
|
||
"is_skill_sealed": int2b(y[7]),
|
||
"rating": rating,
|
||
"join_date": int(y[3]),
|
||
"character": character,
|
||
"recent_score": get_recent_score(c, i[0]),
|
||
"name": y[1],
|
||
"user_id": i[0]
|
||
})
|
||
|
||
s.sort(key=lambda item: item["recent_score"][0]["time_played"] if len(
|
||
item["recent_score"]) > 0 else 0, reverse=True)
|
||
return s
|
||
|
||
|
||
def get_user_me(c, user_id):
|
||
# 构造user/me的数据，返回字典
|
||
c.execute('''select * from user where user_id = :x''', {'x': user_id})
|
||
x = c.fetchone()
|
||
r = {}
|
||
if x is not None:
|
||
user_character = server.character.get_user_character(c, user_id)
|
||
# 下面没有使用get_user_characters函数是为了节省一次查询
|
||
characters = []
|
||
for i in user_character:
|
||
characters.append(i['character_id'])
|
||
|
||
character_id = x[6]
|
||
if character_id not in characters:
|
||
character_id = 0
|
||
c.execute(
|
||
'''update user set character_id=0 where user_id=?''', (user_id,))
|
||
|
||
favorite_character_id = x[23]
|
||
if favorite_character_id not in characters: # 这是考虑有可能favourite_character设置了用户未拥有的角色
|
||
favorite_character_id = -1
|
||
c.execute(
|
||
'''update user set favorite_character=-1 where user_id=?''', (user_id,))
|
||
|
||
# 计算世界排名
|
||
c.execute(
|
||
'''select world_rank_score from user where user_id=?''', (user_id,))
|
||
y = c.fetchone()
|
||
if y and y[0]:
|
||
c.execute(
|
||
'''select count(*) from user where world_rank_score > ?''', (y[0],))
|
||
y = c.fetchone()
|
||
if y and y[0] + 1 <= Config.WORLD_RANK_MAX:
|
||
world_rank = y[0] + 1
|
||
else:
|
||
world_rank = 0
|
||
else:
|
||
world_rank = 0
|
||
|
||
# 源点强化
|
||
prog_boost = 0
|
||
if x[27] and x[27] != 0:
|
||
prog_boost = 300
|
||
|
||
# 体力计算
|
||
next_fragstam_ts = -1
|
||
max_stamina_ts = 1586274871917
|
||
stamina = 12
|
||
if x[31]:
|
||
next_fragstam_ts = x[31]
|
||
if x[32]:
|
||
max_stamina_ts = x[32]
|
||
if x[33]:
|
||
stamina = x[33]
|
||
|
||
r = {"is_aprilfools": Config.IS_APRILFOOLS,
|
||
"curr_available_maps": server.arcworld.get_available_maps(),
|
||
"character_stats": user_character,
|
||
"friends": get_user_friend(c, user_id),
|
||
"settings": {
|
||
"favorite_character": favorite_character_id,
|
||
"is_hide_rating": int2b(x[10]),
|
||
"max_stamina_notification_enabled": int2b(x[24])
|
||
},
|
||
"user_id": user_id,
|
||
"name": x[1],
|
||
"user_code": x[4],
|
||
"display_name": x[1],
|
||
"ticket": x[26],
|
||
"character": character_id,
|
||
"is_locked_name_duplicate": False,
|
||
"is_skill_sealed": int2b(x[7]),
|
||
"current_map": x[25],
|
||
"prog_boost": prog_boost,
|
||
"next_fragstam_ts": next_fragstam_ts,
|
||
"max_stamina_ts": max_stamina_ts,
|
||
"stamina": server.arcworld.calc_stamina(max_stamina_ts, stamina),
|
||
"world_unlocks": server.item.get_user_items(c, user_id, 'world_unlock'),
|
||
"world_songs": server.item.get_user_items(c, user_id, 'world_song'),
|
||
"singles": server.item.get_user_items(c, user_id, 'single'),
|
||
"packs": server.item.get_user_items(c, user_id, 'pack'),
|
||
"characters": characters,
|
||
"cores": server.item.get_user_cores(c, user_id),
|
||
"recent_score": get_recent_score(c, user_id),
|
||
"max_friend": 50,
|
||
"rating": x[5],
|
||
"join_date": int(x[3]),
|
||
"global_rank": world_rank
|
||
}
|
||
|
||
return r
|
||
|
||
|
||
def arc_aggregate_small(user_id):
|
||
# 返回用户数据
|
||
r = {"success": False}
|
||
with Connect() as c:
|
||
r = {"success": True,
|
||
"value": [{
|
||
"id": 0,
|
||
"value": get_user_me(c, user_id)
|
||
}]}
|
||
|
||
return r
|
||
|
||
|
||
def arc_aggregate_big(user_id):
|
||
# 返回比较全的用户数据
|
||
r = {"success": False}
|
||
with Connect() as c:
|
||
# 防止数据库锁
|
||
id_2 = server.arcdownload.get_all_songs(user_id, url_flag=False)
|
||
id_5 = {
|
||
"current_map": server.arcworld.get_current_map(user_id),
|
||
"user_id": user_id,
|
||
"maps": server.arcworld.get_world_all(user_id)
|
||
}
|
||
r = {"success": True,
|
||
"value": [{
|
||
"id": 0,
|
||
"value": get_user_me(c, user_id)
|
||
}, {
|
||
"id": 1,
|
||
"value": server.arcpurchase.get_purchase(c, 'pack')
|
||
}, {
|
||
"id": 2,
|
||
"value": id_2
|
||
}, {
|
||
"id": 3,
|
||
"value": {
|
||
"max_stamina": MAX_STAMINA,
|
||
"stamina_recover_tick": STAMINA_RECOVER_TICK,
|
||
"core_exp": CORE_EXP,
|
||
"curr_ts": int(time.time()*1000),
|
||
"level_steps": server.character.get_level_steps(),
|
||
"world_ranking_enabled": True,
|
||
"is_byd_chapter_unlocked": True
|
||
}
|
||
}, {
|
||
"id": 4,
|
||
"value": server.arcpurchase.get_user_present(c, user_id)
|
||
}, {
|
||
"id": 5,
|
||
"value": id_5
|
||
}
|
||
]}
|
||
|
||
return r
|