1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import json
import sqlite3
'''
Convert legacy Python kimikuri-server's users.json to sqlite3 database used by kimikuri_rs.
'''
# noinspection SqlNoDataSourceInspection
def convert(db_file: str = 'kimikuri.db', json_file: str = 'users.json'):
db = sqlite3.connect(db_file)
with db, open(json_file, 'r', encoding='utf-8') as f:
j = json.load(f)
cur = db.cursor()
cur.execute('''CREATE TABLE IF NOT EXISTS "user" (
"id" INTEGER NOT NULL CHECK(id >= 0) UNIQUE,
"name" TEXT,
"token" TEXT NOT NULL UNIQUE,
"chat_id" INTEGER NOT NULL CHECK(chat_id >= 0) UNIQUE,
PRIMARY KEY("id","token","chat_id")
)''')
cur.executemany(
'INSERT INTO user (id, name, token, chat_id) VALUES (?,?,?,?)',
[(x['user_id'], None, x['token'], x['chat_id']) for x in j]
)
db.close()
if __name__ == '__main__':
convert()
|
