diff --git a/safesite/views.py b/safesite/views.py index 142c7804..3b63478c 100644 --- a/safesite/views.py +++ b/safesite/views.py @@ -1674,11 +1674,10 @@ def getdickey(req): ret=[] for i in r_list: parent=i['parent'] - if parent: - if 'children' in r_dict[parent]: - r_dict[parent]['children'].append(i) - else: - r_dict[parent]['children'] = [i] + if parent in r_dict and 'children' in r_dict[parent]: + r_dict[parent]['children'].append(i) + elif parent in r_dict: + r_dict[parent]['children'] = [i] else: ret.append(i) return HttpResponse(json.dumps(ret), content_type="application/json") @@ -1702,11 +1701,10 @@ def getdickey(req): ret=[] for i in r_list: parent=i['parent'] - if parent: - if 'children' in r_dict[parent]: - r_dict[parent]['children'].append(i) - else: - r_dict[parent]['children'] = [i] + if parent in r_dict and 'children' in r_dict[parent]: + r_dict[parent]['children'].append(i) + elif parent in r_dict: + r_dict[parent]['children'] = [i] else: ret.append(i) return HttpResponse(json.dumps(ret), content_type="application/json")