mirror of
https://github.com/jiojciojsioe3/a3cjroijsiojiorj.git
synced 2024-11-15 07:21:50 +08:00
115 lines
3.5 KiB
Python
115 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
# Migrate old SQLite taiko.db to MongoDB
|
|
|
|
import sqlite3
|
|
from pymongo import MongoClient
|
|
|
|
import os,sys,inspect
|
|
current_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
|
|
parent_dir = os.path.dirname(current_dir)
|
|
sys.path.insert(0, parent_dir)
|
|
import config
|
|
|
|
client = MongoClient(config.MONGO['host'])
|
|
client.drop_database(config.MONGO['database'])
|
|
db = client[config.MONGO['database']]
|
|
sqdb = sqlite3.connect('taiko.db')
|
|
sqdb.row_factory = sqlite3.Row
|
|
curs = sqdb.cursor()
|
|
|
|
def migrate_songs():
|
|
curs.execute('select * from songs order by id')
|
|
rows = curs.fetchall()
|
|
|
|
for row in rows:
|
|
song = {
|
|
'id': row['id'],
|
|
'title': row['title'],
|
|
'title_lang': {'ja': row['title'], 'en': None, 'cn': None, 'tw': None, 'ko': None},
|
|
'subtitle': row['subtitle'],
|
|
'subtitle_lang': {'ja': row['subtitle'], 'en': None, 'cn': None, 'tw': None, 'ko': None},
|
|
'courses': {'easy': None, 'normal': None, 'hard': None, 'oni': None, 'ura': None},
|
|
'enabled': True if row['enabled'] else False,
|
|
'category_id': row['category'],
|
|
'type': row['type'],
|
|
'offset': row['offset'] or 0,
|
|
'skin_id': row['skin_id'],
|
|
'preview': row['preview'] or 0,
|
|
'volume': row['volume'] or 1.0,
|
|
'maker_id': row['maker_id'],
|
|
'hash': row['hash'],
|
|
'order': row['id']
|
|
}
|
|
|
|
for diff in ['easy', 'normal', 'hard', 'oni', 'ura']:
|
|
if row[diff]:
|
|
spl = row[diff].split(' ')
|
|
branch = False
|
|
if len(spl) > 1 and spl[1] == 'B':
|
|
branch = True
|
|
|
|
song['courses'][diff] = {'stars': int(spl[0]), 'branch': branch}
|
|
|
|
if row['title_lang']:
|
|
langs = row['title_lang'].splitlines()
|
|
for lang in langs:
|
|
spl = lang.split(' ', 1)
|
|
if spl[0] in ['ja', 'en', 'cn', 'tw', 'ko']:
|
|
song['title_lang'][spl[0]] = spl[1]
|
|
else:
|
|
song['title_lang']['en'] = lang
|
|
|
|
if row['subtitle_lang']:
|
|
langs = row['subtitle_lang'].splitlines()
|
|
for lang in langs:
|
|
spl = lang.split(' ', 1)
|
|
if spl[0] in ['ja', 'en', 'cn', 'tw', 'ko']:
|
|
song['subtitle_lang'][spl[0]] = spl[1]
|
|
else:
|
|
song['subtitle_lang']['en'] = lang
|
|
|
|
db.songs.insert_one(song)
|
|
last_song = song['id']
|
|
|
|
db.seq.insert_one({'name': 'songs', 'value': last_song})
|
|
|
|
def migrate_makers():
|
|
curs.execute('select * from makers')
|
|
rows = curs.fetchall()
|
|
|
|
for row in rows:
|
|
db.makers.insert_one({
|
|
'id': row['maker_id'],
|
|
'name': row['name'],
|
|
'url': row['url']
|
|
})
|
|
|
|
def migrate_categories():
|
|
curs.execute('select * from categories')
|
|
rows = curs.fetchall()
|
|
|
|
for row in rows:
|
|
db.categories.insert_one({
|
|
'id': row['id'],
|
|
'title': row['title']
|
|
})
|
|
|
|
def migrate_song_skins():
|
|
curs.execute('select * from song_skins')
|
|
rows = curs.fetchall()
|
|
|
|
for row in rows:
|
|
db.song_skins.insert_one({
|
|
'id': row['id'],
|
|
'name': row['name'],
|
|
'song': row['song'],
|
|
'stage': row['stage'],
|
|
'don': row['don']
|
|
})
|
|
|
|
if __name__ == '__main__':
|
|
migrate_songs()
|
|
migrate_makers()
|
|
migrate_categories()
|
|
migrate_song_skins()
|