27 lines
615 B
JavaScript
27 lines
615 B
JavaScript
|
import { Sequelize, DataTypes, Model } from 'sequelize'
|
||
|
|
import { Data } from '#miao'
|
||
|
|
|
||
|
|
Data.createDir('/data/db', 'root')
|
||
|
|
let dbPath = process.cwd() + '/data/db/data.db'
|
||
|
|
|
||
|
|
// TODO DB自定义
|
||
|
|
const sequelize = new Sequelize({
|
||
|
|
dialect: 'sqlite',
|
||
|
|
storage: dbPath,
|
||
|
|
logging: false
|
||
|
|
})
|
||
|
|
|
||
|
|
await sequelize.authenticate()
|
||
|
|
|
||
|
|
export default class BaseModel extends Model {
|
||
|
|
static Types = DataTypes
|
||
|
|
|
||
|
|
static initDB (model, columns) {
|
||
|
|
let name = model.name
|
||
|
|
console.log('Model Name', name)
|
||
|
|
name = name.replace(/DB$/, 's')
|
||
|
|
model.init(columns, { sequelize, tableName: name })
|
||
|
|
model.COLUMNS = columns
|
||
|
|
}
|
||
|
|
}
|