You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.5 KiB
102 lines
2.5 KiB
8 years ago
|
/*
|
||
|
|
* This file is part of SYZOJ.
|
||
|
|
*
|
||
|
|
* Copyright (c) 2016 Menci <huanghaorui301@gmail.com>
|
||
|
|
*
|
||
|
|
* SYZOJ is free software: you can redistribute it and/or modify
|
||
|
|
* it under the terms of the GNU Affero General Public License as
|
||
|
|
* published by the Free Software Foundation, either version 3 of the
|
||
|
|
* License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* SYZOJ is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
* GNU Affero General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU Affero General Public
|
||
|
|
* License along with SYZOJ. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
|
||
|
|
'use strict';
|
||
|
|
|
||
|
|
class Model {
|
||
|
|
constructor(record) {
|
||
|
|
this.record = record;
|
||
|
|
this.loadFields();
|
||
|
|
}
|
||
|
|
|
||
|
|
loadFields() {
|
||
|
|
let model = this.getModel();
|
||
|
|
let obj = JSON.parse(JSON.stringify(this.record.get({ plain: true })));
|
||
|
|
for (let key in obj) {
|
||
|
|
if (model.tableAttributes[key].json) {
|
||
|
|
try {
|
||
|
|
this[key] = eval(`(${obj[key]})`);
|
||
|
|
} catch (e) {
|
||
|
|
this[key] = {};
|
||
|
|
}
|
||
|
|
} else this[key] = obj[key];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
toPlain() {
|
||
|
|
let model = this.getModel();
|
||
|
|
let obj = JSON.parse(JSON.stringify(this.record.get({ plain: true })));
|
||
|
|
for (let key in obj) {
|
||
|
|
if (model.tableAttributes[key].json) obj[key] = JSON.stringify(this[key]);
|
||
|
|
else obj[key] = this[key];
|
||
|
|
}
|
||
|
|
return obj;
|
||
|
|
}
|
||
|
|
|
||
|
|
async save() {
|
||
|
|
let obj = this.toPlain();
|
||
|
|
for (let key in obj) this.record.set(key, obj[key]);
|
||
|
|
|
||
|
|
let isNew = this.record.isNewRecord;
|
||
|
|
await this.record.save();
|
||
|
|
if (!isNew) return;
|
||
|
|
|
||
|
|
await this.reload();
|
||
|
|
}
|
||
|
|
|
||
|
|
async reload() {
|
||
|
|
await this.record.reload();
|
||
|
|
this.loadFields();
|
||
|
|
}
|
||
|
|
|
||
|
|
async destroy() {
|
||
|
|
return this.record.destroy();
|
||
|
|
}
|
||
|
|
|
||
|
|
static async fromRecord(record) {
|
||
|
|
record = await record;
|
||
|
|
if (!record) return null;
|
||
|
|
return new this(await record);
|
||
|
|
}
|
||
|
|
|
||
|
|
static async fromID(id) {
|
||
|
|
return this.fromRecord(this.model.findById(id))
|
||
|
|
}
|
||
|
|
|
||
|
|
static async findOne(options) {
|
||
|
|
return this.fromRecord(this.model.findOne(options));
|
||
|
|
}
|
||
|
|
|
||
|
|
static async count(where) {
|
||
|
|
return this.model.count({ where: where });
|
||
|
|
}
|
||
|
|
|
||
|
|
static async query(page, perpage, where, order) {
|
||
|
|
let records = await this.model.findAll({
|
||
|
|
offset: (page - 1) * perpage,
|
||
|
|
limit: perpage,
|
||
|
|
where: where,
|
||
|
|
order: order
|
||
|
|
});
|
||
|
|
return records.mapAsync(record => (this.fromRecord(record)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = Model;
|