24 lines
584 B
Python
24 lines
584 B
Python
# -*- coding: utf-8 -*-
|
|
import sqlite3
|
|
import os
|
|
|
|
DATA = 'data'
|
|
MIGRATE = os.path.join(DATA, 'migrate')
|
|
|
|
db = sqlite3.connect(os.path.join(DATA, 'nxy.db'))
|
|
|
|
|
|
def read_file(file):
|
|
with open(os.path.join(MIGRATE, file), 'r') as fp:
|
|
return [(l.replace('\n', ''),) for l in fp]
|
|
|
|
|
|
def batch_insert(table, column, data):
|
|
db.executemany('insert into {table} ({column}) values (?)'.
|
|
format(table=table, column=column), data)
|
|
db.commit()
|
|
|
|
|
|
batch_insert('mcmaniacs', 'item', read_file('mcmaniac.txt'))
|
|
batch_insert('yiffs', 'item', read_file('yiff.txt'))
|