mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-06 15:37:26 +08:00
361 lines
11 KiB
Python
361 lines
11 KiB
Python
from flask import (
|
||
Blueprint, flash, g, redirect, render_template, request, url_for
|
||
)
|
||
from web.login import login_required
|
||
from werkzeug.utils import secure_filename
|
||
import sqlite3
|
||
import web.webscore
|
||
import web.system
|
||
import time
|
||
import server.arcscore
|
||
import os
|
||
|
||
UPLOAD_FOLDER = 'database'
|
||
ALLOWED_EXTENSIONS = {'db'}
|
||
|
||
bp = Blueprint('index', __name__, url_prefix='/web')
|
||
|
||
|
||
def is_number(s):
|
||
try: # 判断字符串s是浮点数
|
||
float(s)
|
||
return True
|
||
except ValueError:
|
||
pass
|
||
return False
|
||
|
||
|
||
@bp.route('/index')
|
||
@bp.route('/')
|
||
@login_required
|
||
def index():
|
||
# 主页
|
||
return render_template('web/index.html')
|
||
|
||
|
||
@bp.route('/singleplayer', methods=['POST', 'GET'])
|
||
@login_required
|
||
def single_player_score():
|
||
# 单个玩家分数查询
|
||
if request.method == 'POST':
|
||
name = request.form['name']
|
||
user_code = request.form['user_code']
|
||
error = None
|
||
if name or user_code:
|
||
conn = sqlite3.connect('./database/arcaea_database.db')
|
||
c = conn.cursor()
|
||
if user_code:
|
||
c.execute('''select user_id from user where user_code=:a''', {
|
||
'a': user_code})
|
||
else:
|
||
c.execute(
|
||
'''select user_id from user where name=:a''', {'a': name})
|
||
|
||
user_id = c.fetchone()
|
||
posts = []
|
||
if user_id:
|
||
user_id = user_id[0]
|
||
posts = web.webscore.get_user_score(c, user_id)
|
||
if not posts:
|
||
error = '无成绩 No score.'
|
||
else:
|
||
error = '玩家不存在 The player does not exist.'
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
else:
|
||
error = '输入为空 Null Input.'
|
||
|
||
if error:
|
||
flash(error)
|
||
else:
|
||
return render_template('web/singleplayer.html', posts=posts)
|
||
|
||
return render_template('web/singleplayer.html')
|
||
|
||
|
||
@bp.route('/singleplayerptt', methods=['POST', 'GET'])
|
||
@login_required
|
||
def single_player_ptt():
|
||
# 单个玩家PTT详情查询
|
||
if request.method == 'POST':
|
||
name = request.form['name']
|
||
user_code = request.form['user_code']
|
||
error = None
|
||
if name or user_code:
|
||
conn = sqlite3.connect('./database/arcaea_database.db')
|
||
c = conn.cursor()
|
||
if user_code:
|
||
c.execute('''select user_id from user where user_code=:a''', {
|
||
'a': user_code})
|
||
else:
|
||
c.execute(
|
||
'''select user_id from user where name=:a''', {'a': name})
|
||
|
||
user_id = c.fetchone()
|
||
posts = []
|
||
if user_id:
|
||
user_id = user_id[0]
|
||
user = web.webscore.get_user(c, user_id)
|
||
posts = web.webscore.get_user_score(c, user_id, 30)
|
||
recent, recentptt = web.webscore.get_user_recent30(c, user_id)
|
||
if not posts:
|
||
error = '无成绩 No score.'
|
||
else:
|
||
bestptt = 0
|
||
for i in posts:
|
||
if i['rating']:
|
||
bestptt += i['rating']
|
||
bestptt = bestptt / 30
|
||
else:
|
||
error = '玩家不存在 The player does not exist.'
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
else:
|
||
error = '输入为空 Null Input.'
|
||
|
||
if error:
|
||
flash(error)
|
||
else:
|
||
return render_template('web/singleplayerptt.html', posts=posts, user=user, recent=recent, recentptt=recentptt, bestptt=bestptt)
|
||
|
||
return render_template('web/singleplayerptt.html')
|
||
|
||
|
||
@bp.route('/allplayer', methods=['GET'])
|
||
@login_required
|
||
def all_player():
|
||
# 所有玩家数据，按照ptt排序
|
||
conn = sqlite3.connect('./database/arcaea_database.db')
|
||
c = conn.cursor()
|
||
c.execute('''select * from user order by rating_ptt DESC''')
|
||
x = c.fetchall()
|
||
error = None
|
||
if x:
|
||
posts = []
|
||
for i in x:
|
||
join_data = None
|
||
time_played = None
|
||
if i[3]:
|
||
join_date = time.strftime('%Y-%m-%d %H:%M:%S',
|
||
time.localtime(int(i[3])//1000))
|
||
if i[20]:
|
||
time_played = time.strftime('%Y-%m-%d %H:%M:%S',
|
||
time.localtime(int(i[20])//1000))
|
||
posts.append({'name': i[1],
|
||
'user_id': i[0],
|
||
'join_date': join_date,
|
||
'user_code': i[4],
|
||
'rating_ptt': i[5],
|
||
'song_id': i[11],
|
||
'difficulty': i[12],
|
||
'score': i[13],
|
||
'shiny_perfect_count': i[14],
|
||
'perfect_count': i[15],
|
||
'near_count': i[16],
|
||
'miss_count': i[17],
|
||
'time_played': time_played,
|
||
'clear_type': i[21],
|
||
'rating': i[22]
|
||
})
|
||
else:
|
||
error = '没有玩家数据 No player data.'
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
if error:
|
||
flash(error)
|
||
return render_template('web/allplayer.html')
|
||
else:
|
||
return render_template('web/allplayer.html', posts=posts)
|
||
|
||
|
||
@bp.route('/allsong', methods=['GET'])
|
||
@login_required
|
||
def all_song():
|
||
# 所有歌曲数据
|
||
def defnum(x):
|
||
# 定数转换
|
||
if x >= 0:
|
||
return x / 10
|
||
else:
|
||
return None
|
||
|
||
conn = sqlite3.connect('./database/arcsong.db')
|
||
c = conn.cursor()
|
||
c.execute('''select * from songs''')
|
||
x = c.fetchall()
|
||
error = None
|
||
if x:
|
||
posts = []
|
||
for i in x:
|
||
posts.append({'song_id': i[0],
|
||
'name_en': i[1],
|
||
'rating_pst': defnum(i[12]),
|
||
'rating_prs': defnum(i[13]),
|
||
'rating_ftr': defnum(i[14]),
|
||
'rating_byn': defnum(i[15])
|
||
})
|
||
else:
|
||
error = '没有铺面数据 No song data.'
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
if error:
|
||
flash(error)
|
||
return render_template('web/allsong.html')
|
||
else:
|
||
return render_template('web/allsong.html', posts=posts)
|
||
|
||
|
||
@bp.route('/singlecharttop', methods=['GET', 'POST'])
|
||
@login_required
|
||
def single_chart_top():
|
||
# 歌曲排行榜
|
||
if request.method == 'POST':
|
||
song_name = request.form['sid']
|
||
difficulty = request.form['difficulty']
|
||
if difficulty.isdigit():
|
||
difficulty = int(difficulty)
|
||
error = None
|
||
conn = sqlite3.connect('./database/arcsong.db')
|
||
c = conn.cursor()
|
||
song_name = '%'+song_name+'%'
|
||
c.execute('''select sid, name_en from songs where sid like :a limit 1''',
|
||
{'a': song_name})
|
||
x = c.fetchone()
|
||
conn.commit()
|
||
conn.close()
|
||
print(x)
|
||
if x:
|
||
song_id = x[0]
|
||
posts = server.arcscore.arc_score_top(song_id, difficulty, -1)
|
||
for i in posts:
|
||
i['time_played'] = time.strftime('%Y-%m-%d %H:%M:%S',
|
||
time.localtime(i['time_played']))
|
||
else:
|
||
error = '查询为空 No song.'
|
||
|
||
if not error:
|
||
return render_template('web/singlecharttop.html', posts=posts, song_name_en=x[1], song_id=song_id, difficulty=difficulty)
|
||
else:
|
||
flash(error)
|
||
|
||
return render_template('web/singlecharttop.html')
|
||
|
||
|
||
def allowed_file(filename):
|
||
return '.' in filename and \
|
||
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
|
||
|
||
|
||
@bp.route('/updatedatabase', methods=['GET', 'POST'])
|
||
@login_required
|
||
def update_database():
|
||
# 更新数据库
|
||
error = None
|
||
if request.method == 'POST':
|
||
if 'file' not in request.files:
|
||
flash('无文件 No file part.')
|
||
return redirect(request.url)
|
||
file = request.files['file']
|
||
if file.filename == '':
|
||
flash('未选择文件 No selected file.')
|
||
return redirect(request.url)
|
||
|
||
if file and allowed_file(file.filename) and file.filename in ['arcsong.db', 'arcaea_database.db']:
|
||
filename = 'old_' + secure_filename(file.filename)
|
||
file.save(os.path.join(UPLOAD_FOLDER, filename))
|
||
flash('上传成功 Success upload.')
|
||
try:
|
||
web.system.update_database()
|
||
flash('数据更新成功 Success update data.')
|
||
except:
|
||
flash('数据更新失败 Cannot update data.')
|
||
else:
|
||
error = '上传失败 Upload error.'
|
||
|
||
if error:
|
||
flash(error)
|
||
|
||
return render_template('web/updatedatabase.html')
|
||
|
||
|
||
@bp.route('/changesong', methods=['GET'])
|
||
@login_required
|
||
def change_song():
|
||
# 修改歌曲数据
|
||
return render_template('web/changesong.html')
|
||
|
||
|
||
@bp.route('/changesong/addsong', methods=['POST'])
|
||
@login_required
|
||
def add_song():
|
||
# 添加歌曲数据
|
||
def get_rating(x):
|
||
# 换算定数
|
||
if is_number(x):
|
||
x = float(x)
|
||
if x >= 0:
|
||
return int(x*10)
|
||
else:
|
||
return -1
|
||
else:
|
||
return -1
|
||
|
||
error = None
|
||
song_id = request.form['sid']
|
||
name_en = request.form['name_en']
|
||
rating_pst = get_rating(request.form['rating_pst'])
|
||
rating_prs = get_rating(request.form['rating_prs'])
|
||
rating_ftr = get_rating(request.form['rating_ftr'])
|
||
rating_byd = get_rating(request.form['rating_byd'])
|
||
if len(song_id) >= 256:
|
||
song_id = song_id[:200]
|
||
if len(name_en) >= 256:
|
||
name_en = name_en[:200]
|
||
conn = sqlite3.connect('./database/arcsong.db')
|
||
c = conn.cursor()
|
||
c.execute(
|
||
'''select exists(select * from songs where sid=:a)''', {'a': song_id})
|
||
if c.fetchone() == (0,):
|
||
c.execute('''insert into songs(sid,name_en,rating_pst,rating_prs,rating_ftr,rating_byn) values(:a,:b,:c,:d,:e,:f)''', {
|
||
'a': song_id, 'b': name_en, 'c': rating_pst, 'd': rating_prs, 'e': rating_ftr, 'f': rating_byd})
|
||
flash('歌曲添加成功 Successfully add the song.')
|
||
else:
|
||
error = '歌曲已存在 The song exists.'
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
if error:
|
||
flash(error)
|
||
|
||
return redirect(url_for('index.change_song'))
|
||
|
||
|
||
@bp.route('/changesong/deletesong', methods=['POST'])
|
||
@login_required
|
||
def delete_song():
|
||
# 删除歌曲数据
|
||
|
||
error = None
|
||
song_id = request.form['sid']
|
||
conn = sqlite3.connect('./database/arcsong.db')
|
||
c = conn.cursor()
|
||
c.execute(
|
||
'''select exists(select * from songs where sid=:a)''', {'a': song_id})
|
||
if c.fetchone() == (1,):
|
||
c.execute('''delete from songs where sid=:a''', {'a': song_id})
|
||
flash('歌曲删除成功 Successfully delete the song.')
|
||
else:
|
||
error = "歌曲不存在 The song doesn't exist."
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
if error:
|
||
flash(error)
|
||
|
||
return redirect(url_for('index.change_song'))
|