eorzea-songbook/build_db.py
2023-05-23 12:50:44 +10:00

68 lines
1.6 KiB
Python

import sqlite3
import os
import pyexcel_odsr
os.remove('songbook.sqlite')
db = sqlite3.connect("songbook.sqlite")
cur = db.cursor()
with open('db.sql') as file:
db.executescript(file.read())
data = pyexcel_odsr.get_data('reference.ods')
# import song
cur.executemany(
'insert or ignore into song(id,name,name_jp) values(?, ?, ?)',
((row[0],row[3],row[4]) for row in data['Song'][1:])
)
# import album
cur.executemany(
'insert into album(id,name,date,code) values (?, ?, ?, ?)',
data['Album'][1:]
)
# import artist
cur.executemany(
'insert into artist(id,name_rm,name) values(?, ?, ?)',
((row[0],row[1],row[2]) for row in data['Artist'][1:])
)
# import credit
cur.executemany(
'insert into credit(id,name) values(?, ?)',
data['Credit'][1:]
)
# import motif
cur.executemany(
'insert into motif(id,name) values(?, ?)',
data['Motif'][1:]
)
# import clip
cur.executemany(
'insert into clip(start_ms,duration_ms,song_id,motif_id) values (?, ?, ?, ?)',
data['Clip'][1:]
)
# import song x artist
def creditparams(rows):
for row in rows[1:]:
for index,artist in enumerate(row[5:], 5):
if len(artist)>0:
yield (row[0],artist,rows[0][index])
cur.executemany(
'insert or ignore into song_artist(song_id,artist_id,credit_id) values (?, (select id from artist where coalesce(name_rm,name) = ?), (select id from credit where name = ?))',
creditparams(data['Song'])
)
cur.executemany(
"insert into song_album(song_id,album_id,track) values (?, ?, ?)",
((row[0],row[1],row[2]) for row in data['Song'][1:])
)
db.commit()