mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2025-12-14 08:06:23 +08:00
ba36190f30
- Fix a small bug that `best30` of API cannot have scores whose songs are not in database - At present the setting file can be a module or a file with some of options - Limiter can have multiple rules together now
91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
import os
|
|
from shutil import copy, copy2
|
|
from core.sql import Connect
|
|
from database.database_initialize import main, ARCAEA_SERVER_VERSION
|
|
from web.system import update_database
|
|
|
|
|
|
def try_rename(path, new_path):
|
|
# 尝试重命名文件，并尝试避免命名冲突，返回最终路径
|
|
final_path = new_path
|
|
if os.path.exists(new_path):
|
|
i = 1
|
|
while os.path.exists(new_path + str(i)):
|
|
i += 1
|
|
os.rename(path, new_path + str(i))
|
|
final_path = new_path + str(i)
|
|
else:
|
|
os.rename(path, new_path)
|
|
|
|
return final_path
|
|
|
|
|
|
def check_before_run(app):
|
|
# 运行前检查关键文件，返回布尔值，其实是因为有人经常忘了
|
|
|
|
f = True
|
|
|
|
if not os.path.exists('database'):
|
|
app.logger.warning('Folder `database` is missing.')
|
|
f = False
|
|
|
|
if not os.path.exists('database/songs'):
|
|
app.logger.warning('Folder `database/songs` is missing.')
|
|
f = False
|
|
|
|
if not os.path.exists('database/arcaea_database.db'):
|
|
app.logger.warning('File `database/arcaea_database.db` is missing.')
|
|
f = False
|
|
try:
|
|
app.logger.info(
|
|
'Try to new the file `database/arcaea_database.db`.')
|
|
main('./database/')
|
|
app.logger.info(
|
|
'Success to new the file `database/arcaea_database.db`.')
|
|
f = True
|
|
except:
|
|
app.logger.warning(
|
|
'Fail to new the file `database/arcaea_database.db`.')
|
|
|
|
else:
|
|
with Connect() as c:
|
|
try:
|
|
c.execute('''select value from config where id="version"''')
|
|
x = c.fetchone()
|
|
except:
|
|
x = None
|
|
|
|
# 数据库自动更新，不强求
|
|
if not x or x[0] != ARCAEA_SERVER_VERSION:
|
|
app.logger.warning(
|
|
'Maybe the file `database/arcaea_database.db` is an old version.')
|
|
try:
|
|
app.logger.info(
|
|
'Try to update the file `database/arcaea_database.db`.')
|
|
|
|
path = try_rename('database/arcaea_database.db',
|
|
'database/arcaea_database.db.bak')
|
|
|
|
try:
|
|
copy2(path, 'database/arcaea_database.db')
|
|
except:
|
|
copy(path, 'database/arcaea_database.db')
|
|
|
|
if os.path.isfile("database/old_arcaea_database.db"):
|
|
os.remove('database/old_arcaea_database.db')
|
|
|
|
try_rename('database/arcaea_database.db',
|
|
'database/old_arcaea_database.db')
|
|
|
|
main('./database/')
|
|
update_database()
|
|
|
|
app.logger.info(
|
|
'Success to update the file `database/arcaea_database.db`.')
|
|
|
|
except:
|
|
app.logger.warning(
|
|
'Fail to update the file `database/arcaea_database.db`.')
|
|
|
|
return f
|