github
/
element
mirror of https://github.com/ElemeFE/element
1
0
Fork 0
Code Issues Releases Wiki Activity

Tree: fix setCheckedKeys bug

pull/5962/head
Dreamacro 2017-07-20 11:49:46 +08:00 committed by 杨奕
parent 02459f1833
commit cdd4aa7a8b
2 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/tree/src/model/node.js
View File

@ -5,8 +5,8 @@ export const getChildState = node => {
let all = true;
let none = true;
let allWithoutDisable = true;
for (let n of node) {
for (let i = 0, j = node.length; i < j; i++) {
const n = node[i];
if (n.checked !== true || n.indeterminate) {
all = false;
if (!n.disabled) {
@ -23,7 +23,6 @@ export const getChildState = node => {
const reInitChecked = function(node) {
const {all, none, half} = getChildState(node.childNodes);
if (all) {
node.checked = true;
node.indeterminate = false;

35
packages/tree/src/model/tree-store.js
View File

@ -1,4 +1,4 @@
import Node, { getChildState } from './node';
import Node from './node';
import { getNodeKey } from './util';
export default class TreeStore {
@ -189,35 +189,36 @@ export default class TreeStore {
_setCheckedKeys(key, leafOnly = false, checkedKeys) {
const allNodes = this._getAllNodes().sort((a, b) => b.level - a.level);
const cache = Object.create(null);
const keys = Object.keys(checkedKeys);
for (let node of allNodes) {
let checked = keys.indexOf(node.data[key] + '') > -1;
allNodes.forEach(node => node.setChecked(false, false));
for (let i = 0, j = allNodes.length; i < j; i++) {
const node = allNodes[i];
const nodeKey = node.data[key].toString();
let checked = keys.indexOf(nodeKey) > -1;
if (!checked) {
node.setChecked(false, false);
if (node.checked && !cache[nodeKey]) {
node.setChecked(false, false);
}
continue;
}
let parent = node.parent;
while (parent && parent.level > 0) {
cache[parent.data[key]] = true;
parent = parent.parent;
}
if (node.isLeaf || this.checkStrictly) {
node.setChecked(checked, false);
node.setChecked(true, false);
continue;
}
const { all, none, half } = getChildState(node.childNodes);
if (all) {
node.setChecked(true, !this.checkStrictly);
} else if (half) {
checked = checked ? true : 'half';
node.setChecked(checked, !this.checkStrictly && checked === true);
} else if (none) {
node.setChecked(checked, !this.checkStrictly);
}
node.setChecked(true, true);
if (leafOnly) {
node.setChecked(false, false);
const traverse = function(node) {
const childNodes = node.childNodes;
childNodes.forEach((child) => {
if (!child.isLeaf) {
child.setChecked(false, false);